Redis+Lua脚本实现计数器接口防刷功能(升级版)
来源:脚本之家
2022-12-30 20:15:13
0浏览
收藏
IT行业相对于一般传统行业,发展更新速度更快,一旦停止了学习,很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习,精进自己的技术,尤其是初学者。今天golang学习网给大家整理了《Redis+Lua脚本实现计数器接口防刷功能(升级版)》,聊聊Redis计数器、接口防刷,我们一起来看看吧!
【前言】
Cash Loan(一):Redis实现计数器防刷 中介绍了项目中应用redis来做计数器的实现过程,最近自己看了些关于Redis实现分布式锁的代码后,发现在Redis分布式锁中出现一个问题在这版计数器中同样会出现,于是融入了Lua脚本进行升级改造有了Redis+Lua版本。
【实现过程】
一、问题分析
如果set命令设置上,但是在设置失效时间时由于网络抖动等原因导致没有设置成功,这时就会出现死计数器(类似死锁);
二、解决方案
Redis+Lua是一个很好的解决方案,使用脚本使得set命令和expire命令一同达到Redis被执行且不会被干扰,在很大程度上保证了原子操作;
为什么说是很大程度上保证原子操作而不是完全保证?因为在Redis内部执行的时候出问题也有可能出现问题不过概率非常小;即使针对小概率事件也有相应的解决方案,比如解决死锁一个思路值得参考:防止死锁会将锁的值存成一个时间戳,即使发生没有将失效时间设置上在判断是否上锁时可以加上看看其中值距现在是否超过一个设定的时间,如果超过则将其删除重新设置锁。
三、代码改造
1、Redis+Lua锁的实现
package han.zhang.utils; import org.springframework.data.redis.core.StringRedisTemplate; import org.springframework.data.redis.core.script.DigestUtils; import org.springframework.data.redis.core.script.RedisScript; import java.util.Collections; import java.util.UUID; public class RedisLock { private static final LogUtils logger = LogUtils.getLogger(RedisLock.class); private final StringRedisTemplate stringRedisTemplate; private final String lockKey; private final String lockValue; private boolean locked = false; /** * 使用脚本在redis服务器执行这个逻辑可以在一定程度上保证此操作的原子性 * (即不会发生客户端在执行setNX和expire命令之间,发生崩溃或失去与服务器的连接导致expire没有得到执行,发生永久死锁) * <p> * 除非脚本在redis服务器执行时redis服务器发生崩溃,不过此种情况锁也会失效 */ private static final RedisScript<boolean> SETNX_AND_EXPIRE_SCRIPT; static { StringBuilder sb = new StringBuilder(); sb.append("if (redis.call('setnx', KEYS[1], ARGV[1]) == 1) then\n"); sb.append("\tredis.call('expire', KEYS[1], tonumber(ARGV[2]))\n"); sb.append("\treturn true\n"); sb.append("else\n"); sb.append("\treturn false\n"); sb.append("end"); SETNX_AND_EXPIRE_SCRIPT = new RedisScriptImpl(sb.toString(), Boolean.class); } private static final RedisScript<boolean> DEL_IF_GET_EQUALS; sb.append("if (redis.call('get', KEYS[1]) == ARGV[1]) then\n"); sb.append("\tredis.call('del', KEYS[1])\n"); DEL_IF_GET_EQUALS = new RedisScriptImpl(sb.toString(), Boolean.class); public RedisLock(StringRedisTemplate stringRedisTemplate, String lockKey) { this.stringRedisTemplate = stringRedisTemplate; this.lockKey = lockKey; this.lockValue = UUID.randomUUID().toString() + "." + System.currentTimeMillis(); private boolean doTryLock(int lockSeconds) { if (locked) { throw new IllegalStateException("already locked!"); } locked = stringRedisTemplate.execute(SETNX_AND_EXPIRE_SCRIPT, Collections.singletonList(lockKey), lockValue, String.valueOf(lockSeconds)); return locked; * 尝试获得锁,成功返回true,如果失败立即返回false * * @param lockSeconds 加锁的时间(秒),超过这个时间后锁会自动释放 public boolean tryLock(int lockSeconds) { try { return doTryLock(lockSeconds); } catch (Exception e) { logger.error("tryLock Error", e); return false; * 轮询的方式去获得锁,成功返回true,超过轮询次数或异常返回false * @param lockSeconds 加锁的时间(秒),超过这个时间后锁会自动释放 * @param tryIntervalMillis 轮询的时间间隔(毫秒) * @param maxTryCount 最大的轮询次数 public boolean tryLock(final int lockSeconds, final long tryIntervalMillis, final int maxTryCount) { int tryCount = 0; while (true) { if (++tryCount >= maxTryCount) { // 获取锁超时 return false; } try { if (doTryLock(lockSeconds)) { return true; } } catch (Exception e) { logger.error("tryLock Error", e); Thread.sleep(tryIntervalMillis); } catch (InterruptedException e) { logger.error("tryLock interrupted", e); * 解锁操作 public void unlock() { if (!locked) { throw new IllegalStateException("not locked yet!"); locked = false; // 忽略结果 stringRedisTemplate.execute(DEL_IF_GET_EQUALS, Collections.singletonList(lockKey), lockValue); private static class RedisScriptImpl<t> implements RedisScript<t> { private final String script; private final String sha1; private final Class<t> resultType; public RedisScriptImpl(String script, Class<t> resultType) { this.script = script; this.sha1 = DigestUtils.sha1DigestAsHex(script); this.resultType = resultType; @Override public String getSha1() { return sha1; public Class<t> getResultType() { return resultType; public String getScriptAsString() { return script; }</t></t></t></t></t></boolean></boolean></p>
2、借鉴锁实现Redis+Lua计数器
(1)工具类
package han.zhang.utils; import org.springframework.data.redis.core.StringRedisTemplate; import org.springframework.data.redis.core.script.DigestUtils; import org.springframework.data.redis.core.script.RedisScript; import java.util.Collections; public class CountUtil { private static final LogUtils logger = LogUtils.getLogger(CountUtil.class); private final StringRedisTemplate stringRedisTemplate; /** * 使用脚本在redis服务器执行这个逻辑可以在一定程度上保证此操作的原子性 * (即不会发生客户端在执行setNX和expire命令之间,发生崩溃或失去与服务器的连接导致expire没有得到执行,发生永久死计数器) * <p> * 除非脚本在redis服务器执行时redis服务器发生崩溃,不过此种情况计数器也会失效 */ private static final RedisScript<boolean> SET_AND_EXPIRE_SCRIPT; static { StringBuilder sb = new StringBuilder(); sb.append("local visitTimes = redis.call('incr', KEYS[1])\n"); sb.append("if (visitTimes == 1) then\n"); sb.append("\tredis.call('expire', KEYS[1], tonumber(ARGV[1]))\n"); sb.append("\treturn false\n"); sb.append("elseif(visitTimes > tonumber(ARGV[2])) then\n"); sb.append("\treturn true\n"); sb.append("else\n"); sb.append("end"); SET_AND_EXPIRE_SCRIPT = new RedisScriptImpl(sb.toString(), Boolean.class); } public CountUtil(StringRedisTemplate stringRedisTemplate) { this.stringRedisTemplate = stringRedisTemplate; public boolean isOverMaxVisitTimes(String key, int seconds, int maxTimes) throws Exception { try { return stringRedisTemplate.execute(SET_AND_EXPIRE_SCRIPT, Collections.singletonList(key), String.valueOf(seconds), String.valueOf(maxTimes)); } catch (Exception e) { logger.error("RedisBusiness>>>isOverMaxVisitTimes; get visit times Exception; key:" + key + "result:" + e.getMessage()); throw new Exception("already Over MaxVisitTimes"); } private static class RedisScriptImpl<t> implements RedisScript<t> { private final String script; private final String sha1; private final Class<t> resultType; public RedisScriptImpl(String script, Class<t> resultType) { this.script = script; this.sha1 = DigestUtils.sha1DigestAsHex(script); this.resultType = resultType; @Override public String getSha1() { return sha1; public Class<t> getResultType() { return resultType; public String getScriptAsString() { return script; }</t></t></t></t></t></boolean></p>
(2)调用测试代码
public void run(String... strings) { CountUtil countUtil = new CountUtil(SpringUtils.getStringRedisTemplate()); try { for (int i = 0; i <p>(3)测试结果</p> <p style="text-align:center"><img alt="" src="/uploads/20221230/167240342063aed9dc1092f.png"></p> <h2>【总结】</h2> <p> 1、用心去不断的改造自己的程序;</p> <p> 2、用代码改变世界。</p> <p>今天带大家了解了Redis计数器、接口防刷的相关知识,希望对你有所帮助;关于数据库的技术知识我们会一点点深入介绍,欢迎大家关注golang学习网公众号,一起学习编程~</p>
版本声明
本文转载于:脚本之家 如有侵犯,请联系study_golang@163.com删除

- 上一篇
- Redis命令处理过程源码解析

- 下一篇
- 基于Redis zSet实现滑动窗口对短信进行防刷限流的问题
评论列表
-
- 现代的金毛
- 这篇技术文章太及时了,太全面了,感谢大佬分享,已收藏,关注作者了!希望作者能多写数据库相关的文章。
- 2023-02-05 11:23:02
-
- 喜悦的战斗机
- 这篇文章出现的刚刚好,太全面了,受益颇多,码起来,关注老哥了!希望老哥能多写数据库相关的文章。
- 2023-01-23 01:26:00
-
- 高挑的硬币
- 写的不错,一直没懂这个问题,但其实工作中常常有遇到...不过今天到这,看完之后很有帮助,总算是懂了,感谢作者大大分享博文!
- 2023-01-09 21:32:12
-
- 无聊的牛排
- 太细致了,已收藏,感谢大佬的这篇技术贴,我会继续支持!
- 2023-01-06 01:47:36
查看更多
最新文章
-
- 数据库 · Redis | 8小时前 |
- Redis与MySQL缓存同步方法解析
- 245浏览 收藏
-
- 数据库 · Redis | 9小时前 |
- Redis性能监控工具有哪些
- 124浏览 收藏
-
- 数据库 · Redis | 10小时前 |
- RedisList队列优化方法分享
- 378浏览 收藏
-
- 数据库 · Redis | 10小时前 |
- Redis位图实现用户签到优化方案
- 322浏览 收藏
-
- 数据库 · Redis | 19小时前 |
- Redis数据安全防护全攻略
- 112浏览 收藏
-
- 数据库 · Redis | 19小时前 |
- Redis哈希技巧与实战应用
- 204浏览 收藏
-
- 数据库 · Redis | 19小时前 |
- 扩展Redis集群节点的步骤与注意事项
- 163浏览 收藏
-
- 数据库 · Redis | 20小时前 |
- 高并发Redis优化技巧分享
- 147浏览 收藏
-
- 数据库 · Redis | 20小时前 |
- Redis主从复制故障排查指南
- 477浏览 收藏
-
- 数据库 · Redis | 1天前 |
- Redis与HBase存储方案详解
- 414浏览 收藏
-
- 数据库 · Redis | 1天前 |
- Redis与MongoDB缓存优化方法
- 193浏览 收藏
-
- 数据库 · Redis | 1天前 |
- Redis安全配置更新操作教程
- 313浏览 收藏
查看更多
课程推荐
-
- 前端进阶之JavaScript设计模式
- 设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
- 542次学习
-
- GO语言核心编程课程
- 本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
- 511次学习
-
- 简单聊聊mysql8与网络通信
- 如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
- 498次学习
-
- JavaScript正则表达式基础与实战
- 在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
- 487次学习
-
- 从零制作响应式网站—Grid布局
- 本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
- 484次学习
查看更多
AI推荐
-
- 千音漫语
- 千音漫语,北京熠声科技倾力打造的智能声音创作助手,提供AI配音、音视频翻译、语音识别、声音克隆等强大功能,助力有声书制作、视频创作、教育培训等领域,官网:https://qianyin123.com
- 96次使用
-
- MiniWork
- MiniWork是一款智能高效的AI工具平台,专为提升工作与学习效率而设计。整合文本处理、图像生成、营销策划及运营管理等多元AI工具,提供精准智能解决方案,让复杂工作简单高效。
- 89次使用
-
- NoCode
- NoCode (nocode.cn)是领先的无代码开发平台,通过拖放、AI对话等简单操作,助您快速创建各类应用、网站与管理系统。无需编程知识,轻松实现个人生活、商业经营、企业管理多场景需求,大幅降低开发门槛,高效低成本。
- 107次使用
-
- 达医智影
- 达医智影,阿里巴巴达摩院医疗AI创新力作。全球率先利用平扫CT实现“一扫多筛”,仅一次CT扫描即可高效识别多种癌症、急症及慢病,为疾病早期发现提供智能、精准的AI影像早筛解决方案。
- 98次使用
-
- 智慧芽Eureka
- 智慧芽Eureka,专为技术创新打造的AI Agent平台。深度理解专利、研发、生物医药、材料、科创等复杂场景,通过专家级AI Agent精准执行任务,智能化工作流解放70%生产力,让您专注核心创新。
- 98次使用
查看更多
相关文章
-
- Redis实现高并发计数器
- 2023-01-07 225浏览