Redis：原子GET与条件SET操作实现方案咨询

针对你要实现的单进程独占磁盘写入的场景，有两种靠谱的方案，我给你详细拆解下：

方案一：WATCH + 事务（MULTI/EXEC） #

Redis的WATCH命令可以帮你实现"乐观锁"逻辑，配合事务就能完成条件更新的原子操作，步骤如下：

• 用WATCH监听目标键（比如disk_writing_flag），如果这个键在事务执行前被其他客户端修改，事务会直接失败
• 获取当前键的值，判断是否等于你的预期值（比如预期是0，代表无进程在写入）
• 如果值匹配，开启事务（MULTI），执行SET操作修改键值
• 提交事务（EXEC），如果事务执行成功，说明你成功获取了写入权限；如果EXEC返回空/nil，说明其他进程抢先修改了键，你需要重试整个流程

举个实际的命令序列（假设用Redis-cli操作）：

# 第一步：监听目标键
WATCH disk_writing_flag

# 第二步：获取当前值，客户端自行判断是否等于预期（比如0）
GET disk_writing_flag

# 如果值匹配，执行后续事务
MULTI
SET disk_writing_flag 1  # 把标识设为1，表示正在写入
EXEC

这种方式需要你的客户端代码处理重试逻辑——一旦事务失败，就重新走一遍流程。

方案二：Lua脚本（更推荐！） #

Redis会把整个Lua脚本作为单个原子操作执行，脚本里可以直接写条件判断逻辑，不需要依赖客户端的额外处理，更简洁可靠。

你可以写一个Lua脚本，直接在Redis内部完成"取值-判断-更新"的全流程：

-- 脚本逻辑：如果当前键值等于预期值，就更新为新值，返回1；否则返回0
local current_val = redis.call('GET', KEYS[1])
if current_val == ARGV[1] then
    redis.call('SET', KEYS[1], ARGV[2])
    -- 可选：给键加过期时间，防止进程崩溃导致锁永久持有
    -- redis.call('EXPIRE', KEYS[1], 30)
    return 1
else
    return 0
end

然后用EVAL命令执行这个脚本：

EVAL "local current_val = redis.call('GET', KEYS[1]) if current_val == ARGV[1] then redis.call('SET', KEYS[1], ARGV[2]) return 1 else return 0 end" 1 disk_writing_flag 0 1

参数解释：

• 1：表示后面有1个键参数（KEYS[1]）
• disk_writing_flag：目标键名
• 0：预期的匹配值
• 1：要设置的新值

脚本返回1代表你成功获取了写入权限，返回0代表已经有进程在执行写入操作了。

额外建议 #

为了避免进程崩溃导致锁一直被占用，建议在SET的时候加上过期时间，比如把脚本里的SET改成：

redis.call('SET', KEYS[1], ARGV[2], 'EX', 30)  # 30秒后自动释放锁

当然，进程完成写入后，也可以主动执行DEL disk_writing_flag来释放锁。

内容的提问来源于stack exchange，提问作者duhaime