当前位置: 首页 > ai >正文

redis+lua脚本


一、Lua脚本操作Redis的优势

特性说明
原子性Lua脚本在Redis中单线程执行,所有操作要么全部成功,要么全部失败。
减少网络开销将多个Redis命令合并为一个脚本执行,减少客户端与服务端的通信次数
复杂逻辑封装可实现条件判断、循环、计算等复杂逻辑(Redis原生命令无法直接实现)。
集群兼容性通过显式声明KEYS,保证在Redis集群模式下正确路由到目标节点。

二、Lua脚本基本规范

1. 参数传递
  • KEYS数组:所有操作的Redis键(Key)必须显式声明,通过KEYS[1], KEYS[2]访问。
  • ARGV数组:其他参数(如值、标志位)通过ARGV[1], ARGV[2]访问。
  • 示例
    -- KEYS[1]=user:123:likes, ARGV[1]=pic_456
    redis.call('SADD', KEYS[1], ARGV[1])
    
2. 返回值
  • Lua脚本的最后一个值会作为执行结果返回给客户端。
  • 可返回nil、数值、字符串、表(自动转为Redis多行回复)。
3. 脚本编写原则
  • 禁止使用全局变量:所有变量需用local声明。
  • 避免长耗时操作:Lua脚本会阻塞Redis其他请求,需确保高效性。

三、常用Lua脚本操作

1. 数据操作
操作示例代码
Stringredis.call('SET', KEYS[1], ARGV[1])
Hashredis.call('HSET', KEYS[1], 'field', ARGV[1])
Setredis.call('SADD', KEYS[1], ARGV[1])
ZSetredis.call('ZADD', KEYS[1], ARGV[1], ARGV[2])
Listredis.call('LPUSH', KEYS[1], ARGV[1])
2. 条件判断
if redis.call('EXISTS', KEYS[1]) == 1 thenreturn redis.call('GET', KEYS[1])
elsereturn nil
end
3. 循环操作
for i=1, #ARGV doredis.call('SADD', KEYS[1], ARGV[i])
end
4. 错误处理
local ok, err = pcall(redis.call, 'INCRBY', KEYS[1], ARGV[1])
if not ok thenreturn {err = err}
end

四、注意事项

场景解决方案
集群模式确保所有KEYS在同一个哈希槽(可通过Hash Tag实现,如{user}:123:likes)。
脚本性能避免复杂循环或大范围数据遍历,优先用Redis原生命令。
脚本缓存使用SCRIPT LOAD预加载脚本,通过EVALSHA执行(减少网络传输)。
调试脚本通过redis.log(redis.LOG_DEBUG, 'message')输出日志(需配置Redis日志级别)。

五、典型应用场景

场景Lua脚本作用
分布式锁原子化实现锁的获取、续期、释放(避免锁误删)。
计数器原子化增减计数(如点赞数、库存扣减)。
排行榜计算分数并更新ZSet,返回排名结果。
批量操作合并多个命令(如先检查条件再删除数据)。

六、调试与测试

  1. 直接执行脚本(通过redis-cli):
    redis-cli --eval script.lua key1 key2 , arg1 arg2
    
  2. 捕获错误
    local ok, result = pcall(redis.call, 'COMMAND', params)
    if not ok thenreturn {error = result}
    end
    

点赞实现

编写lua脚本

