diff options
Diffstat (limited to 'kafka')
| -rw-r--r-- | kafka/client.py | 2 | ||||
| -rw-r--r-- | kafka/conn.py | 3 |
2 files changed, 3 insertions, 2 deletions
diff --git a/kafka/client.py b/kafka/client.py index 23b0a48..1c7fc93 100644 --- a/kafka/client.py +++ b/kafka/client.py @@ -116,7 +116,7 @@ class KafkaClient(object): for payload in payloads: payloads_by_broker[self._get_leader_for_partition(payload.topic, payload.partition)].append(payload) original_keys.append((payload.topic, payload.partition)) - + # Accumulate the responses in a dictionary acc = {} diff --git a/kafka/conn.py b/kafka/conn.py index 203488b..c9ba4cc 100644 --- a/kafka/conn.py +++ b/kafka/conn.py @@ -71,8 +71,9 @@ class KafkaConnection(object): def send(self, requestId, payload): "Send a request to Kafka" + log.debug("About to send %d bytes to Kafka" % len(payload)) sent = self._sock.sendall(payload) - if sent == 0: + if sent != None: raise RuntimeError("Kafka went away") self.data = self._consume_response() |