首页
所有分类
TAGS
文字工具 EMOJI BIBLE
数据库 Redis

Redis+Lua脚本实现计数器接口防刷功能(升级版)

Posted in Redis onFebruary 12, 2022

【前言】

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;
}

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;
}

（2）调用测试代码

public void run(String... strings) {
        CountUtil countUtil = new CountUtil(SpringUtils.getStringRedisTemplate());
        try {
            for (int i = 0; i < 10; i++) {
                boolean overMax = countUtil.isOverMaxVisitTimes("zhanghantest", 600, 2);
                if (overMax) {
                    System.out.println("超过i:" + i + ":" + overMax);
                } else {
                    System.out.println("没超过i:" + i + ":" + overMax);
                }
            }
        } catch (Exception e) {
            logger.error("Exception {}", e.getMessage());
        }
    }

（3）测试结果

Redis+Lua脚本实现计数器接口防刷功能(升级版)

【总结】

       1、用心去不断的改造自己的程序；

       2、用代码改变世界。

到此这篇关于Redis+Lua实现计数器接口防刷(升级版)的文章就介绍到这了,更多相关Redis计数器内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木！

Redis+Lua脚本实现计数器接口防刷功能(升级版)

- Author -

当年的春天

- Original Sources -

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

Redis 相关文章推荐
使用Redis实现秒杀功能的简单方法
May 08 Redis
详解Redis主从复制实践
May 19 Redis
在redisCluster中模糊获取key方式
Jul 09 Redis
解析redis hash应用场景和常用命令
Aug 04 Redis
Redis的字符串是如何实现的
Oct 24 Redis
Redis+Lua脚本实现计数器接口防刷功能(升级版)
Feb 12 Redis
浅谈Redis跟MySQL的双写问题解决方案
Feb 24 Redis
浅谈Redis 中的过期删除策略和内存淘汰机制
Apr 03 Redis
Redis超详细讲解高可用主从复制基础与哨兵模式方案
Apr 07 Redis
Redis基本数据类型Zset有序集合常用操作
Jun 01 Redis
Redis实现分布式锁的五种方法详解
Jun 14 Redis
一文教你快速生成MySQL数据库关系图
Jun 28 Redis
Spring Boot实战解决高并发数据入库之 Redis 缓存+MySQL 批量入库问题
基于Redis zSet实现滑动窗口对短信进行防刷限流的问题
Feb 12 #Redis
聊聊redis-dump工具安装问题
Jan 18 #Redis
redis的list数据类型相关命令介绍及使用
Jan 18 #Redis
关于使用Redisson订阅数问题
Jan 18 #Redis
Redis中缓存穿透/击穿/雪崩问题和解决方法
linux下安装redis图文详细步骤
荷兰购物网站(99) 阿根廷购物网站(9) 南非购物网站(8) 比利时购物网站(33) 挪威购物网站(7) 欧洲购物网站(15) 肯尼亚购物网站(2) 瑞典购物网站(55) 葡萄牙购物网站(13) 芬兰购物网站(10)
You might like
解析PHP 5.5 新特性
2013/07/02 PHP
php求正负数数组中连续元素最大值示例
2014/04/11 PHP
PHP判断一个字符串是否是回文字符串的方法
2015/03/23 PHP
php实现有趣的人品测试程序实例
2015/06/08 PHP
PHP微信模板消息操作示例
2017/06/29 PHP
PHP实现压缩图片尺寸并转为jpg格式的方法示例
2018/05/10 PHP
Laravel 框架基于自带的用户系统实现登录注册及错误处理功能分析
2020/04/14 PHP
JavaScript 设计模式学习 Singleton
2009/07/27 Javascript
通过jquery的$.getJSON做一个跨域ajax请求试验
2011/05/03 Javascript
jQuery中:empty选择器用法实例
2014/12/30 Javascript
vue 使用高德地图vue-amap组件过程解析
2019/09/07 Javascript
vuex的使用和简易实现
2021/01/07 Vue.js
python操作mongodb根据_id查询数据的实现方法
2015/05/20 Python
Python文件处理
2016/02/29 Python
django限制匿名用户访问及重定向的方法实例
2018/02/07 Python
Python随机生成身份证号码及校验功能
2018/12/04 Python
Python面向对象之继承和多态用法分析
2019/06/08 Python
Python中字符串List按照长度排序
2019/07/01 Python
Python占用的内存优化教程
2019/07/28 Python
PyQt5通信机制 信号与槽详解
2019/08/07 Python
Python3的unicode编码转换成中文的问题及解决方案
2019/12/10 Python
python和go语言的区别是什么
2020/07/20 Python
HTML5对手机页面长按会粘贴复制禁用的解决方法
2016/07/19 HTML / CSS
html Table 表头固定的实现
2019/01/22 HTML / CSS
英国虚拟主机服务商：eUKhost
2016/08/16 全球购物
Puccini乌克兰：购买行李箱、女士手袋网上商店
2020/08/06 全球购物
CAD制图人员的自荐信
2014/02/07 职场文书
八年级语文教学反思
2014/02/11 职场文书
副董事长岗位职责
2014/04/02 职场文书
金融专业毕业生自荐信
2014/06/26 职场文书
“四风”查摆问题自我剖析材料
2014/09/27 职场文书
离职感谢信怎么写
2015/01/22 职场文书
2015幼儿园新学期寄语
2015/02/27 职场文书
幼儿园保育员随笔
2015/08/14 职场文书
利用Python实现模拟登录知乎
2022/05/25 Python
项目中Nginx多级代理是如何获取客户端的真实IP地址
2022/05/30 Servers