使用 Redis Lua 脚本实现 CAS (check-and-set)?

我正在尝试理解 Redis/Lua 脚本,并且想知道下面的代码是否存在问题。

这是我尝试实现非常简单的“CAS”语义:使用一个键和两个参数来调用它。它将检查该键在服务器上关联的值是否以第一个参数开头,如果是,则将键的新值设置为第二个参数并返回1,否则返回0;如果键与字符串之外的其他类型的数据相关联,则Redis将返回错误,就像在此类键/值组合上尝试SET命令一样。如果在调用之前键不存在,则函数将返回0(失败)。

以下是脚本:

local x=string.len(ARGV[1]);
if redis.call('GETRANGE', KEYS[1], 0, x-1) == ARGV[1] then
    redis.call('SET', KEYS[1], ARGV[2]);
    return 1;
    end;
return 0

这是在一个键为“foo”的情况下使用“bar”前缀值调用脚本的示例(在“redis-cli”中):

eval "local x=string.len(ARGV[1]); if redis.call('GETRANGE', KEYS[1], 0, x-1) == ARGV[1] then redis.call('SET', KEYS[1], ARGV[2]); return 1; end; return 0" 1 foo bar barbazzle

我认为这种用法模式可能出现的情况是,您想同时存储“竞技令牌”和带有键值的值……允许并发客户端尝试更新值,如果它们持有正确的竞技令牌。

与 WATCH/MULTI/EXEC 语义相比,这似乎是一个适合的用法模式吗? (似乎您可以获取当前值,在您的本地代码中拆分竞技令牌,构建一个新值,然后随时尝试使用似乎比WATCH/MULTI/EXEC调用更不令人困惑的语义更新键)。

(我知道我的脚本的语义与 memcached CAS 命令略有不同;这是有意的)。

这经过了我的有限测试……所以我真正询问的是潜在的并发原子性问题,以及Lua中是否存在任何愚蠢之处---因为我几乎从未接触过Lua)。

点赞
用户2629095
用户2629095

根据 Redis 文档,你的原子性操作应该是没有问题的:

Redis 使用相同的 Lua 解释器来运行所有命令。此外,Redis 保证以原子方式执行脚本:在执行脚本时不会执行任何其他脚本或 Redis 命令。这种语义类似于 MULTI/EXEC。从其他所有客户端的角度来看,脚本的效果要么尚未可见,要么已经完成。

但是,如果脚本执行得太慢,就会出现问题。因此,脚本最适合简单的操作,这些操作需要一些逻辑和原子性。

另一个你可能会遇到的漏洞是,如果脚本在中途失败,那么你所做的调用可能无法回滚,尽管脚本将返回错误。

例如:

你有一个这样的脚本:

redis.call('set', 'foo', 1)
redis.call('rpush', 'foo', 2)

脚本执行将返回错误,但 foo 已经被设置为 1


与你的问题无关的另一件事情:我注意到你使用了

eval "your_raw_code" key_count keys argv

实际上,当你在终端中时,你可以调用 lua 脚本文件在 eval 中:

> redis-cli eval "$(cat path/to/script/script_name.lua)" key_count keys argv
2016-04-06 06:33:20