diff options
author | Mark Roberts <wizzat@gmail.com> | 2015-01-02 09:54:55 -0800 |
---|---|---|
committer | Mark Roberts <wizzat@gmail.com> | 2015-01-02 09:54:55 -0800 |
commit | 02c2b469003e2ddcb051dbb4d95977137050c19f (patch) | |
tree | 9ed2d4143bd8cdae99cf5260b1cf37e5eda5b7e4 | |
parent | 9c5216a97fde23c2619d3c4d72cab3f912949fbf (diff) | |
parent | 6603088f0adb4489d3031cfbbbf2ccd3d1dc55be (diff) | |
download | kafka-python-02c2b469003e2ddcb051dbb4d95977137050c19f.tar.gz |
Merge pull request #281 from meandthewallaby/compressed-key
Added keys to compressed messages (both gzip and snappy).
-rw-r--r-- | kafka/protocol.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/kafka/protocol.py b/kafka/protocol.py index 13b973e..a85c7eb 100644 --- a/kafka/protocol.py +++ b/kafka/protocol.py @@ -568,7 +568,7 @@ def create_gzip_message(payloads, key=None): key: bytes, a key used for partition routing (optional) """ message_set = KafkaProtocol._encode_message_set( - [create_message(payload) for payload in payloads]) + [create_message(payload, key) for payload in payloads]) gzipped = gzip_encode(message_set) codec = ATTRIBUTE_CODEC_MASK & CODEC_GZIP @@ -589,7 +589,7 @@ def create_snappy_message(payloads, key=None): key: bytes, a key used for partition routing (optional) """ message_set = KafkaProtocol._encode_message_set( - [create_message(payload) for payload in payloads]) + [create_message(payload, key) for payload in payloads]) snapped = snappy_encode(message_set) codec = ATTRIBUTE_CODEC_MASK & CODEC_SNAPPY |