Skip to content

Commit 5446d18

Browse files
everpcpcdpkp
authored andcommitted
Support alternative lz4framed (#1395)
1 parent d90cf22 commit 5446d18

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

kafka/codec.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,11 @@
2626
except ImportError:
2727
lz4f = None
2828

29+
try:
30+
import lz4framed
31+
except ImportError:
32+
lz4framed = None
33+
2934
try:
3035
import xxhash
3136
except ImportError:
@@ -46,6 +51,8 @@ def has_lz4():
4651
return True
4752
if lz4f is not None:
4853
return True
54+
if lz4framed is not None:
55+
return True
4956
return False
5057

5158

@@ -198,6 +205,8 @@ def snappy_decode(payload):
198205
lz4_encode = lz4.compress # pylint: disable-msg=no-member
199206
elif lz4f:
200207
lz4_encode = lz4f.compressFrame # pylint: disable-msg=no-member
208+
elif lz4framed:
209+
lz4_encode = lz4framed.compress # pylint: disable-msg=no-member
201210
else:
202211
lz4_encode = None
203212

@@ -220,6 +229,8 @@ def lz4f_decode(payload):
220229
lz4_decode = lz4.decompress # pylint: disable-msg=no-member
221230
elif lz4f:
222231
lz4_decode = lz4f_decode
232+
elif lz4framed:
233+
lz4_decode = lz4framed.decompress # pylint: disable-msg=no-member
223234
else:
224235
lz4_decode = None
225236

0 commit comments

Comments
 (0)