sv/tsun-gen3-proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add two more callbacks

Browse Source
This commit is contained in:
Stefan Allius
2024-09-26 00:11:11 +02:00
parent bc54944077
commit 209956865b
1 changed files with 5 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
app/src/async_stream.py
View File

@@ -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: