mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-02 01:10:23 +08:00
1d9c77522a
- 删除全部 Python 源码 (src/) 及 Alembic 迁移脚本，归档至 _deprecated_py_src/ - 重构 Rust gateway ai_pipeline: 拆分 planner/finalize 模块，新增 contracts/adaptation 层 - 重组 handlers 模块为 admin/public/proxy/internal/shared 子模块结构 - 新增 executor 模块，引入 Rust 原生数据库迁移 (aether-data/migrations) - 简化 CI/Docker 构建流程，移除 base image 二级构建，统一为单一 app image - 移除 Python 相关基础设施文件 (entrypoint.sh, gunicorn_conf.py, Dockerfile.base)
68 lines
1.8 KiB
Python
68 lines
1.8 KiB
Python
"""
|
|
Tunnel relay 配置。
|
|
|
|
控制 worker 是否通过本机 gateway relay 转发 tunnel 帧。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from dataclasses import dataclass
|
|
from urllib.parse import quote
|
|
|
|
_DEFAULT_DOCKER_TUNNEL_URL = "http://127.0.0.1:8084"
|
|
_DOCKER_TUNNEL_CONNECT_TIMEOUT_SECONDS = 5.0
|
|
_DOCKER_TUNNEL_URL_ENV_KEYS = ("AETHER_TUNNEL_BASE_URL",)
|
|
|
|
|
|
@dataclass(frozen=True)
|
|
class TunnelRelayConfig:
|
|
enabled: bool
|
|
url: str
|
|
connect_timeout_seconds: float
|
|
|
|
@property
|
|
def local_relay_base_url(self) -> str:
|
|
return f"{self.url.rstrip('/')}/api/internal/tunnel/relay"
|
|
|
|
def local_relay_url(self, node_id: str) -> str:
|
|
return f"{self.local_relay_base_url}/{quote(node_id, safe='')}"
|
|
|
|
|
|
_tunnel_relay_config: TunnelRelayConfig | None = None
|
|
|
|
|
|
def _is_docker_runtime() -> bool:
|
|
if os.getenv("DOCKER_CONTAINER", "").strip().lower() == "true":
|
|
return True
|
|
return os.path.exists("/.dockerenv")
|
|
|
|
|
|
def _resolve_docker_tunnel_url() -> str:
|
|
for key in _DOCKER_TUNNEL_URL_ENV_KEYS:
|
|
value = os.getenv(key, "").strip()
|
|
if value:
|
|
return value.rstrip("/")
|
|
return _DEFAULT_DOCKER_TUNNEL_URL
|
|
|
|
|
|
def get_tunnel_relay_config() -> TunnelRelayConfig:
|
|
"""读取 tunnel relay 配置（进程内缓存）。"""
|
|
global _tunnel_relay_config
|
|
if _tunnel_relay_config is not None:
|
|
return _tunnel_relay_config
|
|
|
|
docker_runtime = _is_docker_runtime()
|
|
_tunnel_relay_config = TunnelRelayConfig(
|
|
enabled=docker_runtime,
|
|
url=_resolve_docker_tunnel_url(),
|
|
connect_timeout_seconds=_DOCKER_TUNNEL_CONNECT_TIMEOUT_SECONDS,
|
|
)
|
|
return _tunnel_relay_config
|
|
|
|
|
|
def reset_tunnel_relay_config_cache() -> None:
|
|
"""测试或热更新场景下清理配置缓存。"""
|
|
global _tunnel_relay_config
|
|
_tunnel_relay_config = None
|