mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-21 04:46:26 +00:00
zhangtao
6a5f8cf0dd
这是一次综合性的项目迭代，包含以下核心变更：
1. **目录与模块重构**
- 调整工作流节点类型模块目录结构，迁移节点类型相关代码
- 重命名platform模块为system模块，更新插件配置信息
- 重构代码生成模块导入路径
2. **数据库与CRUD优化**
- 统一所有CRUD类构造函数，新增数据库会话参数
- 修复权限过滤器数据库会话使用问题
- 更新模板生成器的CRUD代码模板
3. **认证与安全改进**
- 重构JWT密钥配置，移除默认密钥强制要求环境变量
- 重命名密码工具类，统一密码加密校验逻辑
- 优化OAuth认证流程，修复匿名认证使用问题
4. **前端与静态资源**
- 重构前端挂载逻辑，增加目录存在性校验
- 使用标准StaticFiles替换自定义前端挂载实现
5. **工具类与依赖更新**
- 修复导入工具的表名重复检测逻辑
- 优化限流回调代码，移除冗余依赖
- 更新用户、租户等模块的响应模型字段
6. **数据与配置修正**
- 修复系统版本数据字段命名不统一问题
- 简化枚举类校验逻辑，移除冗余注释
- 修复测试用例中的密码工具类导入错误
54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
from math import ceil
|
|
|
|
from slowapi import Limiter
|
|
from slowapi.util import get_remote_address
|
|
from starlette.websockets import WebSocket
|
|
|
|
from app.config.setting import settings
|
|
|
|
# 全局限流器 —— slowapi 通过 storage_uri 创建自己的 Redis 连接
|
|
limiter = Limiter(
|
|
key_func=get_remote_address,
|
|
default_limits=["200/10seconds"],
|
|
storage_uri=settings.REDIS_URI,
|
|
)
|
|
|
|
|
|
async def ws_limit_callback(ws: WebSocket, expire: int) -> None:
|
|
"""WebSocket 触发限流时的默认回调：关闭连接。"""
|
|
expires = ceil(expire / 1000)
|
|
await ws.close(
|
|
code=1008,
|
|
reason=f"请求过于频繁，请稍后重试！{expires} 秒后重试",
|
|
)
|
|
|
|
|
|
class WebSocketRateLimiter:
|
|
"""WebSocket 限流依赖（slowapi 不内置支持 WS，自行实现）。"""
|
|
|
|
def __init__(self, max_calls: int = 200, period: int = 10):
|
|
self.max_calls = max_calls
|
|
self.period = period
|
|
|
|
async def __call__(self, ws: WebSocket) -> None:
|
|
import time
|
|
|
|
redis = getattr(ws.app.state, "redis", None)
|
|
if not redis:
|
|
return
|
|
|
|
now = int(time.time())
|
|
window = now // self.period
|
|
key = f"fastapiadmin:request_limiter:ws:{ws.client.host if ws.client else 'unknown'}:{window}"
|
|
|
|
count = await redis.incr(key)
|
|
if count == 1:
|
|
await redis.expire(key, self.period + 1)
|
|
|
|
if count > self.max_calls:
|
|
retry_after = self.period - (now % self.period) + 1
|
|
await ws.close(
|
|
code=1008,
|
|
reason=f"请求过于频繁，请稍后重试！{retry_after} 秒后重试",
|
|
)
|