redis使用lua脚本实现原子性执行多个命令

1、场景

使用 redis 作为延迟队列时,在 zrangebyscorezrem 之间会存在 context switch 的情况,造成其他的消费者空转
在其他场景下可能会发生更严重的情况
在掘金小册《Redis 深度历险:核心原理与应用实践》第四章讲的比较清楚,最后作者留了一个思考题
使用 Lua Scripting 来优化延时队列的逻辑。
于是,我就用 redis 试了一下

2、编写脚本

local value = redis.call("zrangebyscore", "delay-queue", 0, ARGV[1], "LIMIT", "0", "1")
if not value or not value[1] then
    return nil
end
redis.log(redis.LOG_DEBUG, cjson.encode(value))
redis.call("zrem", "delay-queue", value[1])
return value[1]
  1. 脚本中执行 redis 命令可以使用 call 和 pcall ,区别是 call 遇见错误会抛出,pcall 会返回出来一个带错误的 err

  2. 脚本的返回值会转换成 redis 的回复返回给客户端

  3. 脚本的参数会传到 ARGV table 中

  4. lua 的 index 从 1 开始

3、加载脚本

脚本我写在了文件里面 popzset.lua

redis-cli -x script load < popzset.lua

加载脚本之后会返回一个摘要值

image-20200621194900496

4、执行脚本

evalsha "520a9dd32adf29721b0d925e0fb338aa3c32ea66" {程序生成的时间}

image-20200621195734194

5、参考资料

https://redisbook.readthedocs.io/en/latest/feature/scripting.html

http://redis.io/commands/eval

http://redis.io/commands/evalsha

http://redis.io/commands/script-load

猜你喜欢

转载自www.cnblogs.com/wudanyang/p/13173913.html