服务器之家：专注于VPS、云服务器配置技术及软件下载分享
分类导航

Mysql|Sql Server|Oracle|Redis|MongoDB|PostgreSQL|Sqlite|DB2|mariadb|Access|数据库技术|

服务器之家 - 数据库 - Redis - 基于redis+lua进行限流的方法

基于redis+lua进行限流的方法

2022-07-24 12:28枯枫叶 Redis

这篇文章主要介绍了基于redis+lua进行限流,通过实例代码详细介绍了lua+redis进行限流的做法，开发环境使用idea+redis+lua，本文给大家介绍的非常详细，需要的朋友可以参考下

1，首先我们redis有很多限流的算法(比如：令牌桶，计数器，时间窗口)等，但是都有一定的缺点，令牌桶在单项目中相对来说比较稳定，但是在分布式集群里面缺显的不那么友好，这时候，在分布式里面进行限流的话，我们则可以使用redis+lua脚本进行限流，能抗住亿级并发

2，下面说说lua+redis进行限流的做法
开发环境：idea+redis+lua
第一：
打开idea的插件市场，然后搜索lua，点击右边的安装，然后安装好了，重启即可

基于redis+lua进行限流的方法

第二：写一个自定义限流注解

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
package com.sport.sportcloudmarathonh5.config;
 
import java.lang.annotation.*;
 
/**
 * @author zdj
 * @version 1.0.0
 * @description 自定义注解实现分布式限流
 */
