mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-10-14 13:06:51 +00:00
zhangtao
07fa789d2e
本批为 t10 → t13 → t15 三轮实现 + 一轮复核驳回整改的最终形态（pytest 83 passed / ruff clean，
已部署并线上验证）。
异常语义
- CustomException 默认状态码 500 → 400（约 433 处裸调用不再被计为服务端故障），
且「包装底层异常」的 130 处显式恢复 500；get_or_404 → 404；响应体结构不变
代码生成注入面（复核 blocker F1，二次修复）
- column_type 不再做字符串拼接：改为常量白名单派生（类型名 ∈ DB_TO_SQLALCHEMY ∪ {Boolean}，
长度仅取数字/正则捕获），彻底移除 split("(")[1]
- 列类型加形状白名单；字符串/注释/文档字符串用 \uXXXX，标识符过 safe_ident（含 Python/JS 关键字），
.vue 文本与静态属性改 HTML 实体
- 落盘闸门由 ast.parse 升级为 assert_python_structure（语法 + 顶层语句白名单 + 与哨兵骨架逐节点比对）
与 assert_text_structure（.vue/.ts/.toml 行结构）；U+2028/2029 与 C1 控制符纳入转义
安全默认值
- 生产 CORS 不再回落 ["*"]（未配置即拒绝跨域，可选 CORS_STRICT_STARTUP 直接启动失败）
- OAUTH_ALLOWED_HOSTS 不再默认 "*"；新增 ALLOW_LOCALHOST_HOSTS 控制 Host 校验范围
- ALLOWED_HOSTS 收窄为按配置派生；健康检查暴露 code_exec_enabled / cors_origins_configured
定时任务
- SCHEDULER_ALLOW_CODE_EXEC 默认 False；新增 builtin:<模块名>[.<函数名>] 内置处理器路线（不经 exec）
- 判定收敛到 describe_schedulability；列表返回 schedulable / unschedulable_reason；
启动 reconcile 与状态查询对「已启用但不可调度」打 WARN（去重、恢复后清除）
- 加固：内置处理器必须在本模块内定义或经 HANDLERS 声明，拒绝模块内导入的外部对象（复核 R1）
- 顺带修复：_WARNED_UNSCHEDULABLE 的 set.pop TypeError、job.next_run_time 缺失导致的列表 500
测试：新增 tests/test_generator_template_safety.py、test_security_defaults.py、
test_scheduler_schedulability.py（含恶意载荷、闸门红绿、静态护栏）
59 lines
2.1 KiB
Python
59 lines
2.1 KiB
Python
"""健康检查服务：仅探测 DB / Redis 网络连通（SELECT 1 / PING）。"""
|
|
|
|
import asyncio
|
|
from typing import Any
|
|
|
|
from sqlalchemy import text
|
|
|
|
from app.config.setting import settings
|
|
from app.core.database import async_db_session
|
|
from app.core.logger import logger
|
|
|
|
from .schema import ServiceInfoOut
|
|
|
|
|
|
class HealthService:
|
|
"""健康检查服务：探测 DB / Redis 网络连通并组装状态。"""
|
|
|
|
@classmethod
|
|
async def check_database(cls) -> int:
|
|
"""数据库连通：SELECT 1 成功返回 1，失败返回 0。"""
|
|
try:
|
|
async with async_db_session() as session:
|
|
await session.execute(text("SELECT 1"))
|
|
return 1
|
|
except Exception as e:
|
|
logger.warning(f"数据库连通检查失败: {e}")
|
|
return 0
|
|
|
|
@classmethod
|
|
async def check_redis(cls, redis: Any | None) -> int:
|
|
"""Redis 连通：PING 成功返回 1，失败返回 0。"""
|
|
if redis is None:
|
|
return 0
|
|
try:
|
|
await redis.ping()
|
|
return 1
|
|
except Exception as e:
|
|
logger.warning(f"Redis 连通检查失败: {e}")
|
|
return 0
|
|
|
|
@classmethod
|
|
async def collect(cls, redis: Any | None) -> ServiceInfoOut:
|
|
"""并行探活 DB / Redis，组装健康状态。"""
|
|
db_status, redis_status = await asyncio.gather(
|
|
cls.check_database(),
|
|
cls.check_redis(redis),
|
|
)
|
|
return ServiceInfoOut(
|
|
name=settings.TITLE,
|
|
version=settings.VERSION,
|
|
environment=settings.ENVIRONMENT.value,
|
|
db_status=db_status,
|
|
redis_status=redis_status,
|
|
# 升级可见性：这两个开关会静默改变行为（代码块任务不可调度 / 生产拒绝全部跨域），
|
|
# 在健康检查里显式暴露，避免运维只看到"任务不动了"而不知原因。
|
|
code_exec_enabled=settings.SCHEDULER_ALLOW_CODE_EXEC,
|
|
cors_origins_configured=bool(settings.PROD_CORS_ORIGINS) if settings.ENVIRONMENT.value == "prod" else True,
|
|
)
|