Update kafka-src to latest 0.8

Fix a broken test (100k was too much to send in one batch)

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()