使用lua脚本基于redis缓存实现计时器（令牌桶算法）

9居不挂

于 2024-09-29 16:51:42 发布

阅读量305

点赞数 11

文章标签：缓存 redis 算法

本文链接：https://blog.csdn.net/2402_84037826/article/details/142638543

版权

本次给大家带来的是lua脚本，相信很多大佬都是懂的，所以不感兴趣可以移步，不懂的老铁们继续往下看哈！

当然，这个是我转载别人的，在github上下载的开源项目，看到这个方法确实不错，于是乎就分享出来了。

先说出处！转自‘少年阿斌’分享的MallChat 抹茶聊天项目项目地址：MallChat

接下来我们看下代码：

private static final String LUA_INCR_EXPIRE =
            "local key,ttl=KEYS[1],ARGV[1] \n" +
                    " \n" +
                    "if redis.call('EXISTS',key)==0 then   \n" +
                    "  redis.call('SETEX',key,ttl,1) \n" +
                    "  return 1 \n" +
                    "else \n" +
                    "  return tonumber(redis.call('INCR',key)) \n" +
                    "end ";

这段是lua脚本的执行代码，具体含义就是每次访问key时将其值加1，并设置为过期时间为ARGV[1]秒。如果该键不存在，就设置为过期时间为ARGV[1]秒并将其值设置为1。如果key已经存在，则将其值加1并返回

然后就是获取了


    public static Integer integerInc(String key, int time, TimeUnit unit) {
        RedisScript<Long> redisScript = new DefaultRedisScript<>(LUA_INCR_EXPIRE, Long.class);
        Long result = stringRedisTemplate.execute(redisScript, Collections.singletonList(key), String.valueOf(unit.toSeconds(time)));
        try {
            return Integer.parseInt(result.toString());
        } catch (Exception e) {
            RedisUtils.del(key);
            throw e;
        }
    }

使用DefaultRedisScript类来创建Redis脚本，然后执行脚本获取结果，报错捕获异常

此方法一方面可以当计时器用，另一方面可以确保唯一不重复作为扩展id使用，具体看自己的使用场景！

好了就说到这，有感兴趣的老铁可以去拉源代码研究，该项目使用了很多设计模式，相信一定帮助你快速成长！

收尾！感谢收看，能帮到各位是在下的荣幸！