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

use socket from pool in the creation order

Browse Source
This commit is contained in:
Alexander Bersenev
2019-11-12 04:31:47 +05:00
parent e436792992
commit 8c15fc8fe0
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
mtprotoproxy.py
View File

@@ -449,29 +449,34 @@ class TgConnectionPool:
def register_host_port(self, host, port): def register_host_port(self, host, port):
if (host, port) not in self.pools: if (host, port) not in self.pools:
self.pools[(host, port)] = set() self.pools[(host, port)] = []
while len(self.pools[(host, port)]) < TgConnectionPool.MAX_CONNS_IN_POOL: while len(self.pools[(host, port)]) < TgConnectionPool.MAX_CONNS_IN_POOL:
connect_task = asyncio.ensure_future(self.open_tg_connection(host, port)) connect_task = asyncio.ensure_future(self.open_tg_connection(host, port))
self.pools[(host, port)].add(connect_task) self.pools[(host, port)].append(connect_task)
async def get_connection(self, host, port): async def get_connection(self, host, port):
self.register_host_port(host, port) self.register_host_port(host, port)
for task in self.pools[(host, port)].copy(): ret = None
for task in self.pools[(host, port)][::]:
if task.done(): if task.done():
self.pools[(host, port)].remove(task)
self.register_host_port(host, port)
if task.exception(): if task.exception():
self.pools[(host, port)].remove(task)
continue continue
reader, writer = task.result() reader, writer = task.result()
if writer.transport.is_closing(): if writer.transport.is_closing():
self.pools[(host, port)].remove(task)
continue continue
return reader, writer if not ret:
self.pools[(host, port)].remove(task)
ret = (reader, writer)
self.register_host_port(host, port)
if ret:
return ret
return await self.open_tg_connection(host, port) return await self.open_tg_connection(host, port)