gandc/mtprotoproxy
1
0
Fork 0
mirror of https://github.com/alexbers/mtprotoproxy.git synced 2026-03-14 07:13:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

better handling for server-side socket closing in connection pool

Browse Source
This commit is contained in:
Alexander Bersenev
2026-02-17 21:14:42 +05:00
parent 34f743858c
commit 8920faf650
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
mtprotoproxy.py
View File

@@ -483,7 +483,7 @@ myrandom = MyRandom()
class TgConnectionPool:
MAX_CONNS_IN_POOL = 64
MAX_CONNS_IN_POOL = 16
def __init__(self):
self.pools = {}
@@ -500,6 +500,16 @@ class TgConnectionPool:
timeout=config.TG_CONNECT_TIMEOUT)
return reader_tgt, writer_tgt
def is_conn_dead(self, reader, writer):
if writer.transport.is_closing():
return True
raw_reader = reader
while hasattr(raw_reader, 'upstream'):
raw_reader = raw_reader.upstream
if raw_reader.at_eof():
return True
return False
def register_host_port(self, host, port, init_func):
if (host, port, init_func) not in self.pools:
self.pools[(host, port, init_func)] = []
@@ -519,8 +529,9 @@ class TgConnectionPool:
continue
reader, writer, *other = task.result()
if writer.transport.is_closing():
if self.is_conn_dead(reader, writer):
self.pools[(host, port, init_func)].remove(task)
writer.transport.abort()
continue
if not ret: