From 209956865bebb9903e53450b5182aa7174c7686c Mon Sep 17 00:00:00 2001 From: Stefan Allius Date: Thu, 26 Sep 2024 00:11:11 +0200 Subject: [PATCH] add two more callbacks --- app/src/async_stream.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/app/src/async_stream.py b/app/src/async_stream.py index 0cc449c..1e1ac84 100644 --- a/app/src/async_stream.py +++ b/app/src/async_stream.py @@ -36,8 +36,6 @@ class AsyncIfcImpl(AsyncIfc): def close(self): self.timeout_cb = None - self.init_new_client_conn_cb = None - self.update_header_cb = None self.fwd_fifo.reg_trigger(None) self.tx_fifo.reg_trigger(None) self.rx_fifo.reg_trigger(None) @@ -351,12 +349,11 @@ class AsyncStream(AsyncIfcImpl): await self.remote.ifc.__async_write() if self.remote.stream: - if self.remote.ifc.update_header_cb is callable: - self.remote.ifc.update_header_cb(self.fwd_fifo.peek()) - self.fwd_fifo.logging(logging.INFO, 'Forward to ' - f'{self.remote.ifc.addr}:') - self.remote.ifc._writer.write(self.fwd_fifo.get()) - await self.remote.ifc._writer.drain() + self.remote.ifc.update_header_cb(self.fwd_fifo.peek()) + self.fwd_fifo.logging(logging.INFO, 'Forward to ' + f'{self.remote.ifc.addr}:') + self.remote.ifc._writer.write(self.fwd_fifo.get()) + await self.remote.ifc._writer.drain() except OSError as error: if self.remote.stream: