From 6603088f0adb4489d3031cfbbbf2ccd3d1dc55be Mon Sep 17 00:00:00 2001 From: Eric Hewitt Date: Fri, 2 Jan 2015 11:34:38 -0500 Subject: Added keys to compressed messages (both gzip and snappy). --- kafka/protocol.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'kafka/protocol.py') 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 -- cgit v1.2.1