Davte/filebridging
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Test 6

Browse Source
This commit is contained in:
Davte 2020-04-19 19:45:48 +02:00
parent f56fc6d60c
commit bd8eb7da4c
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
filebridging/server.py
View File

@ -73,7 +73,8 @@ class Server:
def set_ssl_context(self, ssl_context: ssl.SSLContext): def set_ssl_context(self, ssl_context: ssl.SSLContext):
self._ssl_context = ssl_context self._ssl_context = ssl_context
async def run_reader(self, reader, connection_token): async def run_reader(self, reader: asyncio.StreamReader, connection_token):
received_bytes = 0
while 1: while 1:
try: try:
# Wait one second if buffer is full # Wait one second if buffer is full
@ -81,6 +82,7 @@ class Server:
await asyncio.sleep(1) await asyncio.sleep(1)
continue continue
input_data = await reader.read(self.buffer_chunk_size) input_data = await reader.read(self.buffer_chunk_size)
received_bytes += len(input_data)
if connection_token not in self.buffers: if connection_token not in self.buffers:
break break
self.buffers[connection_token].append(input_data) self.buffers[connection_token].append(input_data)
@ -89,6 +91,7 @@ class Server:
break break
except Exception as e: except Exception as e:
logging.error(f"Unexpected exception:\n{e}", exc_info=True) logging.error(f"Unexpected exception:\n{e}", exc_info=True)
print(received_bytes)
async def run_writer(self, writer, connection_token): async def run_writer(self, writer, connection_token):
consecutive_interruptions = 0 consecutive_interruptions = 0