Merge pull request #518 from dpkp/lz4

Add support for LZ4 compression / decompression

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