admin
d703ab64e3
CI / docs-check (push) Canceled after 0s
CI / backend-test (push) Canceled after 0s
CI / service-test (push) Canceled after 0s
CI / frontend-test (push) Canceled after 0s
CI / rust-core (push) Canceled after 0s
CI / docs-check (pull_request) Canceled after 0s
CI / backend-test (pull_request) Canceled after 0s
CI / service-test (pull_request) Canceled after 0s
CI / frontend-test (pull_request) Canceled after 0s
CI / rust-core (pull_request) Canceled after 0s
27 lines
1.3 KiB
Python
27 lines
1.3 KiB
Python
"""仅删除过期的上传暂存；引用的历史对象永远不会是 GC'd。"""
|
|
import re
|
|
import time
|
|
from pathlib import Path
|
|
from .database import row, rows, run
|
|
|
|
|
|
def cleanup_expired_uploads(db, staging: Path, *, now=None, limit=500):
|
|
now = int(time.time() if now is None else now)
|
|
with db.transaction() as conn:
|
|
expired = rows(conn, "SELECT id,vault_id FROM uploads WHERE expires<=:now ORDER BY expires LIMIT :limit", now=now, limit=limit)
|
|
removed = 0
|
|
for candidate in expired:
|
|
# 与 PUT 相同的锁定顺序/完成。获取锁后重新检查过期时间。
|
|
with db.transaction() as conn:
|
|
suffix = " FOR UPDATE" if not db.sqlite else ""
|
|
row(conn, "SELECT id FROM vaults WHERE id=:v" + suffix, v=candidate["vault_id"])
|
|
upload = row(conn, "SELECT id FROM uploads WHERE id=:id AND expires<=:now", id=candidate["id"], now=now)
|
|
if upload:
|
|
if not re.fullmatch(r"[0-9a-f]{32}", upload["id"]):
|
|
raise RuntimeError("UPLOAD_ID_INVALID")
|
|
# 文件优先：中断留下可以重试的过期行。
|
|
(staging / upload["id"]).unlink(missing_ok=True)
|
|
run(conn, "DELETE FROM uploads WHERE id=:id", id=upload["id"])
|
|
removed += 1
|
|
return {"expired_uploads_removed": removed}
|