feat(fam-edge): 新增 DiskGuard 磁盘空间守护,剩余空间不足自动清理旧素材

背景:Oracle 磁盘曾经被写满(gdrive_videos 持续下载新素材、旧文件迟迟没
清理),触发 rclone 的一个安全机制——同步遇到 IO 错误就整体拒绝执行删除,
形成"越满越删不掉,越删不掉越满"的死循环,最终连新视频都下载不了。这次
排查+手动清理已经解决了当次故障,但需要一道独立于 rclone 同步之外的兜底,
防止再次悄悄写满没人发现。

- oracle_db.py 新增 get_oldest_purgeable_material():只挑最旧的、已完成
  分割阶段(status='done')的整段素材(非 motion_ 前缀),绝不碰运动片段
  (事件时间轴/人物头像依赖它)和还在处理中的素材
- disk_guard.py 新增 DiskGuard 后台线程:5 分钟检查一次,剩余空间 <10GB
  触发清理,删到 15GB 水位为止(留缓冲避免刚清完又立刻触发),复用已有的
  delete_video() 完成实际删除
- app.py 启动这个后台服务;/api/oracle/activity 新增 disk 字段(实时剩余
  空间 + 最近一次清理动作),供服务状态页展示
- 新增 9 个单元测试,全部通过(139/139)

已部署 Oracle 验证:DiskGuard 正常启动,配置生效。

Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
This commit is contained in:
ericwyuan
2026-08-28 12:08:04 +08:00
parent 3ec79de911
commit ef56ae5662
7 changed files with 320 additions and 1 deletions

View File

@@ -0,0 +1,114 @@
"""
DiskGuard - 磁盘空间守护2026-08-28 新增)
背景Oracle 机器磁盘曾经被写满gdrive_videos 持续下载新素材、旧文件迟迟
没被清理),触发了 rclone 的一个安全机制——同步过程中一旦遇到 IO 错误(写
不进去)就整体拒绝执行删除,导致"越满越删不掉,越删不掉越满"的死循环,
最终连新视频都下载不了。这个模块是一道独立于 rclone 同步逻辑之外的兜底:
不管上游同步是否正常,只要本机磁盘剩余空间跌破阈值,就主动清理最旧的、已
经处理完成的整段素材文件,把空间抢回来。
清理对象:仅限"整段素材"gdrive_videos 里落地的原始录像文件filename
不是 motion_ 前缀)且 status='done'(已完成分割阶段,不会再被 Video-Queue
重新捡起——不碰运动片段motion_clips 里的文件,是独立的分析产物,事件
时间轴展示、人物头像裁剪都依赖它,删了会导致图片丢失)、不碰还在
pending/processing 中的素材(避免删掉还没来得及处理的数据)。
"""
import shutil
import threading
import time
import os
from .logger import setup_logger
from .config_loader import load_config
logger = setup_logger('fam-edge.disk_guard')
class DiskGuard:
def __init__(self, db):
self.db = db
cfg = load_config().get('disk_guard', {})
self.enabled = bool(cfg.get('enabled', True))
self.check_interval_sec = int(cfg.get('check_interval_sec', 300))
self.min_free_gb = float(cfg.get('min_free_gb', 10))
# 清理到这个水位就停:留出缓冲,避免刚清理完又立刻因为新文件写入
# 跌破阈值、频繁触发清理循环
self.target_free_gb = float(cfg.get('target_free_gb', 15))
self.watch_path = cfg.get('watch_path', '/opt/fam-edge')
# 单轮最多清理几个文件:防止候选异常多时一次性删太多,先清一部分
# 观察效果下一轮检查再继续check_interval_sec 后很快就会再跑一次)
self.max_delete_per_round = int(cfg.get('max_delete_per_round', 50))
self._running = False
self._thread = None
def start(self):
if not self.enabled:
logger.info("DiskGuard 未启用")
return
self._running = True
self._thread = threading.Thread(target=self._run, daemon=True, name='disk-guard')
self._thread.start()
logger.info(
f"DiskGuard 已启动(阈值 {self.min_free_gb}GB"
f"目标水位 {self.target_free_gb}GB检查间隔 {self.check_interval_sec}s")
def stop(self):
self._running = False
if self._thread:
self._thread.join(timeout=5)
def is_alive(self) -> bool:
return self._thread is not None and self._thread.is_alive()
def _free_gb(self) -> float:
return shutil.disk_usage(self.watch_path).free / (1024 ** 3)
def _run(self):
while self._running:
try:
self.check_once()
except Exception as e:
logger.error(f"DiskGuard 检查异常: {e}", exc_info=True)
for _ in range(self.check_interval_sec):
if not self._running:
return
time.sleep(1)
def check_once(self):
"""检查一次磁盘空间,不足则清理最旧的已完成素材直到恢复到目标水位。
供后台循环调用,也可单独调用做一次性检查(比如手动触发/测试)。
"""
free_gb = self._free_gb()
if free_gb >= self.min_free_gb:
return
logger.warning(f"磁盘剩余 {free_gb:.1f}GB 低于阈值 {self.min_free_gb}GB开始清理旧素材")
self.db.record_activity(
'disk_guard', 'low_space',
f"剩余 {free_gb:.1f}GB 低于阈值 {self.min_free_gb}GB开始清理")
deleted = 0
freed_bytes = 0
while deleted < self.max_delete_per_round and self._free_gb() < self.target_free_gb:
candidate = self.db.get_oldest_purgeable_material()
if not candidate:
logger.warning("磁盘空间仍然紧张,但已经没有可清理的素材了")
self.db.record_activity(
'disk_guard', 'no_candidate',
f"剩余 {self._free_gb():.1f}GB 仍不足,且没有可清理的素材")
break
video_id, local_path = candidate['id'], candidate.get('local_path')
size = os.path.getsize(local_path) if local_path and os.path.isfile(local_path) else 0
self.db.delete_video(video_id)
deleted += 1
freed_bytes += size
logger.info(f"DiskGuard 清理素材 video_id={video_id}(约 {size/1024/1024:.0f}MB")
if deleted:
free_gb = self._free_gb()
self.db.record_activity(
'disk_guard', 'cleaned',
f"清理 {deleted} 个素材,释放约 {freed_bytes/1024**3:.1f}GB"
f"当前剩余 {free_gb:.1f}GB")
logger.info(f"DiskGuard 本轮清理完成:{deleted} 个文件,当前剩余 {free_gb:.1f}GB")