diff options
| author | Eric Hewitt <eric@understoryweather.com> | 2015-01-02 11:34:38 -0500 |
|---|---|---|
| committer | Eric Hewitt <eric@understoryweather.com> | 2015-01-02 11:34:38 -0500 |
| commit | 6603088f0adb4489d3031cfbbbf2ccd3d1dc55be (patch) | |
| tree | 9ed2d4143bd8cdae99cf5260b1cf37e5eda5b7e4 | |
| parent | 9c5216a97fde23c2619d3c4d72cab3f912949fbf (diff) | |
| download | kafka-python-6603088f0adb4489d3031cfbbbf2ccd3d1dc55be.tar.gz | |
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 |