perf: 注册开通复用进程内管理员会话

实测(真网关):单次开通链路约 0.29s,其中管理员登录 111ms、
建号 80ms、用户登录 79ms、查组 5ms、发 key 10ms、授余额 9ms。
外部 OAuth 两步为必需串行调用;查组/授余额仅个位数毫秒,
不值得为其加缓存复杂度,故本批只做一项:同一进程内第二
次起的开通复用管理员 token 元组(实例不跨线程,401 时走
刷新->重登录自愈,密码轮换后自动用新密码),免一次登录。
真网关对照:冷开通 0.294s,热开通 0.179s(约 -39%),
对照组每次新登录两次均为 0.255s 左右。
测试:新增共享会话复用/过期重取 2 个;全量 cloud 794
(4 跳过)、host 69、deploy 19 全绿。
This commit is contained in:
host committed 2026-10-27 22:16:43 +08:00
1 parent cc0c707ccd
commit a9ff7cb498
2 files changed
+110 -3

No files matched your search

+59 -3
View File
@@ -16,6 +16,7 @@
import json
import os
import secrets
import threading
import time
import urllib.error
import urllib.request
@@ -112,8 +113,57 @@ def get_user_gateway_key(conn, user_id):
return None
# 进程内共享管理员会话:(origin, email) -> (token, refresh, expires_at)。
# 只复用 token 元组,不共用实例(实例不跨线程);注册与首任务开通
# 每次新建实例都要登录一次(实测约 111ms,占开通链路约 1/3),
# 共享后同一进程内第二次起免登录。401/刷新失败会回落到重登录,
# 密码轮换后下一次登录即用新密码刷新缓存,不会卡死。
_SHARED_ADMIN = {}
_SHARED_ADMIN_LOCK = threading.Lock()
def _store_shared_session(admin):
key = getattr(admin, "_shared_key", None)
if key is None or not admin._token:
return
with _SHARED_ADMIN_LOCK:
_SHARED_ADMIN[key] = (admin._token, admin._refresh,
admin._expires_at)
def _evict_shared_session(admin):
key = getattr(admin, "_shared_key", None)
if key is None:
return
with _SHARED_ADMIN_LOCK:
_SHARED_ADMIN.pop(key, None)
def shared_admin_session(request=None):
"""带进程内 token 复用的管理员会话(生产开通路径默认用它)。
缓存命中且未过期时直接播种 token,免一次管理员登录;
未命中/已过期时照常登录,成功后写回缓存。调用方显式传
admin 的单测不受影响(不走这条工厂)。
"""
admin = Sub2apiAdmin.from_env(request=request)
key = (admin.origin, admin.email)
admin._shared_key = key
with _SHARED_ADMIN_LOCK:
cached = _SHARED_ADMIN.get(key)
if cached and cached[2] > time.time():
admin._token, admin._refresh, admin._expires_at = cached
return admin
class Sub2apiAdmin:
"""管理员会话:登录、建/删用户、查组。token 自动刷新一次。"""
"""管理员会话:登录、建/删用户、查组。token 自动刷新一次。
每次实例只在进程内复用 token 元组(见 shared_admin_session):
实例本身不跨线程共用,只复用 (token, refresh, expires_at),
401 时走已有刷新->重登录链路自愈,密码轮换后下一次登录
即用新密码并刷新缓存,不会卡在旧 token 上。
"""
def __init__(self, origin, email, password, request=None):
if not origin:
@@ -127,6 +177,7 @@ class Sub2apiAdmin:
self._token = None
self._refresh = None
self._expires_at = 0
self._shared_key = None
@classmethod
def from_env(cls, request=None):
@@ -141,6 +192,7 @@ class Sub2apiAdmin:
{"email": self.email, "password": self.password})
payload = (data or {}).get("data") or {}
if st != 200 or not payload.get("access_token"):
_evict_shared_session(self)
raise Sub2apiError(
"sub2api 管理员登录失败: %s"
% ((data or {}).get("message") or st), status=st)
@@ -151,6 +203,7 @@ class Sub2apiAdmin:
except (TypeError, ValueError):
ttl = 3600
self._expires_at = time.time() + max(60, ttl - 60)
_store_shared_session(self)
def _refresh_token(self):
if not self._refresh:
@@ -169,6 +222,7 @@ class Sub2apiAdmin:
except (TypeError, ValueError):
ttl = 3600
self._expires_at = time.time() + max(60, ttl - 60)
_store_shared_session(self)
return
self._login()
@@ -310,7 +364,9 @@ def ensure_user_gateway_key(conn, user_id, admin=None, request=None):
入库失败时尽力清理刚建的 key 与账号(避免孤儿 key),再抛原错;
授余额失败同样走这条清理(余额为 0 的 key 在网关侧直接 403,
留着没有意义)。同一用户并发开通时建号撞 email 已存在,
则轮询复用胜出者入库的行。调用方负责 conn 的提交语义:本函数在入库后
则轮询复用胜出者入库的行。管理员工单默认走进程内共享会话
(同一进程内第二次起免管理员登录,见 shared_admin_session)。
调用方负责 conn 的提交语义:本函数在入库后
commit(开通是独立事务,不跟调用方的业务事务绑在一起)。
"""
ensure_gateway_keys_table(conn)
@@ -320,7 +376,7 @@ def ensure_user_gateway_key(conn, user_id, admin=None, request=None):
origin = gateway_origin()
if not origin:
raise Sub2apiError("SUB2API_BASE_URL 未配置,无法开通网关账号")
admin = admin or Sub2apiAdmin.from_env(request=request)
admin = admin or shared_admin_session(request=request)
req = request or _default_request
email = "codex-%s@codex.local" % user_id
username = "codex-" + str(user_id)[:12]