redis实现API限流

最新推荐文章于 2024-12-20 13:53:41 发布

码农记事本

最新推荐文章于 2024-12-20 13:53:41 发布

阅读量826

点赞数

分类专栏： java笔记本文章标签： redis 限流

本文链接：https://blog.youkuaiyun.com/leopast/article/details/123125319

版权

java笔记本专栏收录该内容

32 篇文章

订阅专栏

本文介绍了如何在SpringBoot应用中实现接口限流，通过自定义注解定义限流时间和最大请求次数，利用拦截器检查请求，并结合Redis进行计数和超时管理。示例代码展示了注解定义、拦截器实现以及配置过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

思路

自定义注解定义限流片段时间和最大请求个数
拦截器拦截所有请求，如果发现使用了限流注解的API，查看redis里是否记录了该接口的请求次数，如果有，就判断没有超过自定义注解的最大阈值，达到阈值就拦截该请求，否则请求次数加1并放行
使用到redis key自过过期方法，自增量方法

springboot示例代码

注解

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface AccessLimit {

    int seconds() default 1;
    int maxCount() default 20;

}

拦截器

@Slf4j
public class AccessLimitInterceptor implements HandlerInterceptor {

    @Autowired
    private RedisUtil redisUtil;

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        if(handler instanceof HandlerMethod){
            HandlerMethod handlerMethod = (HandlerMethod) handler;
            AccessLimit accessLimit = handlerMethod.getMethodAnnotation(AccessLimit.class);
            if(accessLimit == null) return true;
            int seconds = accessLimit.seconds();
            int maxCount = accessLimit.maxCount();
            String url = request.getRequestURI();
            String key = RedisKey.accessLimit(url);
            if(redisUtil.hasKey(key)){
                Integer count = redisUtil.get(key);
                if(count == null) return true;
                if(count >= maxCount) throw new APIException(APICode.SYSTEM_BUSY);
                redisUtil.increment(key, 1);
            }else {
                redisUtil.set(key, 1, seconds);
            }
        }
        return true;
    }

}

配置

@Configuration
public class WebConfig implements WebMvcConfigurer {

    @Bean
    public AccessLimitInterceptor accessLimitInterceptor(){
        return new AccessLimitInterceptor();
    }

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registration = registry.addInterceptor(accessLimitInterceptor());
        registration.addPathPatterns("/**");
        registration.excludePathPatterns(
                "/",
                "/**/swagger-resources/**",
                "/**/*.html",
                "/**/*.js",
                "/**/*.css"
        );
    }
}