自定义ip限流注解

最新推荐文章于 2024-07-23 07:45:00 发布

阳十三

最新推荐文章于 2024-07-23 07:45:00 发布

阅读量421

点赞数

分类专栏： java

本文为wdful原创文章，未经wdful允许不得转载。

本文链接：https://blog.youkuaiyun.com/qq_28114645/article/details/106620275

版权

一:定义注解

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

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

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

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

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

    // 限制访问次数
    int count();

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

}

二:注解实现

限流枚举

public enum LimitType {
    // 默认
    CUSTOMER,
    //  by ip addr
    IP;
}

@Aspect
@Component
public class LimitAspect {

    private final RedisTemplate<Object,Object> redisTemplate;
    private static final Logger logger = LoggerFactory.getLogger(LimitAspect.class);

    public LimitAspect(RedisTemplate<Object,Object> redisTemplate) {
        this.redisTemplate = redisTemplate;