diff --git a/app/src/async_stream.py b/app/src/async_stream.py index 2650235..b023c23 100644 --- a/app/src/async_stream.py +++ b/app/src/async_stream.py @@ -221,7 +221,6 @@ class AsyncStream(AsyncIfcImpl): async def disc(self) -> None: """Async disc handler for graceful disconnect""" - self.remote = None if self._writer.is_closing(): return logger.debug(f'AsyncStream.disc() l{self.l_addr} | r{self.r_addr}') @@ -370,6 +369,11 @@ class AsyncStreamClient(AsyncStream): AsyncStream.__init__(self, reader, writer, rstream) self.close_cb = close_cb + async def disc(self) -> None: + logging.debug('AsyncStreamClient.disc()') + self.remote = None + await super().disc() + def close(self) -> None: logging.debug('AsyncStreamClient.close()') self.close_cb = None