diff options
| author | Dana Powers <dana.powers@rd.io> | 2016-01-24 22:01:09 -0800 |
|---|---|---|
| committer | Dana Powers <dana.powers@rd.io> | 2016-01-25 10:09:15 -0800 |
| commit | 0d5899020a75e22fec14d3e3d7aec8f043d60a31 (patch) | |
| tree | e227d12865560026a3170196b331d4d843a9e7fd /kafka/codec.py | |
| parent | 2c7b7452a8ca761672e70ee56b3779e4a96c1997 (diff) | |
| download | kafka-python-0d5899020a75e22fec14d3e3d7aec8f043d60a31.tar.gz | |
Add support for LZ4 compressed messages using python-lz4 module
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 |