Skip to content

Commit 56f3186

Browse files
authored
ref(profile): remove compression param (#93349)
1 parent 0ded31f commit 56f3186

File tree

1 file changed

+16
-18
lines changed

1 file changed

+16
-18
lines changed

src/sentry/profiles/task.py

Lines changed: 16 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -92,25 +92,24 @@ def _get_profiles_producer_from_topic(topic: Topic) -> KafkaProducer:
9292
logger = logging.getLogger(__name__)
9393

9494

95-
def decode_payload(encoded: str, compressed_profile: bool) -> dict[str, Any]:
96-
if compressed_profile:
97-
try:
98-
res = msgpack.unpackb(
99-
zlib.decompress(b64decode(encoded.encode("utf-8"))), use_list=False
100-
)
101-
metrics.incr("profiling.profile_metrics.decompress", tags={"status": "ok"})
102-
return res
103-
except Exception as e:
104-
logger.exception("Failed to decompress compressed profile", extra={"error": e})
105-
metrics.incr("profiling.profile_metrics.decompress", tags={"status": "err"})
106-
raise
107-
108-
# not compressed
109-
return msgpack.unpackb(b64decode(encoded.encode("utf-8")), use_list=False)
95+
def decode_payload(encoded: str) -> dict[str, Any]:
96+
try:
97+
res = msgpack.unpackb(zlib.decompress(b64decode(encoded.encode("utf-8"))), use_list=False)
98+
metrics.incr("profiling.profile_metrics.decompress", tags={"status": "ok"})
99+
return res
100+
except Exception as e:
101+
logger.exception("Failed to decompress compressed profile", extra={"error": e})
102+
metrics.incr("profiling.profile_metrics.decompress", tags={"status": "err"})
103+
raise
110104

111105

112106
def encode_payload(message: dict[str, Any]) -> str:
113-
return b64encode(msgpack.packb(message)).decode("utf-8")
107+
return b64encode(
108+
zlib.compress(
109+
msgpack.packb(message),
110+
level=1,
111+
)
112+
).decode("utf-8")
114113

115114

116115
@instrumented_task(
@@ -137,14 +136,13 @@ def process_profile_task(
137136
profile: Profile | None = None,
138137
payload: str | None = None,
139138
sampled: bool = True,
140-
compressed_profile: bool = False,
141139
**kwargs: Any,
142140
) -> None:
143141
if not sampled and not options.get("profiling.profile_metrics.unsampled_profiles.enabled"):
144142
return
145143

146144
if payload:
147-
message_dict = decode_payload(payload, compressed_profile)
145+
message_dict = decode_payload(payload)
148146

149147
profile = json.loads(message_dict["payload"], use_rapid_json=True)
150148

0 commit comments

Comments
 (0)