From cb3184554b5e61663371f3043ac22214046e3713 Mon Sep 17 00:00:00 2001 From: Dana Powers Date: Tue, 19 Aug 2014 22:33:07 -0700 Subject: Shutdown socket before closing in kafka.conn --- kafka/conn.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'kafka/conn.py') diff --git a/kafka/conn.py b/kafka/conn.py index 2e37ecf..33c799c 100644 --- a/kafka/conn.py +++ b/kafka/conn.py @@ -155,6 +155,15 @@ class KafkaConnection(local): """ log.debug("Closing socket connection for %s:%d" % (self.host, self.port)) if self._sock: + # Call shutdown to be a good TCP client + # But expect an error if the socket has already been + # closed by the server + try: + self._sock.shutdown(socket.SHUT_RDWR) + except socket.error: + pass + + # Closing the socket should always succeed self._sock.close() self._sock = None else: -- cgit v1.2.1