mirror of
https://github.com/fawney19/Aether.git
synced 2026-09-03 09:50:21 +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)
88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
"""
|
||
速率限制配置
|
||
|
||
提供灵活的端点速率限制策略配置
|
||
"""
|
||
|
||
from dataclasses import dataclass
|
||
from typing import Literal
|
||
|
||
RateLimitScope = Literal["server_ip", "user", "api_key", "skip"]
|
||
|
||
|
||
@dataclass
|
||
class RateLimitPolicy:
|
||
"""速率限制策略"""
|
||
|
||
scope: RateLimitScope # 限制范围
|
||
limit: int # 限制值（请求数/分钟）
|
||
description: str = ""
|
||
|
||
|
||
class RateLimitConfig:
|
||
"""
|
||
速率限制配置管理
|
||
|
||
定义不同路径前缀的速率限制策略
|
||
"""
|
||
|
||
# 默认策略配置
|
||
POLICIES: dict[str, RateLimitPolicy] = {
|
||
# 客户端 API 端点 - 服务器级别 IP 限制
|
||
"/v1/": RateLimitPolicy(
|
||
scope="server_ip", limit=60, description="Claude/OpenAI API 端点，服务器级别限制"
|
||
),
|
||
# 公共 API 端点 - 服务器级别 IP 限制
|
||
"/api/public/": RateLimitPolicy(
|
||
scope="server_ip", limit=60, description="公共只读 API，服务器级别限制"
|
||
),
|
||
# 管理后台端点 - 用户级别限制
|
||
"/api/admin/": RateLimitPolicy(
|
||
scope="user", limit=1000, description="管理后台，用户级别限制"
|
||
),
|
||
# 认证端点 - 跳过中间件（在路由层处理）
|
||
"/api/auth/": RateLimitPolicy(scope="skip", limit=0, description="认证端点，路由层处理"),
|
||
# 用户端点 - 用户级别限制
|
||
"/api/users/": RateLimitPolicy(scope="skip", limit=0, description="用户端点，路由层处理"),
|
||
# 监控端点 - 跳过限制
|
||
"/api/monitoring/": RateLimitPolicy(scope="skip", limit=0, description="监控端点"),
|
||
}
|
||
|
||
@classmethod
|
||
def get_policy_for_path(cls, path: str) -> RateLimitPolicy | None:
|
||
"""
|
||
根据路径获取速率限制策略
|
||
|
||
按照最长匹配原则，优先匹配更具体的路径
|
||
|
||
Args:
|
||
path: 请求路径
|
||
|
||
Returns:
|
||
匹配的速率限制策略，如果没有匹配则返回 None
|
||
"""
|
||
# 按路径长度降序排序，确保最长匹配优先
|
||
sorted_prefixes = sorted(cls.POLICIES.keys(), key=len, reverse=True)
|
||
|
||
for prefix in sorted_prefixes:
|
||
if path.startswith(prefix):
|
||
return cls.POLICIES[prefix]
|
||
|
||
return None
|
||
|
||
@classmethod
|
||
def register_policy(cls, prefix: str, policy: RateLimitPolicy) -> None:
|
||
"""
|
||
注册新的速率限制策略
|
||
|
||
Args:
|
||
prefix: 路径前缀
|
||
policy: 速率限制策略
|
||
"""
|
||
cls.POLICIES[prefix] = policy
|
||
|
||
@classmethod
|
||
def get_all_policies(cls) -> dict[str, RateLimitPolicy]:
|
||
"""获取所有策略配置"""
|
||
return cls.POLICIES.copy()
|