Slightly improve exiting of tcp connections

This commit is contained in:
Tony Garnock-Jones 2021-08-19 17:27:33 -04:00
parent aa2baa62f1
commit 79f433167c
1 changed files with 6 additions and 2 deletions

View File

@ -333,8 +333,12 @@ class _StreamTunnelRelay(TunnelRelay, asyncio.Protocol):
def _disconnect(self):
if self.stop_signal:
self.stop_signal.get_loop().call_soon_threadsafe(
lambda: self.stop_signal.set_result(True))
def set_stop_signal():
try:
self.stop_signal.set_result(True)
except:
pass
self.stop_signal.get_loop().call_soon_threadsafe(set_stop_signal)
async def _create_connection(self, loop):
raise Exception('subclassresponsibility')