63 lines
1.8 KiB
Go
63 lines
1.8 KiB
Go
|
package luascript
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
KEYS[1]:锁的键名（key）,通常是锁的唯一标识。
|
|||
|
|
ARGV[1]:锁的过期时间（lockLeaseTime）,单位为秒。
|
|||
|
|
ARGV[2]:当前客户端的唯一标识（token）,用于区分不同的客户端。
|
|||
|
|
*/
|
|||
|
|
var LockScript = `
|
|||
|
|
-- 锁不存在的情况下加锁
|
|||
|
|
if (redis.call('exists', KEYS[1]) == 0) then
|
|||
|
|
redis.call('hset', KEYS[1], ARGV[2], 1);
|
|||
|
|
redis.call('expire', KEYS[1], ARGV[1]);
|
|||
|
|
return 1;
|
|||
|
|
end;
|
|||
|
|
-- 重入锁逻辑
|
|||
|
|
if (redis.call('hexists', KEYS[1], ARGV[2]) == 1) then
|
|||
|
|
redis.call('hincrby', KEYS[1], ARGV[2], 1);
|
|||
|
|
redis.call('expire', KEYS[1], ARGV[1]);
|
|||
|
|
return 1;
|
|||
|
|
end;
|
|||
|
|
-- 持有锁的 token 不是当前客户端的 token,返回加锁失败
|
|||
|
|
return -9;
|
|||
|
|
`
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
KEYS[1]:锁的键名（key）,通常是锁的唯一标识。
|
|||
|
|
ARGV[1]:锁的过期时间（lockLeaseTime）,单位为秒。
|
|||
|
|
ARGV[2]:当前客户端的唯一标识（token）,用于区分不同的客户端。
|
|||
|
|
*/
|
|||
|
|
var RefreshLockScript = `
|
|||
|
|
if (redis.call('hexists', KEYS[1], ARGV[2]) == 1) then
|
|||
|
|
redis.call('expire', KEYS[1], ARGV[1]);
|
|||
|
|
return 1;
|
|||
|
|
end;
|
|||
|
|
return -8;
|
|||
|
|
`
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
KEYS[1]:锁的键名（key）,通常是锁的唯一标识。
|
|||
|
|
KEYS[2]:锁的释放通知频道（chankey）,用于通知其他客户端锁已释放。
|
|||
|
|
ARGV[1]:解锁消息（unlockMessage）,用于通知其他客户端锁已释放。
|
|||
|
|
ARGV[2]:当前客户端的唯一标识（token）,用于区分不同的客户端。
|
|||
|
|
*/
|
|||
|
|
var UnLockScript = `
|
|||
|
|
if (redis.call('exists', KEYS[1]) == 0) then
|
|||
|
|
redis.call('publish', KEYS[2], ARGV[1]);
|
|||
|
|
return 1;
|
|||
|
|
end;
|
|||
|
|
if (redis.call('hexists', KEYS[1], ARGV[2]) == 0) then
|
|||
|
|
return 1;
|
|||
|
|
end;
|
|||
|
|
local counter = redis.call('hincrby', KEYS[1], ARGV[2], -1);
|
|||
|
|
if (counter > 0) then
|
|||
|
|
return 1;
|
|||
|
|
else
|
|||
|
|
redis.call('del', KEYS[1]);
|
|||
|
|
redis.call('publish', KEYS[2], ARGV[1]);
|
|||
|
|
return 1;
|
|||
|
|
end;
|
|||
|
|
-- 持有锁的 token 不是当前客户端的 token,返回解锁失败
|
|||
|
|
return -10;
|
|||
|
|
`
|