summaryrefslogtreecommitdiff
path: root/kafka/codec.py
diff options
context:
space:
mode:
Diffstat (limited to 'kafka/codec.py')
-rw-r--r--kafka/codec.py13
1 files changed, 13 insertions, 0 deletions
diff --git a/kafka/codec.py b/kafka/codec.py
index c27d89b..c8195ee 100644
--- a/kafka/codec.py
+++ b/kafka/codec.py
@@ -13,6 +13,15 @@ try:
except ImportError:
_HAS_SNAPPY = False
+try:
+ import lz4
+ from lz4 import compress as lz4_encode
+ from lz4 import decompress as lz4_decode
+except ImportError:
+ lz4 = None
+ lz4_encode = None
+ lz4_decode = None
+
def has_gzip():
return True
@@ -22,6 +31,10 @@ def has_snappy():
return _HAS_SNAPPY
+def has_lz4():
+ return lz4 is not None
+
+
def gzip_encode(payload, compresslevel=None):
if not compresslevel:
compresslevel = 9