diff options
author | everpcpc <git@everpcpc.com> | 2018-03-10 00:21:32 +0800 |
---|---|---|
committer | Dana Powers <dana.powers@gmail.com> | 2018-03-09 11:21:32 -0500 |
commit | 5446d1832c07f436cb306b2d27e8d74b6906b363 (patch) | |
tree | 91567bdb2429d7d6cdf09da92764210e978b73e9 | |
parent | d90cf22bcf5c49b6d75cac4282d7cc2ae973651c (diff) | |
download | kafka-python-5446d1832c07f436cb306b2d27e8d74b6906b363.tar.gz |
Support alternative lz4framed (#1395)
-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 |