@Target(value = ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface RedisLimitStream {
    /**
     * 请求限制，一秒内可以允许好多个进入(默认一秒可以支持100个)
     * @return
     */
    int reqLimit() default 1000;
 
    /**
     * 模块名称
     * @return
     */
    String reqName() default "";
}

第三：在指定的方法上面添加该注解

?
1
2
3
4
5
6
7
8
9
10
11
/**
     * 压测接口
     * @return
     */
    @Login(isLogin = false)
    @RedisLimitStream(reqName = "名额秒杀", reqLimit = 1000)
    @ApiOperation(value = "压测接口", notes = "压测接口", httpMethod = "GET")
    @RequestMapping(value = "/pressure", method = RequestMethod.GET)
    public ResultVO<Object> pressure(){
        return ResultVO.success("抢购成功！");
    }

第四：添加一个拦截器对访问的方法在访问之前进行拦截：

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
package com.sport.sportcloudmarathonh5.config;
 
import com.alibaba.fastjson.JSONObject;
import com.sport.sportcloudmarathonh5.service.impl.RedisService;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.script.RedisScript;
import org.springframework.stereotype.Component;
import org.springframework.util.ObjectUtils;
 
import javax.servlet.http.HttpServletResponse;
import java.io.PrintWriter;
import java.util.ArrayList;
import java.util.List;
 
/**
 * @author zdj
 * @version 1.0.0
 * @description MyRedisLimiter注解的切面类
 */
@Aspect
@Component
public class RedisLimiterAspect {
    private final Logger logger = LoggerFactory.getLogger(RedisLimitStream.class);
    /**
     * 当前响应请求
     */
    @Autowired
    private HttpServletResponse response;
 
    /**
     * redis服务
     */
    @Autowired
    private RedisService redisService;
 
    /**
     * 执行redis的脚本文件
     */
    @Autowired
    private RedisScript<Boolean> rateLimitLua;
 
    /**
     * 对所有接口进行拦截
     */
    @Pointcut("execution(public * com.sport.sportcloudmarathonh5.controller.*.*(..))")
    public void pointcut(){}
 
    /**
     * 对切点进行继续处理
     */
    @Around("pointcut()")
    public Object process(ProceedingJoinPoint proceedingJoinPoint) throws Throwable{
        //使用反射获取RedisLimitStream注解
        MethodSignature signature = (MethodSignature) proceedingJoinPoint.getSignature();
        //没有添加限流注解的方法直接放行
        RedisLimitStream redisLimitStream = signature.getMethod().getDeclaredAnnotation(RedisLimitStream.class);
        if(ObjectUtils.isEmpty(redisLimitStream)){
            return proceedingJoinPoint.proceed();
        }
 
        //List设置Lua的KEYS[1]
        List<String> keyList = new ArrayList<>();
        keyList.add("ip:" + (System.currentTimeMillis() / 1000));
 
        //获取注解上的参数，获取配置的速率
        //List设置Lua的ARGV[1]
        int value = redisLimitStream.reqLimit();
 
        // 调用Redis执行lua脚本，未拿到令牌的，直接返回提示
        boolean acquired = redisService.execute(rateLimitLua, keyList, value);
        logger.info("执行lua结果：" + acquired);
        if(!acquired){
            this.limitStreamBackMsg();
            return null;
        }
 
        //获取到令牌，继续向下执行
        return proceedingJoinPoint.proceed();
    }
 
    /**
     * 被拦截的人，提示消息
     */
    private void limitStreamBackMsg() {
        response.setHeader("Content-Type", "text/html;charset=UTF8");
        PrintWriter writer = null;
        try {
            writer = response.getWriter();
            writer.println("{\"code\":503,\"message\":\"当前排队人较多，请稍后再试！\",\"data\":\"null\"}");
            writer.flush();
        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            if (writer != null) {
                writer.close();
            }
        }
    }
}

第五：写个配置类，在启动的时候将我们的lua脚本代码加载到redisscript中

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
package com.sport.sportcloudmarathonh5.config;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.io.ClassPathResource;
import org.springframework.data.redis.core.script.DefaultRedisScript;
 
/**
 * @author zdj
 * @version 1.0.0
 * @description 实现redis的编码方式
 */
@Configuration
public class RedisConfiguration {
 
    /**
     * 初始化将lua脚本加载到redis脚本中
     * @return
     */
    @Bean
    public DefaultRedisScript loadRedisScript() {
        DefaultRedisScript redisScript = new DefaultRedisScript();
        redisScript.setLocation(new ClassPathResource("limit.lua"));
        redisScript.setResultType(Boolean.class);
        return redisScript;
    }
}

第六：redis执行lua的方法

?
1
2
3
4
5
6
7
8
9
10
/**
   * 执行lua脚本
   * @param redisScript lua源代码脚本
   * @param keyList
   * @param value
   * @return
   */
  public boolean execute(RedisScript<Boolean> redisScript, List<String> keyList, int value) {
      return redisTemplate.execute(redisScript, keyList, String.valueOf(value));
  }

第七：在resources目录下面新加一个lua脚本文件，将下面代码拷贝进去即可：

?
1
2
3
4
5
6
7
8
9
10
local key = KEYS[1] --限流KEY（一秒一个）
local limit = tonumber(ARGV[1]) --限流大小
local current = tonumber(redis.call('get', key) or "0")
if current + 1 > limit then --如果超出限流大小
    return false
else --请求数+1，并设置2秒过期
    redis.call("INCRBY", key, "1")
    redis.call("expire", key, "2")
end
return true

基于redis+lua进行限流的方法

最后执行即可：
可以使用jemster进行测试：

基于redis+lua进行限流的方法

到此这篇关于基于redis+lua进行限流的文章就介绍到这了,更多相关redis lua限流内容请搜索服务器之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持服务器之家！

原文链接：https://blog.csdn.net/gelinwangzi_juge/article/details/125919151

  • redis
  • 限流
  • Lua

    • 延伸 · 阅读

    精彩推荐
    • RedisRedis为什么要存两次数据

      Redis为什么要存两次数据

      这篇文章主要介绍了Redis为什么要存两次数据的相关资料,需要的朋友可以参考下...

      双子孤狼11172021-07-26
    • RedisRedis分布式缓存：微信抢红包解决方案

      Redis分布式缓存：微信抢红包解决方案

      微信抢红包已经在我们生活中很常见的场景了，特别是年底公司开年会和春节2个时间段。本文主要介绍了通过Redis实现微信抢红包功能，感兴趣的小伙伴可...

      小伙子vae3412022-01-24
    • RedisRedis实现分布式锁的实例讲解

      Redis实现分布式锁的实例讲解

      在本篇文章里小编给大家整理了一篇关于Redis实现分布式锁的实例讲解内容，有兴趣的朋友们可以学习参考下。...

      青灯夜游9732022-01-22
    • RedisRedis如何实现分布式锁

      Redis如何实现分布式锁

      相信大家对锁已经不陌生了，本文主要介绍了Redis如何实现分布式锁，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参...

      公众号程序员学长10462021-09-17
    • RedisRedis配合SSDB实现持久化存储代码示例

      Redis配合SSDB实现持久化存储代码示例

      这篇文章主要介绍了Redis配合SSDB实现持久化存储代码示例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友...

      -零3982021-01-05
    • RedisRedis三种集群模式详解

      Redis三种集群模式详解

      redis有三种集群模式，其中主从是最常见的模式，今天通过本文给大家分享Redis三种集群模式介绍，感兴趣的朋友一起看看吧...

      coffeebabe3812021-11-19
    • RedisLinux下redis的安装与使用图文教程

      Linux下redis的安装与使用图文教程

      这篇文章主要介绍了Linux下redis的安装与使用,结合图文形式分析了Linux环境下redis的下载、编译、安装、部署、访问等相关操作技巧,需要的朋友可以参考下...

      Tony9184532019-11-27
    • RedisRedis基本知识、安装、部署、配置笔记

      Redis基本知识、安装、部署、配置笔记

      这篇文章主要介绍了Redis基本知识、安装、部署、配置笔记,本文讲解了基本知识、安装Redis、Redis的配置等知识,需要的朋友可以参考下...

      Redis教程网7202021-01-08
    最近更新
    编辑推荐
    阅读排行
    热门标签
    16384 　 批量提权 　 intset 　 节点迁移 　 事务机制 　 主从同步 　 十六进制 　 内部数据结构 　 Redis集群 　 Redis集群搭建 　 持久化方案 　 AOF 　 ziplist 　 限流器 　 内存型数据库 　 RDB 　 简介 　 哨兵机制 　 批量读写 　 密码认证 　 redis命令 　 未授权 　 访问漏洞 　 redislive 　 发布 　 集合类型 　 列表类型 　 快照模式 　 集群搭建 　 键 　
    © 2019-2022 服务器之家(www.tuohang.net) 版权所有关于我们联系我们版权申明网站地图