lua+redis：分布式锁解决方案分享

local redis = require 'resty.redis'
local red = redis:new()
 
-- 连接Redis
red:set_timeout(1000) -- 设置超时时间（毫秒）
local ok, err = red:connect('127.0.0.1', 6379)
if not ok then
    ngx.say("连接Redis失败: ", err)
    return
end
 
-- 获取分布式锁
local lock_key = "my_lock"
local identifier = ngx.worker_pid() -- 使用worker进程PID作为锁的标识
local expire_time = 5 -- 锁的过期时间（秒）
local elapsed, err = red:setnx(lock_key, identifier)
if not elapsed then
    ngx.say("获取分布式锁失败: ", err)
    return
end
 
if elapsed == 1 then
    red:expire(lock_key, expire_time) -- 设置锁的过期时间，避免死锁
    ngx.say("获取分布式锁成功")
else
    ngx.say("已有其他进程持有锁")
end
 
-- 执行业务逻辑...
 
-- 释放分布式锁
local unlock_script = [[
    if redis.call("get", KEYS[1]) == ARGV[1] then
        return redis.call("del", KEYS[1])
    else
        return 0
    end
]]
local unlock_res, err = red:eval(unlock_script, 1, lock_key, identifier)
if not unlock_res then
    ngx.say("释放分布式锁失败: ", err)
    return
end
 
if unlock_res == 1 then
    ngx.say("释放分布式锁成功")
else
    ngx.say("释放分布式锁失败，无法匹配锁的标识")
end

这段代码使用Lua结合OpenResty环境中的resty.redis库来实现分布式锁。首先，它尝试获取一个名为my_lock的锁，如果这个锁不存在，它就设置这个锁，并设置一个过期时间来避免死锁。在完成业务逻辑处理后，它使用一个Lua脚本来安全地释放锁，只有当锁的标识与期望的一致时，才会释放锁。这是一个简单而安全的分布式锁实现方式。