AaronFiore/electrum
1
0
Fork 0
Code Activity

interface.got_disconnected: change from Future to Event

Browse Source 
Events are far easier to reason about.
This commit is contained in:
SomberNight
2021-01-21 03:11:15 +01:00
parent e365eb0d85
commit b1449a0a39
2 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
electrum/network.py
View File

@@ -820,20 +820,20 @@ class Network(Logger, NetworkRetryManager[ServerAddr]):
assert iface.ready.done(), "interface not ready yet"
# try actual request
success_fut = asyncio.ensure_future(func(self, *args, **kwargs))
await asyncio.wait([success_fut, iface.got_disconnected], return_when=asyncio.FIRST_COMPLETED)
await asyncio.wait([success_fut, iface.got_disconnected.wait()], return_when=asyncio.FIRST_COMPLETED)
if success_fut.done() and not success_fut.cancelled():
if success_fut.exception():
try:
raise success_fut.exception()
except RequestTimedOut:
await iface.close()
await iface.got_disconnected
await iface.got_disconnected.wait()
continue # try again
except RequestCorrupted as e:
# TODO ban server?
iface.logger.exception(f"RequestCorrupted: {e}")
await iface.close()
await iface.got_disconnected
await iface.got_disconnected.wait()
continue # try again
return success_fut.result()
# otherwise; try again