diff options
Diffstat (limited to 'kafka/codec.py')
-rw-r--r-- | kafka/codec.py | 13 |
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 |