public class RedisLuaScript {/*** 点赞 Lua 脚本* KEYS[1]       -- 临时计数键* KEYS[2]       -- 用户点赞状态键* ARGV[1]       -- 用户 ID* ARGV[2]       -- 壁纸 ID* 返回:* -1: 已点赞* 1: 操作成功*/public static final RedisScript<Long> LIKE_SCRIPT = new DefaultRedisScript<>("local tempLikeKey = KEYS[1]\n" +"local userLikeKey = KEYS[2]\n" +"local userId = ARGV[1]\n" +"local picId = ARGV[2]\n" +"\n" +"-- 1. 检查是否已点赞(避免重复操作)\n" +"if redis.call('HEXISTS', userLikeKey, picId) == 1 then\n" +"    return -1\n" +"end\n" +"\n" +"-- 2. 获取旧值(不存在则默认为 0)\n" +"local hashKey = userId .. ':' .. picId\n" +"local oldNumber = tonumber(redis.call('HGET', tempLikeKey, hashKey) or 0)\n" +"\n" +"-- 3. 计算新值\n" +"local newNumber = oldNumber + 1\n" +"\n" +"-- 4. 原子性更新:写入临时计数 + 标记用户已点赞\n" +"redis.call('HSET', tempLikeKey, hashKey, newNumber)\n" +"redis.call('HSET', userLikeKey, picId, 1)\n" +"\n" +"return 1", Long.class);/*** 取消点赞 Lua 脚本* 参数同上* 返回:* -1: 未点赞* 1: 操作成功*/public static final RedisScript<Long> UNLIKE_SCRIPT = new DefaultRedisScript<>("local tempLikeKey = KEYS[1]\n" +  // 显式换行"local userLikeKey = KEYS[2]\n" +"local userId = ARGV[1]\n" +"local picId = ARGV[2]\n" +"\n" +  // 空行分隔逻辑块"-- 1. 检查用户是否已点赞(若未点赞,直接返回失败)\n" +"if redis.call('HEXISTS', userLikeKey, picId) ~= 1 then\n" +"    return -1\n" +"end\n" +"\n" +"-- 2. 获取当前临时计数(若不存在则默认为 0)\n" +"local hashKey = userId .. ':' .. picId\n" +"local oldNumber = tonumber(redis.call('HGET', tempLikeKey, hashKey) or 0)\n" +"\n" +"-- 3. 计算新值并更新\n" +"local newNumber = oldNumber - 1\n" +"\n" +"-- 4. 原子性操作:更新临时计数 + 删除用户点赞标记\n" +"redis.call('HSET', tempLikeKey, hashKey, newNumber)\n" +"redis.call('HDEL', userLikeKey, picId)\n" +"\n" +  // 确保return前有换行"return 1",  // 最后一行无需\n(Redis会自动补全)Long.class);
}

执行lua脚本

redisTemplate.execute(RedisLuaScriptConstant.LIKE_SCRIPT,Arrays.asList(tempLikeKey,userLikeKey),loginUser.getId(),picId
);
redisTemplate.execute(RedisLuaScriptConstant.LIKE_SCRIPT,Arrays.asList(tempLikeKey,userLikeKey),loginUser.getId(),picId
);
http://www.xdnf.cn/news/609.html

相关文章:

  • 【英语语法】词法---动词
  • hadoop分布式部署
  • Linux `init 5` 相关命令的完整使用指南
  • Android学习总结之APK打包流程
  • 【踩坑记录】Pico串流SteamVR绿屏解决方案:排查兼容性问题与Windows系统升级指南
  • STM32 HAL库FreeRTOS 中断管理
  • XSS学习1之http回顾
  • 【读书笔记·VLSI电路设计方法解密】问题63:为什么可测试性设计对产品的财务成功至关重要
  • 机器学习周报-文献阅读
  • FastAPI-MCP
  • 8节串联锂离子电池组可重构buck-boost均衡拓扑结构 simulink模型仿真
  • 个人所得税
  • DeepSeek R1 7b,Langchain 实现 RAG 知识库 | LLMs
  • 抽象工厂模式及其在自动驾驶中的应用举例(c++代码实现)
  • 秒杀抢购系统架构与优化全解:从业务特性到技术落地
  • tigase源码学习杂记-组件化设计
  • AI预测3D新模型百十个定位预测+胆码预测+去和尾2025年4月20日第58弹
  • 智能体团队 (Agent Team)
  • 考研系列-计算机网络-第三章、数据链路层
  • Linux:网络基础
  • 深入理解 Transformer:原理、架构与注意力机制全景图解
  • 微信怎么绑定孩子的医保卡
  • w299基于Java的家政服务平台设计与实现
  • idea中运行groovy程序报错
  • FISCO 2.0 安装部署WeBASE与区块链浏览器(环境搭建)
  • 【Linux学习笔记】Linux的环境变量和命令行参数
  • 【支付】支付宝支付
  • FastAPI:现代高性能Python Web框架的技术解析与实践指南
  • 【刷题Day21】TCP(浅)
  • Java枚举