1. 一次线上事故:半截文件与重复支付
大概半年前,我维护的一个内部订单同步服务出了两次看起来毫不相关的事故。第一次,一个下游系统在凌晨拉取对账文件,打开一看,文件最后一行只有半个 JSON——前缀完整,后面的字段全被截断了。第二次,一个用户在极短时间内收到了两条重复的扣款通知,数据库里对应的支付回调被写入了两次。
两个问题单独看都不难查:文件写入时进程被强制终止,没有处理干净;回调接口在超时之后被调用方自动重试,而我们的服务端没有做幂等。但它们放在一起,指向同一个深层问题——这个服务在“数据落盘”和“处理失败”这两件事上,从设计之初就没有认真对待过。当时我把修复方案拆成两部分:存盘要原子,处理要可重试。这就是这篇文章要讲的“原子存盘与重试机制实战”的核心内容。
这篇文章不是教科书式地讲理论。我会用一个真实的“任务状态持久化服务”作为载体,把原子存盘的几种标准做法、重试机制的参数设计、幂等保护的关键要点,以及两者如何配合的全部细节梳理一遍。适合正在写后端服务、中间件、批处理任务,或者负责系统稳定性的人看。如果你现在正被“数据丢了怎么办”“重复执行了怎么办”这类问题困扰,这篇文章应该能给你一套可以直接抄作业的方案。
需要模型API调用? 免费领10W Token,多模型网关一键接入 Claude、DeepSeek 等主流模型。
2. 原子存盘为什么难:问题出在“写了一半”
先明确什么是原子存盘。原子在数据库领域的意思是“要么全部完成,要么什么都不做,不存在中间状态”。放到文件系统或者对象存储的场景里,就是一次完整的数据写入操作,对任何观察者来说,要么看到旧的全部内容,要么看到新的全部内容,绝不会看到新旧混杂、或者只有一半的中间内容。
这个需求听着很朴素,但实现起来有几个容易被忽略的坑。
2.1 直接写目标文件的风险
最朴素的做法是打开目标文件,直接写入全部数据,然后关闭。比如:
python复制with open("/data/order_status.json", "w") as f:
f.write(json.dumps(payload))
这段代码在正常情况下一秒钟能跑无数次,但它有三个致命的非原子点。第一,open(..., "w") 这一步会立刻把原文件截断成 0 字节,如果后面写的过程中进程崩溃,磁盘上留下的就是一个空文件。第二,写操作落盘到文件系统缓存里,真正刷到磁盘需要 fsync,如果没有这条数据,机器掉电时数据可能还停留在页缓存中。第三,写的过程中如果另一个进程来读这个文件,它可能看到的是“截断后的空文件”、“一半数据 + 另一半旧数据”或者“全部新数据”这三种状态之一,具体取决于写入进度。
这不是理论推演。我第一次做文件型状态存储时,就是写“带状态的 JSON 直接覆盖”,结果一次 kill -9 磁盘上留了个 0 字节文件,而我的加载代码刚启动就读到了空数据,直接抛异常。当时还觉得奇怪——“我明明保存了,怎么启动就坏了”。
2.2 原子替换的核心模式:写入临时文件再 rename
解决这个问题的标准姿势是 “同一目录下的临时文件 + 写入完成后的原子重命名”。核心思路是:
- 在同一文件系统内创建一个临时文件,比如
/data/.order_status.json.tmp。 - 把完整数据写入临时文件,调用
flush和fsync确保持久化。 - 关闭文件。
- 调用
os.replace(tmp_path, target_path)完成原子替换。
为什么必须 rename?因为 POSIX 定义了 rename 操作在同一个文件系统内是原子的——它要么把旧路径替换成新路径,要么什么都不变。整个过程对其他进程来说,看到的目标文件永远是“旧的”或“新的”二选一。
示例代码:
python复制import os
import json
import tempfile
def atomic_write_json(target_path: str, data: dict):
dir_path = os.path.dirname(os.path.abspath(target_path))
fd, tmp_path = tempfile.mkstemp(prefix=".tmp_", suffix=".json", dir=dir_path)
try:
with os.fdopen(fd, "w", encoding="utf-8") as f:
json.dump(data, f, ensure_ascii=False, indent=2)
f.flush()
os.fsync(f.fileno())
os.replace(tmp_path, target_path)
except BaseException:
try:
os.unlink(tmp_path)
except OSError:
pass
raise
几个细节值得展开说。
mkstemp创建的临时文件默认权限是 0600,如果这个文件需要被其他进程读取,记得os.chmod调整。- 临时文件必须和目标文件在同一个目录下。如果放在
/tmp,目标文件在/data,rename在不同文件系统之间会退化成“拷贝 + 删除”,那就完全不具备原子性了。 fsync是必须的。只flush只是把数据从用户态缓冲区送到内核页缓存,进程崩溃时数据还在;但机器掉电时内核页缓存会丢。fsync才是真正把数据交给磁盘控制器。对大多数业务场景来说,每次调用fsync的性能损耗是可以接受的,尤其是低频状态保存;但如果你的目标是极高性能,可以做成“周期性批量 fsync”的合并策略,但这超越了本篇文章的范围。
2.3 rename 之外还有好选择:SQLite 与事务
文件替换解决了“单个文件”的原子性问题。但现实中,状态数据很少只有单条——往往是订单和订单明细、任务和任务日志、配置主表和配置历史这种“一对多”结构。如果只用一个 JSON 文件存全部数据,光“加载后加锁修改再整体写回”这一步,在并发高一点的时候就会频繁冲突。
这种场景下,SQLite 是用得最顺手的原子存盘介质。它本质上也是一个文件,但内置了日志和事务机制,BEGIN IMMEDIATE 开始的写事务,如果在提交前崩溃,会自动回滚到上一个完整状态。对“多个数据要同时更新,且不能破坏完整性”的需求来说,SQLite 事务比手撸多文件目录交换要可靠得多,也极少写样板代码。
我自己的实践是:如果数据能整块序列化成一组 JSON 或一个二进制对象,用临时文件 + rename;如果数据有结构性关系、需要按条件查询、还要保证多个字段同时更新,就用 SQLite 或类似嵌入式数据库。不要因为“一个文件多简单”就把所有东西都硬塞进 JSON。
2.4 对象存储、目录级和数据块的原子性
原子存盘的应用场景不只是本地文件。对象存储服务(比如阿里云 OSS、腾讯云 COS、AWS S3)的 PutObject 本身具备“整对象覆盖”的原子性语义,但如果你的流程是“先删除旧对象,再上传新对象”,中间就存在一个空窗期,读请求会拿到 404。正确做法应该是“覆盖写”,让同一个 Key 直接被新内容替换,而不是先删后写。
本地目录替换也有类似思路:用 rename 把整个目录换掉,但前提是挂载点支持目录交换(某些文件系统或某些托管环境并不支持)。所以目录级替换在跨平台、跨环境中常被列为“不推荐用于生产”,我建议优先保持“单数据文件 + 单索引文件”的粒度。
顺带提一句:如果你在用数据库集群或分布式存储,不要假设底层网络文件系统(NFS)的 rename 一定原子。NFS 的语义有时候和本地 POSIX 不一致,遇到这种环境,要多问一句运维同事文件系统是否保证原子性。这个坑我踩过:跨挂载点的 os.replace 在某环境里实际执行的是“copy + unlink”,性能慢不说,还产生了中间态。
3. 重试机制设计:核心难题不是“再试一次”,而是“再试不会坏事”
存盘问题解决之后,第二个核心问题就是“处理失败后怎么办”。最简单的思路是:失败就重试。但重试带来的风险比不重试还大——一个操作的真实执行成功,但因为网络超时或响应丢失被判定为失败,然后重试,这时候如果服务端把同一个指令再执行一遍,就会产生重复数据、重复扣款、重复发消息。
所以,重试机制设计的核心,不是“怎么再调用一次更稳”,而是 “再调用一次也不会改变系统状态的保护设计”。这个保护包含两个方向:叫幂等,以及配合重试的依赖控制。
3.1 三类常见的重试困境
需要重试的场景基本可以分为三类。
第一类是“结果未知”。请求已经发出,但响应超时。网络包可能已经送达并处理,也可能根本没送达。这是最危险的重试场景,必须靠幂等键。
第二类是“明确失败”。服务端返回 5xx 或明确的业务错误码。这种情况下,你至少知道上一次没有执行成功,重试是安全的(前提是服务端没有做“部分写入后报错”这种半成品行为)。
第三类是“前置依赖失败”。当前任务依赖的上游数据还没准备好,比如任务要处理今天的分片,但分片尚未生成。这种重试的意义在于等待条件满足,而不是修复自身错误,往往需要延迟较长的时间。
3.2 幂等键:重试安全的最强护盾
一个可重试接口,在接受请求时应该生成一个全局唯一的幂等 key,通常由请求方生成。服务端在处理请求时,先把幂等 key 存入一个带“唯一约束”的存储(数据库唯一索引、Redis SETNX、文件系统中创建一个专属于该 key 的目录,都可以),然后开始执行逻辑;如果这个 key 已经存在,说明之前接受过同样请求,不再重复处理,直接返回上一次的结果。
举个例子,一个处理“订单状态变更回调”的服务:
python复制from flask import Flask, request, jsonify
import uuid
import redis
app = Flask(__name__)
r = redis.Redis(host="localhost", port=6379, db=0)
IDEMPOTENT_PREFIX = "idem:order_status"
IDEMPOTENT_TTL = 60 * 60 * 24
@app.post("/api/v1/order/report")
def order_report():
payload = request.get_json(force=True)
# 幂等键由调用方生成,并在每次重试时保持不变
idem_key = payload["idempotent_key"]
order_id = payload["order_id"]
target_key = f"{IDEMPOTENT_PREFIX}:{order_id}:{idem_key}"
ok = r.set(target_key, "processing", nx=True, ex=IDEMPOTENT_TTL)
if not ok:
# 如果状态是 DONE,说明已经成功处理过,直接返回成功
cached = r.get(target_key)
if cached == b"DONE":
return jsonify({"code": 0, "msg": "already processed", "data": {}})
if cached == b"PROCESSING":
# 另一个并发请求正在处理中,可以返回冲突或让调用方稍后重试
return jsonify({"code": 409, "msg": "another request in progress"}), 409
if cached == b"FAILED":
# 上次失败但未完成,可以安全删除后重试
pass
try:
# 真正执行业务逻辑
result = apply_order_status_change(order_id, payload["new_status"])
r.set(target_key, "DONE", ex=IDEMPOTENT_TTL)
return jsonify({"code": 0, "data": result})
except Exception as e:
# 标记失败,但不留下脏数据
r.set(target_key, "FAILED", ex=IDEMPOTENT_TTL)
raise e
这个方案的关键点是:
- 幂等键不能是“订单 ID + 时间戳”这种可能变的东西,必须是同一次业务动作的唯一标识,否则重试时 key 不一样,幂等就失效了。
- key 的存储必须带 TTL,否则幂等记录会越堆越多,长期占空间。
- “处理中”和“已完成”最好分开表示,避免并发请求互相误判。如果处理极快,可以只使用 DONE 状态,把“处理中”完全省略,但并发高的场景里“PROCESSING”状态能有效防止同一请求被两个 worker 同时执行。
3.3 指数退避 + 抖动:别让重试变成二次故障
幂等是“重复执行会不会坏事”的问题,退避是“什么时候重试最合理”的问题。大多数人刚写重试时,都会用固定间隔,比如每 5 秒重试一次,总共重试 3 次。这样做在单个客户端时问题不大,但如果系统有 500 个客户端同时遇到上游故障,所有客户端会同时重试,形成一个“重试风暴”,上游还没恢复就已经被第二次冲击打垮了。
标准解法是指数退避加抖动(Exponential Backoff with Jitter)。公式:
code复制sleep_time = min(max_delay, base_delay * (2 ** retry_count)) + random.uniform(0, jitter_amount)
实现一个带抖动的重试器:
python复制import random
import time
from typing import Callable, TypeVar
T = TypeVar("T")
def retry_with_backoff(
func: Callable[[], T],
max_retries: int = 3,
base_delay: float = 0.5,
max_delay: float = 10.0,
jitter: float = 0.3,
) -> T:
last_exc = None
for attempt in range(max_retries + 1):
try:
return func()
except Exception as e:
last_exc = e
if attempt >= max_retries:
break
delay = min(max_delay, base_delay * (2 ** attempt))
delay += random.uniform(0, jitter)
time.sleep(delay)
raise last_exc
为什么要加 jitter?因为指数退避的所有重试者都会在同一时间点醒来,加上随机抖动后,把并发重试的节奏打散,能显著降低“打在同一时刻”的概率。Google 的 SRE 书中也明确提到了这一点,并且在实践中我确实见过没加抖动的系统在故障恢复后收到一波瞬间高峰流量,而加了抖动之后系统平稳回满。
另外要注意:重试不应该吞掉业务异常。只有对“当前操作可能已经执行成功但响应丢失”的未知状态、或者“上游暂时不可用”的超时和 5xx 才是可重试的。像参数错误(400)、签名验证失败(401/403)、资源不存在(404),重试一万次结果也一样,应直接返回给调用方,别浪费 CPU。
3.4 超时设置与连接池大小联动
重试的节奏还要和超时、连接池配合。如果单次 HTTP 请求超时是 3 秒,最多重试 5 次,那么最坏情况下一次操作需要 3 * 6 = 18 秒(含首次)。如果这个操作是在用户请求链路内,用户早就等急了。所以重试策略必须分层:对外部服务的调用,单个请求超时不宜超过必要时间;对用户请求链路的同步调用,尽量不重试或者只重试一次,把多次重试放到异步任务里去做。
如果用了连接池,还要考虑一个隐藏问题:重试的打流会占用连接池线程。假设连接池有 20 个连接,一次上游故障后,如果所有任务都在重试且每个任务都在等同一个上游返回超时,连接池会瞬间被塞满,其他正常请求也被阻塞。所以连接池大小和超时时间要互相折算——最大等待连接数 × 单次请求超时 ≈ 最多能承受的在途请求总时间,一旦超出,系统会真正卡死。
4. 把原子存盘和重试机制装进同一个组件
现在,把两套能力组合起来,做一个“任务状态持久化 + 失败重跑”的统一组件。这个组件解决的实际问题是:一个批量任务处理一批数据,任务执行过程中可能崩溃,崩溃后要从上次完整状态继续,而不是从零开始。
4.1 需求拆解
我需要一个类似这样的持久化组件:
- 任务被分解成多个“步骤”(比如处理文件 A、处理文件 B、上传到远端)。
- 每个步骤执行前先记录状态,执行完成后再次记录状态。
- 任务随时可能崩溃,重启后读取磁盘上的状态,跳过已经完成的步骤。
- 记录状态这个动作本身不能破坏文件,不能出现“半截状态文件”。
- 如果某一步失败,服务可以重试;重试不能导致重复执行已完成步骤。
我把状态保存为 JSON,内容如:
json复制{
"task_id": "task_20240613_001",
"version": 1,
"steps": {
"fetch_file": {"status": "done", "updated_at": "2024-06-13T10:00:02Z"},
"parse_data": {"status": "running", "started_at": "2024-06-13T10:00:03Z"},
"upload_result": {"status": "pending"}
}
}
使用临时文件 + rename 来实现状态文件的原子替换。任务重启后加载状态文件,找到 status: "running" 且已经超过某个超时阈值(比如 20 秒)的步骤,认为上一次执行异常中断,将该步骤重新标记为 pending,继续执行。
4.2 状态管理器的具体实现
python复制import os
import json
import time
import tempfile
class AtomicStateManager:
def __init__(self, state_path: str):
self.state_path = state_path
def _read_state_locked(self) -> dict:
try:
with open(self.state_path, "r", encoding="utf-8") as f:
return json.load(f)
except FileNotFoundError:
return {"steps": {}}
def _write_state_atomic(self, state: dict):
dir_path = os.path.dirname(os.path.abspath(self.state_path))
fd, tmp_path = tempfile.mkstemp(prefix=".state_", suffix=".tmp", dir=dir_path)
try:
with os.fdopen(fd, "w", encoding="utf-8") as f:
json.dump(state, f, ensure_ascii=False, indent=2)
f.flush()
os.fsync(f.fileno())
os.replace(tmp_path, self.state_path)
except BaseException:
try:
os.unlink(tmp_path)
except OSError:
pass
raise
def mark_step_running(self, step_name: str):
state = self._read_state_locked()
state["steps"][step_name] = {
"status": "running",
"started_at": time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime()),
}
self._write_state_atomic(state)
def mark_step_done(self, step_name: str, result_meta: dict = None):
state = self._read_state_locked()
state["steps"][step_name] = {
"status": "done",
"updated_at": time.strftime("%Y-%m-%dT%H:%M:%SZ", time.gmtime()),
"result": result_meta or {},
}
self._write_state_atomic(state)
def recover_interrupted_steps(self, timeout_secs: int = 20):
state = self._read_state_locked()
now = int(time.time())
for step_name, info in state["steps"].items():
if info["status"] != "running":
continue
started_at = info.get("started_at", "")
try:
started_ts = int(time.mktime(time.strptime(started_at, "%Y-%m-%dT%H:%M:%SZ")))
except ValueError:
state["steps"][step_name]["status"] = "pending"
continue
if now - started_ts > timeout_secs:
state["steps"][step_name]["status"] = "pending"
self._write_state_atomic(state)
这段代码的原则就是:状态以文件为真,所有状态变更都通过原子替换生效。任务进程挂在 mark_step_running 之后、mark_step_done 之前时,磁盘上保留的是 running 状态,重启后会被 recover_interrupted_steps 捞起重跑。如果挂在 mark_step_done 之后,则说明程序在保存状态之前已经完成了所有副作用操作,重启后会看到 done 状态,不会重复执行这一步。
要注意的是,这套机制只对“标记状态”本身是原子的,它不能保证“业务副作用”也是原子的。如果步骤“发送通知”已经发出去了,但紧接着标记 done 前进程崩了,重启后会重发一次通知。这句话我放在这里纯粹是想让读者意识到边界——真正需要完全避免重复的业务副作用,靠的是业务侧幂等,而不是状态机的标记。
4.3 重试和状态机的衔接
状态机在恢复时,会做一个“合法的重试路径选择”。假如某个步骤连续失败 3 次,我们不能让它无限重试,需要记录失败次数:
json复制{
"status": "failed",
"retry_count": 2,
"last_error": "connection timeout after 5000ms",
"next_retry_at": "2024-06-13T10:05:00Z"
}
加一个判断:当 retry_count >= max_retries 时,不再执行该步骤,进入 failed_stop 状态,需要人工介入。这样,重试逻辑就被约束在组件内部,不会出现“任务元凶是下游系统、但重试风暴把内部数据库也打崩”的自杀式行为。
5. 分布式并发下的原子存盘细节
现在把视角再往外拉一层。上面的文件级原子替换在单机、单进程下很舒服,但真实生产里经常是多个进程甚至多台机器同时读写同一份状态。这时候,原子替换只能保证文件内容一致性,不能保证并发更新的顺序正确性。两个进程同时读到旧状态,各自修改自己关心的字段,然后分别做 rename,后写的会把先写的覆盖掉,造成更新丢失。
5.1 文件锁与专属目录
最常见的解法是进程间加锁。以 Linux 为例,使用 fcntl.flock 对状态文件上共享锁/排他锁。
python复制import fcntl
with open(self.state_path, "r+", encoding="utf-8") as f:
fcntl.flock(f.fileno(), fcntl.LOCK_EX)
state = json.load(f)
# modify state
f.seek(0)
f.truncate()
json.dump(state, f)
f.flush()
os.fsync(f.fileno())
fcntl.flock(f.fileno(), fcntl.LOCK_UN)
但注意,直接对原文件做“seek + truncate + write”并不是原子操作,虽然它不产生“半截文件”以外的中间态?其实还是会——如果写完过程中崩溃,文件就是半截。更好的做法是“锁定 + 临时文件 + rename”:
- 打开锁文件,对锁文件上
LOCK_EX。 - 读取原状态文件。
- 修改后写入临时文件,
fsync。 os.replace(tmp, target_path)。- 释放锁。
这样做的好处是,读状态的人即使没拿锁,也只会看到新或旧完整文件;写状态的人通过锁串行化,保证不会互相覆盖。锁文件和状态文件可以分开,锁名可以固定为 <state_path>.lock。
5.2 原子存盘与编辑器的对比:为什么不要人为干预
写文件时我还常被别人问到一个问题:“我能不能手动用 VI 改这个 JSON 状态文件?”答案是:可以,但如果你希望状态文件始终被程序原子管理,最好别在运行期间手动编辑。很多编辑器保存时是先写入缓冲、再保存到临时文件、最终 rename,这种方式本身是原子的;但也有编辑器默认会修改文件权限,或者在你保存时产生 vim 的 swap 文件,这些杂糅都可能让程序读状态时遇到预期外的内容格式。如果确实需要人工调整,强烈建议先暂停任务,改完再启动,不要在热运行中直接编辑。
5.3 无锁化思路:单一写者 + 版本号
如果架构允许,最优雅的方案其实是“单一写者”。也就是说,任何时候只允许一个进程或一个线程修改状态文件,其他进程只读。这样避免了大范围加锁的开销。在分布式场景里,可以用一个领导者选举(比如 ZooKeeper / etcd 选主),只有 leader 写本地状态,follower 只读。
如果确实需要多写者且无法引入锁,可以给状态文件加 version 字段,每次更新时比较版本号,遇到版本冲突就回滚重新加载最新状态再合并。本质上是一种乐观并发控制。这个方案实现起来比锁复杂,而且要求每个写者都遵守版本检查逻辑,否则毫无意义。
6. 数据库作为原子存盘介质时,和重试怎么配合
很多场景下,状态数据最终还是要落到数据库。这里边有一个常见的认知偏差:数据库事务保证原子性,却不保证你的重试逻辑“只执行一次”。事务帮你解决的是“数据操作要么全成要么全没”,但这只是重试安全的一部分。
6.1 事务内的唯一约束
在数据库里的防重试手段,最有效的就是唯一索引。给业务表加一个 request_id 字段,并建立唯一索引。插入时如果撞到唯一冲突,说明这条请求已经处理过(或正在处理),程序捕获异常并走“幂等命中”分支,而不是报错返回。
举个例子:
sql复制CREATE TABLE order_status_change (
id BIGINT AUTO_INCREMENT PRIMARY KEY,
order_id VARCHAR(64) NOT NULL,
request_id VARCHAR(64) NOT NULL,
new_status VARCHAR(32) NOT NULL,
created_at DATETIME NOT NULL DEFAULT CURRENT_TIMESTAMP,
UNIQUE KEY uk_order_request (order_id, request_id)
);
重试时,请求方携带同样的 request_id,数据库的 UNIQUE KEY 会拒绝重复插入,这是最牢靠的“先到先得”判重手段。比“先查有没有再插入”的写法安全得多——后者在并发下可能两个请求都查到“没有”,然后双双插入成功,造成重复数据。
6.2 事务里的状态流转
处理任务状态时,推荐在事务里完成“状态前置检查 + 状态更新 + 业务数据变更”这个组合。
sql复制BEGIN;
SELECT status FROM task WHERE task_id = 'T1001' FOR UPDATE;
-- 应用层判断 status 是否允许流转,比如必须从 RUNNING 变为 DONE
UPDATE task SET status = 'DONE', updated_at = NOW() WHERE task_id = 'T1001';
INSERT INTO task_log(task_id, action, created_at) VALUES ('T1001', 'complete', NOW());
COMMIT;
FOR UPDATE 锁行后,其他并发事务在更新同一行时会等待,保证同一个任务不会同时被两个 worker 执行。如果没有这一行锁,两个 worker 同时读到 RUNNING,同时更新为 DONE,从业务结果看可能没问题,但从日志、操作记录看就会缺一条或乱序。
6.3 事务内的重试与“已处理”状态
另一个容易踩的坑是:同一业务操作,既改了业务表,又改了状态表,但两个更新必须在同一个事务里,否则会不一致。比如你改了订单状态为“已支付”,然后向用户发送通知,结果通知发送失败。如果发送动作和订单状态更新在同一个事务里,事务最终回滚时,订单状态没改,发送也失败;如果不在同一事务里,就可能出现“订单显示已支付但通知没发”的不一致状态,只能靠补偿任务对账。
我的习惯是:把“核心状态变更”放在事务内,把“通知、推送、外呼”这类非核心副作用放在事务外,并用消息队列异步处理。事务内只保证核心数据的一致性,事务外靠重试和死信队列兜底。这样数据库不会被外部系统拖垮,重试也只在消息层发生,不会污染核心链路。
7. 实战案例:推演一次“崩溃-重启-重试”的完整路径
用上面的组件,走一遍完整链路,看看崩溃和重启后会发生什么。
场景:一个文件处理任务,处理 100 个文件,每个文件处理完要更新状态。
初始状态:
code复制task_id: T1001
steps:
file_001: pending
file_002: pending
...
file_100: pending
执行流程:
- 加载任务状态,发现所有 file 均为 pending,开始处理。
mark_step_running("file_001"),状态变为 running。- 处理 file_001,成功。
mark_step_done("file_001"),状态变为 done。- 处理 file_002,
mark_step_running("file_002"),正在处理时进程收到SIGKILL。
进程重启后:
- 加载状态文件,看到 file_001 是 done,file_002 是 running。
- 调用
recover_interrupted_steps(timeout_secs=20),发现 file_002 的started_at距离现在已经超过 20 秒,认为上次执行中断,把 file_002 改为 pending。 - 主循环跳过 file_001,从 file_002 开始重新处理。
这里隐藏着一个重要问题:file_002 在处理过程中可能已经做了一部分副作用操作。比如它已经写完了输出文件的一部分,然后崩溃,重启后重新处理整个 file_002,会造成“该文件的输出内容被重复写入”。所以我在设计每个文件的处理函数时,会让“对该文件的处理”在业务上具备幂等性——比如打开输出文件时以 truncate 模式写入,而不是 append 模式。这样重跑时,以前的内容被整体覆盖,不会越积越多。
如果你处理的不是文件而是外部接口,就回到第 3 节说的幂等键设计:每个文件对应一个固定幂等键,重试时传同一键,服务端才能去重。
8. 原子存盘的性能权衡与适用边界
原子存盘虽然可靠,但也不是免费的。一次 fsync 大约耗时 2~10 毫秒不等(取决于磁盘和机器负载),如果每次状态变更都立刻 fsync,高频状态更新会变成磁盘瓶颈。所以要做分级:
- 低频、关键、流失不可接受的状态(比如任务步骤,几十秒才变一次):每次写都 fsync,完全能接受。
- 高频、可容忍少量丢失的统计信息(比如每秒 PV 计数):可以使用内存聚合 + 周期性刷盘,不需要每次 fsync。
- 超高频的日志型数据:直接用日志或消息队列落盘,不要用“JSON 覆盖全文件”这种方案。
另外一个容易忽略的事实是:rename 的原子性只保证单次操作的数据一致性,不保证“原子性 + 可持久性”同时成立。如果你写完临时文件没有 fsync 就 rename,掉电时 rename 可能成功但新文件内容还没刷到磁盘,恢复后可能拿到的是旧的、或者是空文件。完整链路必须包含“临时文件 fsync → rename → 目录 fsync(可选)”。目录 fsync 是为了保证 rename 操作本身可持久化,在极端掉电下可以防止目录项丢失。多数常规场景只做文件 fsync 就够,但如果做的是数据库 WAL 这类对持久性要求极高的系统,目录 fsync 也不能省。
9. 一套可复用的设计清单
最后把我自己实践里总结出来的要点按步骤梳理出来,可以直接当检查清单用。
- 决定原子存盘的粒度:单文件 JSON 还是 SQLite 还是数据库表。数据量大、带关联关系用 SQLite 或 DB;单块数据用文件。
- 所有写文件操作统一走“临时文件 + fsync + rename”模式,禁止在目标路径上直接写。
- 临时文件必须和目标文件同目录,跨文件系统 rename 一律视为“非原子”。
- 高频状态更新场景,接受“周期性批量刷盘”的折衷,但要有明确的数据丢失容忍度。
- 所有可重试的接口,强制要求调用方传入幂等键;服务端用唯一索引或 Redis
SETNX做幂等记录。 - 重试策略采用指数退避 + 随机抖动;区分可重试错误(超时、5xx、网络异常)和不可重试错误(参数错误、鉴权失败)。
- 设置最大重试次数和死信逻辑,失败达到阈值要告警,不要无限循环。
- 恢复机制必须等待“超过超时阈值”的 running 状态,不能启动后立刻把所有 running 都重跑,因为还有可能是另一个 worker 正在正常处理。
- 在事务里用行锁或乐观锁保护状态流转,避免多 worker 并发重复处理同一个任务。
- 核心业务副作用要设计成幂等,状态标记不能替代业务幂等。
- 监控指标至少要有:原子写失败次数、重试次数分布、幂等命中次数、恢复任务次数。这些指标能帮你判断系统是否在良性运行。
10. 写在最后的几个实战心得
这套方案我前前后后改了三版。第一版只做了原子写文件,没做幂等重试,结果一次网络抖动把同一任务重复跑了两次,问题还是没根治。第二版加了重试和幂等,但没处理好“重试风暴”,上游故障恢复时我们的服务被自己的重试流量拖死。第三版才是现在的形态:文件写入统一原子替换,重试统一走指数退避 + 抖动,上游故障时只发出必要的最小重试流量。
我个人在实际操作中的一个体会是:重试机制的设计,本质上是给系统增加了一个“不确定但可容忍的延迟”维度。你不可能完全消灭失败,也不能完全保证不重复,但你可以让“失败后安全恢复”和“重复后不造成影响”成为系统的默认能力。只要这两个底座稳了,上层业务逻辑怎么加都踏实。
如果这个小项目能重来一次,我会把测试部分做得更细。在状态管理器的单元测试里,故意模拟“写完临时文件但没 rename 就崩溃”“rename 完成后但没 fsync 就掉电”的情况,用来验证恢复逻辑是否真的能覆盖每一种崩溃点位。现实世界中的崩溃不挑时间,你要是没测过这些边界,它就一定会在某个凌晨替你测一遍。
