actions/opensearch-pyd
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

close thrift connection upon exception

Browse Source
This commit is contained in:
Michael Schier
2014-09-23 01:40:04 +12:00
parent 8a4c990065
commit 2a1aac8c3b
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
elasticsearch/connection/thrift.py
+5 -3
View File
@@ -49,6 +49,7 @@ class ThriftConnection(PoolingConnection):
protocol = TBinaryProtocol.TBinaryProtocolAccelerated(transport) protocol = TBinaryProtocol.TBinaryProtocolAccelerated(transport)
client = Rest.Client(protocol) client = Rest.Client(protocol)
client.transport = transport
transport.open() transport.open()
return client return client
@@ -64,10 +65,11 @@ class ThriftConnection(PoolingConnection):
duration = time.time() - start duration = time.time() - start
except (TException, SocketTimeout) as e: except (TException, SocketTimeout) as e:
self.log_request_fail(method, url, body, time.time() - start, exception=e) self.log_request_fail(method, url, body, time.time() - start, exception=e)
raise ConnectionError('N/A', str(e), e)
finally:
if tclient: if tclient:
self._release_connection(tclient) tclient.transport.close()
raise ConnectionError('N/A', str(e), e)
self._release_connection(tclient)
if not (200 <= response.status < 300) and response.status not in ignore: if not (200 <= response.status < 300) and response.status not in ignore:
self.log_request_fail(method, url, body, duration, response.status) self.log_request_fail(method, url, body, duration, response.status)