xCAT/confluent
2
0
Fork 0
mirror of https://github.com/xcat2/confluent.git synced 2025-01-12 18:59:06 +00:00
Code Issues Projects Releases Wiki Activity

Protect against stale data in new requests

Browse Source 
If a caller (reasonably) broke out of a loop, a subsequent call would get old data.
Protect against this by discarding data not consumed if previously called.
This commit is contained in:
Jarrod Johnson 2017-02-10 11:20:39 -05:00
parent 9e4bb84932
commit ffbe1ab156
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
confluent_client/confluent/client.py
View File

@ -27,6 +27,8 @@ import confluent.tlvdata as tlvdata
SO_PASSCRED = 16
inflight = False
def _parseserver(string):
if ']:' in string:
server, port = string[1:].split(']:')
@ -226,6 +228,15 @@ def send_request(operation, path, server, parameters=None):
:param server: The socket to send data over
:param parameters: Parameters if any to send along with the request
"""
global inflight
if inflight:
# calling code failed to complete a transaction, flush and discard
# their unused data
result = tlvdata.recv(server)
while '_requestdone' not in result:
yield result
result = tlvdata.recv(server)
inflight = True
payload = {'operation': operation, 'path': path}
if parameters is not None:
payload['parameters'] = parameters
@ -234,6 +245,7 @@ def send_request(operation, path, server, parameters=None):
while '_requestdone' not in result:
yield result
result = tlvdata.recv(server)
inflight = False