From 45946d872fbbaff5b3d05db124283ec88d95dfbe Mon Sep 17 00:00:00 2001
From: Dana Powers <dana.powers@rd.io>
Date: Sun, 17 Aug 2014 20:02:47 -0700
Subject: Initialized _dirty in KafkaConnection __init__() and set _sock to
 None in close()

---
 kafka/conn.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

(limited to 'kafka/conn.py')

diff --git a/kafka/conn.py b/kafka/conn.py
index d54d70a..3a94581 100644
--- a/kafka/conn.py
+++ b/kafka/conn.py
@@ -55,6 +55,7 @@ class KafkaConnection(local):
         self.host = host
         self.port = port
         self.timeout = timeout
+        self._dirty = None
         self._sock = None
 
         self.reinit()
@@ -75,7 +76,7 @@ class KafkaConnection(local):
         responses = []
 
         log.debug("About to read %d bytes from Kafka", num_bytes)
-        if self._dirty:
+        if self._dirty or not self._sock:
             self.reinit()
 
         while bytes_left:
@@ -105,7 +106,7 @@ class KafkaConnection(local):
         "Send a request to Kafka"
         log.debug("About to send %d bytes to Kafka, request %d" % (len(payload), request_id))
         try:
-            if self._dirty:
+            if self._dirty or not self._sock:
                 self.reinit()
             sent = self._sock.sendall(payload)
             if sent is not None:
@@ -143,6 +144,7 @@ class KafkaConnection(local):
         """
         if self._sock:
             self._sock.close()
+            self._sock = None
 
     def reinit(self):
         """
-- 
cgit v1.2.1