95 lines
2.5 KiB
Python
95 lines
2.5 KiB
Python
|
"""
|
|||
|
|
AI Box Sophon Adapter — 告警记录清理
|
|||
|
|
|
|||
|
|
当告警记录超过配置的 max_records 时，自动清理最早的记录及其关联的视频文件。
|
|||
|
|
"""
|
|||
|
|
import logging
|
|||
|
|
import os
|
|||
|
|
import shutil
|
|||
|
|
|
|||
|
|
from sqlalchemy import select, func, asc
|
|||
|
|
|
|||
|
|
from .models.orm import AlarmRecord
|
|||
|
|
|
|||
|
|
logger = logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
|
|||
|
|
async def trim_alarm_records(db_session, upload_root: str, max_records: int) -> int:
|
|||
|
|
"""
|
|||
|
|
裁剪告警记录，保留最新的 max_records 条
|
|||
|
|
|
|||
|
|
删除超出限制的最早记录，同时清理关联的视频文件目录。
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
db_session: SQLAlchemy 异步 session
|
|||
|
|
upload_root: 上传文件根目录（如 ./data/uploads）
|
|||
|
|
max_records: 最大保留条数，<= 0 表示不限制
|
|||
|
|
|
|||
|
|
Returns:
|
|||
|
|
实际删除的记录数
|
|||
|
|
"""
|
|||
|
|
if max_records <= 0:
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
# 查询当前总记录数
|
|||
|
|
count_result = await db_session.execute(
|
|||
|
|
select(func.count()).select_from(AlarmRecord)
|
|||
|
|
)
|
|||
|
|
total = count_result.scalar() or 0
|
|||
|
|
|
|||
|
|
if total <= max_records:
|
|||
|
|
return 0
|
|||
|
|
|
|||
|
|
# 超出数量
|
|||
|
|
overflow = total - max_records
|
|||
|
|
|
|||
|
|
# 查询最早的 overflow 条记录（按 created_at 升序）
|
|||
|
|
stmt = (
|
|||
|
|
select(AlarmRecord)
|
|||
|
|
.order_by(asc(AlarmRecord.created_at)) # type: ignore
|
|||
|
|
.limit(overflow)
|
|||
|
|
)
|
|||
|
|
result = await db_session.execute(stmt)
|
|||
|
|
old_records = result.scalars().all()
|
|||
|
|
|
|||
|
|
deleted_count = 0
|
|||
|
|
for record in old_records:
|
|||
|
|
# 清理关联的视频文件目录
|
|||
|
|
if record.video_file:
|
|||
|
|
_delete_video_dir(upload_root, record.video_file)
|
|||
|
|
|
|||
|
|
await db_session.delete(record)
|
|||
|
|
deleted_count += 1
|
|||
|
|
|
|||
|
|
await db_session.commit()
|
|||
|
|
|
|||
|
|
if deleted_count > 0:
|
|||
|
|
logger.info(
|
|||
|
|
f"[AlarmTrim] 已清理 {deleted_count} 条告警记录 "
|
|||
|
|
f"（保留最新 {max_records} 条，原共 {total} 条）"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
return deleted_count
|
|||
|
|
|
|||
|
|
|
|||
|
|
def _delete_video_dir(upload_root: str, video_id: str) -> None:
|
|||
|
|
"""
|
|||
|
|
删除告警关联的视频文件目录
|
|||
|
|
|
|||
|
|
视频存储路径: {upload_root}/aibox_sophon_adapter/{video_id}/
|
|||
|
|
|
|||
|
|
Args:
|
|||
|
|
upload_root: 上传根目录
|
|||
|
|
video_id: 视频唯一标识（VideoId）
|
|||
|
|
"""
|
|||
|
|
video_dir = os.path.join(upload_root, "aibox_sophon_adapter", video_id)
|
|||
|
|
|
|||
|
|
if not os.path.isdir(video_dir):
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
shutil.rmtree(video_dir)
|
|||
|
|
logger.debug(f"[AlarmTrim] 已删除视频目录: {video_dir}")
|
|||
|
|
except OSError as e:
|
|||
|
|
logger.warning(f"[AlarmTrim] 删除视频目录失败 [{video_dir}]: {e}")
|