zhaolei
16f134e328
核心功能: - 6 层检测管线: L0 结构分析 → L1 ClamAV → L2 文档深度检测 → L3 关键词匹配 → L4 YARA → L5 PE/二进制 - FastAPI + PostgreSQL + Redis + Docker Compose 架构 - 客户端 HMAC-SHA256 签名鉴权 + 限流 + 幂等性 - 控制台 UI (登录/Dashboard/诊断中心/策略管理/任务列表/测试中心/产品介绍页) - 三层诊断定时器 (error_scan 30s / db_aggregate 30s / container_check 5min) 并发与稳定性修复: - 诊断定时器 Redis 时间槽位锁 (多 worker 同一槽位只有一个执行) - JSONL 存储 fcntl 跨进程排他锁 + 容错解码 + 原子重写 - 移除 SQLite fallback, 统一 PostgreSQL - API_CLIENTS 脱敏为环境变量, .env 已 .gitignore
88 lines
2.8 KiB
Python
88 lines
2.8 KiB
Python
"""统一日志配置。
|
||
|
||
特性：
|
||
- 五级日志：DEBUG / INFO / WARNING / ERROR / CRITICAL，默认 INFO
|
||
- 同时输出到控制台和文件（RotatingFileHandler 按大小轮转）
|
||
- 通过环境变量 ``MESHSAFE_LOG_LEVEL`` 控制级别
|
||
- 通过环境变量 ``MESHSAFE_LOG_DIR`` 控制日志文件目录
|
||
- 日志文件按 10 MB 轮转，保留最近 5 份
|
||
- 自动为 FastAPI / uvicorn / sqlalchemy 等第三方库设置一致的级别
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
import logging.handlers
|
||
import os
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
from app.config import get_settings
|
||
|
||
_LOG_FORMAT = "%(asctime)s [%(levelname)s] %(name)s - %(message)s"
|
||
_DATE_FORMAT = "%Y-%m-%d %H:%M:%S"
|
||
_MAX_BYTES = 10 * 1024 * 1024 # 10 MB
|
||
_BACKUP_COUNT = 5
|
||
|
||
|
||
def setup_logging() -> None:
|
||
"""初始化全局日志配置。幂等：多次调用不会重复添加 handler。"""
|
||
settings = get_settings()
|
||
|
||
level_name = os.environ.get("MESHSAFE_LOG_LEVEL", settings.log_level).upper()
|
||
level = getattr(logging, level_name, logging.INFO)
|
||
|
||
log_dir = os.environ.get("MESHSAFE_LOG_DIR", "/var/log/meshsafe")
|
||
Path(log_dir).mkdir(parents=True, exist_ok=True)
|
||
|
||
root_logger = logging.getLogger()
|
||
root_logger.setLevel(level)
|
||
|
||
# 幂等：已初始化则跳过
|
||
if root_logger.handlers:
|
||
# 更新级别即可
|
||
root_logger.setLevel(level)
|
||
for handler in root_logger.handlers:
|
||
handler.setLevel(level)
|
||
return
|
||
|
||
formatter = logging.Formatter(_LOG_FORMAT, datefmt=_DATE_FORMAT)
|
||
|
||
# 控制台 handler
|
||
console = logging.StreamHandler(sys.stdout)
|
||
console.setLevel(level)
|
||
console.setFormatter(formatter)
|
||
root_logger.addHandler(console)
|
||
|
||
# 文件 handler（按大小轮转）
|
||
log_file = os.path.join(log_dir, "meshsafe.log")
|
||
file_handler = logging.handlers.RotatingFileHandler(
|
||
log_file,
|
||
maxBytes=_MAX_BYTES,
|
||
backupCount=_BACKUP_COUNT,
|
||
encoding="utf-8",
|
||
)
|
||
file_handler.setLevel(level)
|
||
file_handler.setFormatter(formatter)
|
||
root_logger.addHandler(file_handler)
|
||
|
||
# 第三方库级别收敛：避免 uvicorn.access / sqlalchemy 刷屏
|
||
logging.getLogger("uvicorn.access").setLevel(logging.WARNING)
|
||
logging.getLogger("sqlalchemy.engine").setLevel(logging.WARNING)
|
||
logging.getLogger("redis").setLevel(logging.WARNING)
|
||
logging.getLogger("pydantic").setLevel(logging.WARNING)
|
||
|
||
# 启动日志
|
||
root_logger.info(
|
||
"日志系统初始化完成：级别=%s，日志文件=%s（按 %d MB 轮转，保留 %d 份）",
|
||
logging.getLevelName(level),
|
||
log_file,
|
||
_MAX_BYTES // (1024 * 1024),
|
||
_BACKUP_COUNT,
|
||
)
|
||
|
||
|
||
def get_logger(name: str) -> logging.Logger:
|
||
"""获取模块级 logger。"""
|
||
return logging.getLogger(name)
|