两种白名单限流方案（redis lua限流，guava方案）

百里东君~

已于 2022-07-31 17:05:37 修改

阅读量671

点赞数 1

分类专栏：技术贴 java 文章标签： lua redis java

于 2022-07-31 17:03:18 首次发布

本文链接：https://blog.csdn.net/yuec1998/article/details/126087630

版权

两种白名单限流方案

1、redis lua方案

创建注解类

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

    // 资源名称，用于描述接口功能
    String name() default "";

    // 资源 key
    String key() default "";

    // key 前缀
    String prefix() default "";

    // 时间单位秒
    int period();

    // 限制单位时间访问的次数
    int count();

    // 限制类型
    LimitType limitType() default LimitType.CUSTOMER;
}

编写AOP拦截类

@Slf4j
@Aspect
@Component
public class LimitAspect {
   

    @Autowired
    @Qualifier("jacksonRedisTemplate")
    private RedisTemplate<String,Object> redisTemplate;


    @Pointcut("@annotation(com.xx.common.annotation.Limit)")
    public void pointcut() {
   
    }

    @Around("pointcut()")
    public Object around(ProceedingJoinPoint point) throws Throwable {
   
        HttpServletRequest request = ((ServletRequestAttributes) Objects.requireNonNull(RequestContextHolder.getRequestAttributes())).getRequest();

        MethodSignature signature = (MethodSignature) point.getSignature();
        Method method = signature.getMethod();
        Limit limitAnnotation = method.getAnnotation(Limit.class);
        LimitType limitType = limitAnnotation.limitType();
        String name = limitAnnotation.name();
        String key;
        String ip = IpUtil.getIpAddr(request);
        int limitPeriod = limitAnnotation.period