redis+lua分布式限流

浅浅的花香味﹌ 2023-07-10 08:21 2阅读 0赞

### 文章目录 ###

*  前言
 *  代码
 *   *  这是什么限流？
     *  核心限流代码
 *  注解
 *  反思

# 前言 #

我们项目有好几个接口，调用公司中台的接口，包括定时器的分片执行还有本人的多线程强行压榨，哈哈。最后加上用户的疯狂请求，导致中台的接口偶尔出现问题，主要是数据库cpu达到100%。

昨晚专门做了个限流，那么技术定型使用什么呢？ratelimiter?这好像是单机才能玩，sentinel？这个还要搭建环境。那就只能redis+lua。

# 代码 #

lua脚本 **放在resource目录下**

local key = KEYS[1];
    local limit = tonumber(ARGV[1]);
    local val = redis.call("INCRBY", key,"1");
    
    if val == 1 then
        redis.call("expire", key,"1")
        return 1;
    end
    if val > limit then
        return 0;
    end
    return 1;

意思：incrby自增一个key，如果是1的话，证明刚刚设置，需要设定过期时间为1秒，如果不是的话返回1，再如果超过我们限制的limit值，那么返回0，表示限流。

## 这是什么限流？ ##

**计数法啦，当然简单也有缺点。**

## 核心限流代码 ##

@Slf4j
    @Component
    public class RedisLimit {
    
        @Autowired
        private RedisTemplate redisTemplate;
    
        private static DefaultRedisScript<Long> GET_REDIS_SCRIPT;
    
        private static final String CACHE_KEY = "xx";
    
        @PostConstruct
        public void init() {
            GET_REDIS_SCRIPT = new DefaultRedisScript<>();
            GET_REDIS_SCRIPT.setResultType(Long.class);
            GET_REDIS_SCRIPT.setLocation(new ClassPathResource("limit.lua"));
        }
    
        /**
         * 1秒内限流
         *
         * @param key
         * @param limit
         * @return true 放行 false gg
         */
        public boolean accquire(String key, Integer limit) {
    
            //校验
            if (StrUtil.isBlank(key) || limit == null) {
                throw new BusinessException(ActivityErrorCodeEnum.AC5200003);
            }
    
            //List设置lua的KEYS
            List<String> keyList = new ArrayList<>();
            keyList.add(String.format(CACHE_KEY, key));
            // 执行lua脚本，传入参数
            Long result = (Long) (redisTemplate.execute(GET_REDIS_SCRIPT, keyList, limit));
            System.out.println(result);
            return result == 1;
        }
    
    }

**解析一下**

使用redisTemplate.execute来执行lua了，注意这里有个坑，如果你是直接去拿connect连接的，记得要释放资源，在spring与redis集成比较低的版本是没有说去释放资源的！！！

# 注解 #

为了通用使用aop注解

**方法注解头**

@Target({ElementType.METHOD, ElementType.FIELD, ElementType.CONSTRUCTOR, ElementType.PARAMETER, ElementType.TYPE_USE})
    @Retention(RetentionPolicy.RUNTIME)  // 保留到运行时，可通过注解获取
    @Documented
    public @interface MyRedisLimit {
    
        String key() default "";
    
        String limit() default "40";
    
    }

一个是缓存的key，一个是限制最大的数量limit，这里不能用Integer.

**AOP逻辑**

@Aspect
    @Component
    public class TestAspect {
    
        @Resource
        RedisLimit limit;
    
        @Pointcut("@annotation(com.yatsenglobal.cloud.limit.MyRedisLimit)")
        public void annotationPoinCut() {
    
        }
    
        @Before(value = "annotationPoinCut()")
        public void beforeTest(JoinPoint point) {
            //获取方法签名
            MethodSignature signature = (MethodSignature) point.getSignature();
            //获取切入方法的对象
            Method method = signature.getMethod();
            //获取方法上的Aop注解
            MyRedisLimit myMethod = method.getAnnotation(MyRedisLimit.class);
    
            //限流
            if (!limit.accquire(myMethod.key(), Integer.valueOf(myMethod.limit()))) {
                throw new BusinessException(ActivityErrorCodeEnum.WZPRESENT21000);
            }
        }
    
    }

到此完成了，但是记住加上这个注解，必须使用spring注入去调用，如果是本地方法一样调用的话，这个注解就失效了！

# 反思 #

ratelimiter在网上看很少用java来配合redis实现分布式限流的，那么sentinel是如何做到的呢？期待下一篇文章