mirror of
https://github.com/Flowseal/tg-ws-proxy.git
synced 2026-08-08 23:33:19 +03:00
hard limit workers pool
This commit is contained in:
+2
-1
@@ -218,6 +218,7 @@ class _WsPool:
|
|||||||
|
|
||||||
class _CfWorkerPool:
|
class _CfWorkerPool:
|
||||||
WS_POOL_MAX_AGE = 100.0
|
WS_POOL_MAX_AGE = 100.0
|
||||||
|
PER_DC_LIMIT = 1
|
||||||
|
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
self._idle: Dict[Tuple[int, str], deque] = {}
|
self._idle: Dict[Tuple[int, str], deque] = {}
|
||||||
@@ -258,7 +259,7 @@ class _CfWorkerPool:
|
|||||||
dc, worker_domain = key
|
dc, worker_domain = key
|
||||||
try:
|
try:
|
||||||
bucket = self._idle.setdefault(key, deque())
|
bucket = self._idle.setdefault(key, deque())
|
||||||
needed = proxy_config.pool_size - len(bucket)
|
needed = min(proxy_config.pool_size - len(bucket), self.PER_DC_LIMIT)
|
||||||
if needed <= 0:
|
if needed <= 0:
|
||||||
return
|
return
|
||||||
tasks = [asyncio.create_task(
|
tasks = [asyncio.create_task(
|
||||||
|
|||||||
Reference in New Issue
Block a user