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

- refactoring

Browse Source 
- remove _forward_buffer
- make async_write private
This commit is contained in:
Stefan Allius
2024-09-22 14:59:18 +02:00
parent edc0f7a6af
commit 0b437cf3bc
1 changed files with 10 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
app/src/async_stream.py
View File

@@ -8,10 +8,10 @@ from itertools import count
if __name__ == "app.src.async_stream":
from app.src.async_ifc import AsyncIfc
from app.src.messages import hex_dump_memory, State
from app.src.messages import State
else: # pragma: no cover
from async_ifc import AsyncIfc
from messages import hex_dump_memory, State
from messages import State
import gc
@@ -122,7 +122,7 @@ class AsyncStream():
dead_conn_to)
if self.unique_id:
await self.async_write()
await self.__async_write()
await self.__async_forward()
await self.async_publ_mqtt()
@@ -156,7 +156,7 @@ class AsyncStream():
f"{traceback.format_exc()}")
await asyncio.sleep(0) # be cooperative to other task
async def async_write(self, headline: str = 'Transmit to ') -> None:
async def __async_write(self, headline: str = 'Transmit to ') -> None:
"""Async write handler to transmit the send_buffer"""
if len(self.ifc.write) > 0:
self.ifc.write.logging(logging.INFO, f'{headline}{self.addr}:')
@@ -212,24 +212,21 @@ class AsyncStream():
async def __async_forward(self) -> None:
"""forward handler transmits data over the remote connection"""
if not self._forward_buffer:
if len(self.ifc.forward) == 0:
return
try:
if not self.remote_stream:
await self.async_create_remote()
if self.remote_stream:
if self.remote_stream._init_new_client_conn():
await self.remote_stream.async_write()
await self.remote_stream.__async_write()
if self.remote_stream:
self.remote_stream._update_header(self._forward_buffer)
hex_dump_memory(logging.INFO,
f'Forward to {self.remote_stream.addr}:',
self._forward_buffer,
len(self._forward_buffer))
self.remote_stream._writer.write(self._forward_buffer)
self.remote_stream._update_header(self.ifc.forward.peek())
self.ifc.forward.logging(logging.INFO, 'Forward to '
f'{self.remote_stream.addr}:')
self.remote_stream._writer.write(self.ifc.forward.get())
await self.remote_stream._writer.drain()
self._forward_buffer = bytearray(0)
except OSError as error:
if self.remote_stream: