JOJO
a997c04cda
os.replace 目标是符号链接时会替换链接本身，产生与共享文件脱钩的普通文件 （分叉）。docker/web 模式工作区的 personalization.json 是指向用户级共享 文件的符号链接，保存后分叉被 user_manager 的链接校准丢弃，导致个人空间 设置（如 external_session_header）保存成功却立刻回退。修复为写入前先 resolve 符号链接到真实文件再原子替换；普通文件路径行为不变。
92 lines
3.4 KiB
Python
92 lines
3.4 KiB
Python
"""原子写文件工具：temp file + os.replace 的 Windows 加固版。
|
||
|
||
背景：Windows 上目标文件被任何句柄以默认共享模式打开时（msvcrt 默认共享读/写，
|
||
不含 FILE_SHARE_DELETE），os.replace 会抛 WinError 5（拒绝访问）；
|
||
并发写入者之间则可能抛 WinError 32（共享冲突）。杀毒软件实时扫描、
|
||
Windows Search 索引、同进程并发的读取线程都会造成这类短暂持锁。
|
||
POSIX 下 rename 不受打开句柄影响，无此问题。
|
||
|
||
对策：仅对这两种 winerror 做短退避重试；其他错误立即抛出。
|
||
|
||
2026-08 调整：默认预算由 6 次/约 1.55s 上调至 8 次/约 4.75s——
|
||
实测 Defender 实时扫描 / Search 索引器等外部程序偶发持锁超过原预算
|
||
（如对话索引 index.json 保存失败 WinError 5），提高预算以覆盖此类场景。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import json
|
||
import os
|
||
import tempfile
|
||
import time
|
||
from pathlib import Path
|
||
from typing import Any, Dict, Union
|
||
|
||
# ERROR_ACCESS_DENIED / ERROR_SHARING_VIOLATION
|
||
_RETRY_WINERRORS = {5, 32}
|
||
|
||
PathLike = Union[str, Path]
|
||
|
||
|
||
def replace_with_retry(
|
||
src: PathLike,
|
||
dst: PathLike,
|
||
*,
|
||
attempts: int = 8,
|
||
initial_delay: float = 0.05,
|
||
max_delay: float = 1.6,
|
||
) -> None:
|
||
"""os.replace 的 Windows 加固版：对瞬时持锁做指数退避重试。
|
||
|
||
attempts 为总尝试次数（含首次），退避序列默认 0.05/0.1/0.2/0.4/0.8/1.6/1.6s，
|
||
最坏情况约 4.75s。最后一次失败时原样抛出该 OSError。
|
||
"""
|
||
delay = initial_delay
|
||
total = max(1, int(attempts))
|
||
for i in range(total):
|
||
try:
|
||
os.replace(str(src), str(dst))
|
||
return
|
||
except OSError as exc:
|
||
if getattr(exc, "winerror", None) not in _RETRY_WINERRORS or i == total - 1:
|
||
raise
|
||
time.sleep(delay)
|
||
delay = min(delay * 2, max_delay)
|
||
|
||
|
||
|
||
def atomic_write_json(path: PathLike, data: Dict[str, Any]) -> None:
|
||
"""原子写入 JSON 文件：唯一临时文件 + fsync + replace_with_retry。
|
||
|
||
防止写中断留下半截文件（直接 open(path,"w") 覆写的风险）。
|
||
临时文件与目标同目录，保证 os.replace 同卷原子性；异常时清理临时文件。
|
||
|
||
符号链接写穿（2026-09-10 修复）：目标是符号链接时必须先解析到真实文件再
|
||
原子替换。否则 os.replace 会把链接本身替换成普通文件（分叉），真实文件
|
||
根本没被写入——个人空间 personalization.json 共享链接曾因此被
|
||
user_manager 的链接校准判为分叉并丢弃，表现为「设置保存后立刻回退」。
|
||
"""
|
||
target = Path(path)
|
||
try:
|
||
if target.is_symlink():
|
||
target = target.resolve()
|
||
except Exception:
|
||
pass
|
||
target.parent.mkdir(parents=True, exist_ok=True)
|
||
fd, tmp_name = tempfile.mkstemp(
|
||
prefix=f".{target.name}.", suffix=".tmp", dir=str(target.parent)
|
||
)
|
||
tmp_path = Path(tmp_name)
|
||
try:
|
||
with os.fdopen(fd, "w", encoding="utf-8") as fp:
|
||
json.dump(data, fp, ensure_ascii=False, indent=2)
|
||
fp.flush()
|
||
os.fsync(fp.fileno())
|
||
# Windows 瞬时持锁（并发读取/杀软扫描）重试，POSIX 行为不变
|
||
replace_with_retry(tmp_path, target)
|
||
finally:
|
||
try:
|
||
if tmp_path.exists():
|
||
tmp_path.unlink()
|
||
except Exception:
|
||
pass
|