优化白名单逻辑

This commit is contained in:
yml2213
2026-06-25 08:19:18 +08:00
parent ef12dcf4a3
commit a62e7c192f
2 changed files with 33 additions and 16 deletions
+27 -16
View File
@@ -3,10 +3,11 @@
专为短效代理设计:无池、无冷却、无复用。
每次调用 fetch_new_proxy() 从 API 取 1 个新代理,用完即弃。
线程安全:白名单同步有全局锁 + 标志位避免重复同步。
白名单策略:批次启动前调用 warmup_whitelist() 主动同步一次本地出口 IP;
之后 fetch_new_proxy 不再主动检测/同步,遇到代理 API 返回"白名单错误"
时由 ProxyResolver 内部被动同步兜底。
"""
import threading
from typing import Optional
from loguru import logger
@@ -34,22 +35,36 @@ class ProxyFetcher:
if whitelist_credentials
else None
)
# 跨线程共享:是否已同步过白名单,避免多线程重复同步
self._synced = False
self._sync_lock = threading.Lock()
def warmup_whitelist(self) -> tuple[bool, str]:
"""批次启动前调用一次:把当前本地出口 IP 同步到白名单。
Returns:
(是否需要关注的结果, 提示消息)
- 无白名单凭据时返回 (True, "无白名单凭据,跳过")
- 出口 IP 获取失败 / 同步失败时返回 (False, 原因)
- 同步成功时返回 (True, 同步消息)
"""
if not self._whitelist_syncer:
return True, "无白名单凭据,跳过"
local_ip = self._whitelist_syncer.get_local_exit_ip()
if not local_ip:
return False, "获取本地出口 IP 失败"
ok, msg = self._whitelist_syncer.sync_ip(local_ip)
if ok:
return True, f"出口 IP {local_ip} 已同步: {msg}"
return False, f"出口 IP {local_ip} 同步失败: {msg}"
def fetch_new_proxy(self, max_attempts: int = 3) -> Optional[str]:
"""从代理 API 获取 1 个可用代理,失败返回 None。"""
if self._whitelist_syncer:
with self._sync_lock:
already_synced = self._synced
else:
already_synced = False
"""从代理 API 获取 1 个可用代理,失败返回 None。
不主动检测/同步本地出口 IP;如果代理 API 返回"白名单错误"
由 ProxyResolver 内部被动同步兜底。
"""
resolver = ProxyResolver(
api_url=self.api_url,
whitelist_syncer=self._whitelist_syncer,
sync_local_exit_ip=bool(self._whitelist_syncer) and not already_synced,
sync_local_exit_ip=False,
sync_whitelist_once=False,
)
result, msg = resolver.fetch_verified(
@@ -57,10 +72,6 @@ class ProxyFetcher:
return_all=False,
)
if self._whitelist_syncer and result:
with self._sync_lock:
self._synced = True
if isinstance(result, str):
logger.info(f"获取新代理: {result}")
return result