133 lines
4.2 KiB
Python
133 lines
4.2 KiB
Python
"""网关配置加载与校验。
|
|
|
|
从 gateway/config.json 读取配置，缺字段给默认值，启动时校验必填项。
|
|
"""
|
|
|
|
import json
|
|
import logging
|
|
from pathlib import Path
|
|
from typing import List
|
|
|
|
logger = logging.getLogger("gateway.config")
|
|
|
|
CONFIG_PATH = Path(__file__).resolve().parent / "config.json"
|
|
|
|
DEFAULTS = {
|
|
"public_base_url": "https://hair.xiangsilian.com",
|
|
"static_dir": "static/annotations",
|
|
"health_check": {
|
|
"path": "/health",
|
|
"interval_seconds": 8,
|
|
"timeout_seconds": 3,
|
|
"unhealthy_threshold": 2,
|
|
"healthy_threshold": 1,
|
|
},
|
|
"dispatch": {
|
|
"per_worker_concurrency": 1,
|
|
"max_global_concurrency": 1,
|
|
"max_queue_wait_seconds": 590,
|
|
"queue_wait_seconds": 30,
|
|
"request_timeout_seconds": 600,
|
|
"retry_on_failure": True,
|
|
"max_retries": 1,
|
|
},
|
|
"request_log": {
|
|
"enabled": True,
|
|
"log_file": "gateway/request_log.jsonl",
|
|
"buffer_size": 2000,
|
|
"max_file_lines": 10000,
|
|
"max_file_age_days": 7,
|
|
},
|
|
}
|
|
|
|
_config_cache = None
|
|
|
|
|
|
def _deep_merge(defaults: dict, overrides: dict) -> dict:
|
|
"""递归合并：overrides 中的值覆盖 defaults，嵌套 dict 递归处理。"""
|
|
result = defaults.copy()
|
|
for key, value in overrides.items():
|
|
if key in result and isinstance(result[key], dict) and isinstance(value, dict):
|
|
result[key] = _deep_merge(result[key], value)
|
|
else:
|
|
result[key] = value
|
|
return result
|
|
|
|
|
|
def load_config() -> dict:
|
|
"""加载并校验配置文件，结果缓存仅加载一次。"""
|
|
global _config_cache
|
|
if _config_cache is not None:
|
|
return _config_cache
|
|
|
|
if not CONFIG_PATH.exists():
|
|
raise FileNotFoundError(
|
|
f"配置文件不存在: {CONFIG_PATH}\n"
|
|
f"请从 gateway/config.example.json 复制并修改: "
|
|
f"cp gateway/config.example.json gateway/config.json"
|
|
)
|
|
|
|
with open(CONFIG_PATH, "r", encoding="utf-8") as f:
|
|
raw = json.load(f)
|
|
|
|
# 合并默认值
|
|
cfg = _deep_merge(DEFAULTS, raw)
|
|
|
|
# --- 校验 ---
|
|
workers: List[str] = cfg.get("workers", [])
|
|
if not workers:
|
|
raise ValueError("配置错误: workers 列表不能为空，至少需要一个 worker 地址")
|
|
|
|
for i, w in enumerate(workers):
|
|
if not isinstance(w, str) or not w:
|
|
raise ValueError(f"配置错误: workers[{i}] 必须是非空字符串")
|
|
|
|
password: str = cfg.get("shared_password", "")
|
|
if not password or password == "REPLACE_ME_ROTATE_PERIODICALLY":
|
|
logger.warning(
|
|
"⚠ 安全警告: shared_password 未设置或仍为占位值 "
|
|
"'REPLACE_ME_ROTATE_PERIODICALLY'，请立即更换！"
|
|
)
|
|
if len(password) < 8:
|
|
logger.warning("⚠ 安全警告: shared_password 长度不足 8 位，建议使用更长的密码")
|
|
|
|
public_base_url: str = cfg.get("public_base_url", "")
|
|
if public_base_url.endswith("/"):
|
|
cfg["public_base_url"] = public_base_url.rstrip("/")
|
|
logger.warning("public_base_url 末尾含 '/'，已自动去除")
|
|
|
|
# 确保 static_dir 是绝对路径
|
|
static_dir = Path(cfg["static_dir"])
|
|
if not static_dir.is_absolute():
|
|
cfg["static_dir"] = str(Path(__file__).resolve().parents[1] / static_dir)
|
|
|
|
logger.info(
|
|
"配置加载完成 | workers=%s | public_base_url=%s | "
|
|
"hc_interval=%ds | dispatch_timeout=%ds | queue_wait=%ds | "
|
|
"global_concurrency=%d | max_queue_wait=%ds",
|
|
cfg["workers"],
|
|
cfg["public_base_url"],
|
|
cfg["health_check"]["interval_seconds"],
|
|
cfg["dispatch"]["request_timeout_seconds"],
|
|
cfg["dispatch"]["queue_wait_seconds"],
|
|
cfg["dispatch"]["max_global_concurrency"],
|
|
cfg["dispatch"]["max_queue_wait_seconds"],
|
|
)
|
|
|
|
_config_cache = cfg
|
|
return cfg
|
|
|
|
|
|
def get_config() -> dict:
|
|
"""获取已加载的配置（必须先调用 load_config）。"""
|
|
if _config_cache is None:
|
|
raise RuntimeError("配置尚未加载，请先调用 load_config()")
|
|
return _config_cache
|
|
|
|
|
|
def reload_config() -> dict:
|
|
"""强制重新加载配置（用于热更新）。"""
|
|
global _config_cache
|
|
_config_cache = None
|
|
return load_config()
|