pep/slixmpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

xmlstream: handle done tasks in wait_until

Browse source 
and handle other loops properly
This commit is contained in:
mathieui 2021-01-23 15:41:13 +01:00
parent 370abb1d98
commit 0eed84d0b2
1 changed files with 8 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
slixmpp/xmlstream/xmlstream.py
View file

@ -1142,9 +1142,15 @@ class XMLStream(asyncio.BaseProtocol):
:param int timeout: Timeout :param int timeout: Timeout
""" """
fut = asyncio.Future() fut = asyncio.Future()
def result_handler(event_data):
if not fut.done():
fut.set_result(event_data)
else:
log.debug("Future registered on event '%s' was alredy done", event)
self.add_event_handler( self.add_event_handler(
event, event,
fut.set_result, result_handler,
disposable=True, disposable=True,
) )
return await asyncio.wait_for(fut, timeout) return await asyncio.wait_for(fut, timeout, loop=self.loop)