alemi/aiocraft
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

safer disconnect sequence

Browse source
This commit is contained in:
əlemi 2021-11-19 16:30:08 +01:00
parent 1b72f94043
commit cf35e2cf24
1 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
aiocraft/dispatcher.py
View file

@ -109,8 +109,11 @@ class Dispatcher:
await asyncio.gather(self._writer, self._reader)
self._logger.debug("Net workers stopped")
if self._up:
if self._up.can_write_eof():
self._up.write_eof()
if not self._up.is_closing() and self._up.can_write_eof():
try:
self._up.write_eof()
except OSError as e:
self._logger.error("Could not write EOF : %s", str(e))
self._up.close()
if block:
await self._up.wait_closed()