解决非标准的gzip压缩 (#722)

* 解决非标准的gzip压缩解决非标准的gzip格式在body_encoding == gzip 时, 使用了deflate压缩, 没有使用标准的gzip压缩, 故做了兼容, 防伪未来服务端修复这个bug * chore: remove trailing spaces Signed-off-by: Zhanhui Li <[email protected]> --------- Signed-off-by: Zhanhui Li <[email protected]> Co-authored-by: Zhanhui Li <[email protected]>
apache · Aug 6, 2024 · 950652b · 950652b
1 parent a363f41
commit 950652b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/python/rocketmq/message.py b/python/rocketmq/message.py
@@ -17,6 +17,7 @@
 import binascii
 import gzip
 import hashlib
+import zlib
 from typing import Dict, List
 
 from rocketmq.definition import MessageQueue
@@ -184,7 +185,10 @@ def from_protobuf(cls, message, message_queue=None):
         body_encoding = system_properties.body_encoding
         body = raw
         if body_encoding == ProtoEncoding.GZIP:
-            body = gzip.decompress(message.body)
+            if message.body and message.body[:2] == b'\x1f\x8b':  # Standard Gzip format
+                body = gzip.decompress(message.body)
+            else:  # deflate zip
+                body = zlib.decompress(message.body)
         elif body_encoding in [ProtoEncoding.IDENTITY, None]:
             pass
         else: