redis 访问限流

---

redis 访问限流

 

利用分布式计数器限制某一路径在一段时间内的访问次数

 

 

******************

示例

 

*****************

拦截器

 

CustomInterceptor

@Component
public class CustomInterceptor implements HandlerInterceptor {

    @Autowired
    private StringRedisTemplate redisTemplate;

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String key=request.getRequestURI();
        if(redisTemplate.boundValueOps(key).get()==null){
            redisTemplate.boundValueOps(key).set("1", Duration.ofMinutes(1L));
            return true;
        }else {
            redisTemplate.opsForValue().increment(key);
        }

        if(Integer.parseInt(Objects.requireNonNull(redisTemplate.opsForValue().get(key)))>5){
            response.sendRedirect("/hello2");
        }

        return true;
    }
}

 

*****************

注册拦截器

 

WebConfig

@Configuration
public class WebConfig implements WebMvcConfigurer {

    @Autowired
    private CustomInterceptor customInterceptor;

    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(customInterceptor).addPathPatterns("/hello");
    }
}

 

*****************

controller 层

 

HelloController

@RestController
public class HelloController {

    @RequestMapping("/hello")
    public String hello(HttpServletRequest request){

        return "success";
    }

    @RequestMapping("/hello2")
    public String error(){
        return "请稍后重试哟";
    }
}

 

 

*****************

使用测试

 

1分钟访问五次后，重定向到"/hello2"路径