openstack/deb-python-kafka
Code Issues Proposed changes

Merge pull request #281 from meandthewallaby/compressed-key

Browse Source 
Added keys to compressed messages (both gzip and snappy).
This commit is contained in:
Mark Roberts
2015-01-02 09:54:55 -08:00
parent 9c5216a97f 6603088f0a
commit 02c2b46900
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
kafka/protocol.py
View File

@@ -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