From 5446d1832c07f436cb306b2d27e8d74b6906b363 Mon Sep 17 00:00:00 2001 From: everpcpc Date: Sat, 10 Mar 2018 00:21:32 +0800 Subject: Support alternative lz4framed (#1395) --- kafka/codec.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/kafka/codec.py b/kafka/codec.py index a527b42..de15e79 100644 --- a/kafka/codec.py +++ b/kafka/codec.py @@ -26,6 +26,11 @@ try: except ImportError: lz4f = None +try: + import lz4framed +except ImportError: + lz4framed = None + try: import xxhash except ImportError: @@ -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 -- cgit v1.2.1