diff options
| author | Dana Powers <dana.powers@gmail.com> | 2015-03-29 15:36:57 -0700 |
|---|---|---|
| committer | Dana Powers <dana.powers@gmail.com> | 2015-03-29 15:36:57 -0700 |
| commit | 1a5cb0376cfc35d77bbc49072a6d55ca2ee3f7ce (patch) | |
| tree | 101bb990f4a02733fd968841563dd711f3924bfe /kafka/protocol.py | |
| parent | a5b1c8d0fc627de228d00db25ad27fa078c9da32 (diff) | |
| parent | 25ad88cbe68c816cf41ae12d6d6bfc7c2a0926e8 (diff) | |
| download | kafka-python-1a5cb0376cfc35d77bbc49072a6d55ca2ee3f7ce.tar.gz | |
Merge pull request #329 from vshlapakov/feature-batch-msg-keys
Correct message keys for async batching mode
Diffstat (limited to 'kafka/protocol.py')
| -rw-r--r-- | kafka/protocol.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/kafka/protocol.py b/kafka/protocol.py index 2a39de6..b34a95d 100644 --- a/kafka/protocol.py +++ b/kafka/protocol.py @@ -559,7 +559,7 @@ def create_gzip_message(payloads, key=None): """ message_set = KafkaProtocol._encode_message_set( - [create_message(payload, key) for payload in payloads]) + [create_message(payload, pl_key) for payload, pl_key in payloads]) gzipped = gzip_encode(message_set) codec = ATTRIBUTE_CODEC_MASK & CODEC_GZIP @@ -580,7 +580,7 @@ def create_snappy_message(payloads, key=None): """ message_set = KafkaProtocol._encode_message_set( - [create_message(payload, key) for payload in payloads]) + [create_message(payload, pl_key) for payload, pl_key in payloads]) snapped = snappy_encode(message_set) codec = ATTRIBUTE_CODEC_MASK & CODEC_SNAPPY @@ -595,7 +595,7 @@ def create_message_set(messages, codec=CODEC_NONE, key=None): return a list containing a single codec-encoded message. """ if codec == CODEC_NONE: - return [create_message(m, key) for m in messages] + return [create_message(m, k) for m, k in messages] elif codec == CODEC_GZIP: return [create_gzip_message(messages, key)] elif codec == CODEC_SNAPPY: |