diff options
-rw-r--r-- | kafka/codec.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/kafka/codec.py b/kafka/codec.py index a527b42..de15e79 100644 --- a/kafka/codec.py +++ b/kafka/codec.py @@ -27,6 +27,11 @@ except ImportError: lz4f = None try: + import lz4framed +except ImportError: + lz4framed = None + +try: import xxhash except ImportError: xxhash = None @@ -46,6 +51,8 @@ def has_lz4(): return True if lz4f is not None: return True + if lz4framed is not None: + return True return False @@ -198,6 +205,8 @@ if lz4: lz4_encode = lz4.compress # pylint: disable-msg=no-member elif lz4f: lz4_encode = lz4f.compressFrame # pylint: disable-msg=no-member +elif lz4framed: + lz4_encode = lz4framed.compress # pylint: disable-msg=no-member else: lz4_encode = None @@ -220,6 +229,8 @@ if lz4: lz4_decode = lz4.decompress # pylint: disable-msg=no-member elif lz4f: lz4_decode = lz4f_decode +elif lz4framed: + lz4_decode = lz4framed.decompress # pylint: disable-msg=no-member else: lz4_decode = None |