java自定义注解，通过aop实现限制频繁调用接口功能

java通过aop实现限制频繁调用接口功能

---

1、引入pom依赖

        <!--        aop依赖-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-aop</artifactId>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>org.aspectj</groupId>
            <artifactId>aspectjweaver</artifactId>
        </dependency>
        <!--        redis依赖-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-redis</artifactId>
        </dependency>

2、创建自定义注解类，RequestLimit.java

import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;

import java.lang.annotation.*;

@Retention(RetentionPolicy.RUNTIME)     // 注解会在class字节码文件中存在，在运行时可以通过反射获取到
@Target({ElementType.TYPE, ElementType.METHOD})     //在类和方法上都能引用
@Documented     //如果使用@Documented标注了，在生成javadoc的时候就会把@Documented注解给显示出来。此处可以不引入
@Order(Ordered.HIGHEST_PRECEDENCE)      //最高优先级
public @interface RequestLimit {

    /**
     *
     * 时间段，单位为毫秒，默认值5秒
     */
    long time() default 5000;

}

3.创建NoRepeatSubmitAop.java，实现接口的拦截并做处理

import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.concurrent.TimeUnit;

@Aspect
@Component
@Slf4j
public class NoRepeatSubmitAop {


    @Resource
    public RedisTemplate redisTemplate;

    //@Synchronized 作用是创建一个互斥锁，保证此时没有其它线程对self对象进行修改。这个是objective-c的一个锁定令牌，防止self对象在同一时间内被其它线程访问，起到线程的保护作用。 一般在公用变量的时候使用，如单例模式或者操作类的static变量中使用。
//    @Synchronized
    @Around("execution(* com.example.mongodbdemo.controller..*.*(..)) && @annotation(nrs)")
    public Object arround(ProceedingJoinPoint pjp, RequestLimit nrs) throws Throwable {

        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = attributes.getRequest();
        Object[] args = pjp.getArgs();  //获取方法参数
        String key = "repeat" + "：" + request.getServletPath();
        if (!redisTemplate.hasKey(key)) {// 如果缓存中没有这个url就执行方法
            Object o = pjp.proceed();
            Long time = nrs.time();     //获取RequestLimit注解中参数设置的值
            System.out.println("限制 "+time+"毫秒内只能访问一次，能正常访问");
            redisTemplate.opsForValue().set(key, 0, time.intValue(), TimeUnit.MILLISECONDS);
            return o;
        } else {    // 如果缓存中有这个url视为重复提交
            System.out.println("请勿重复提交或者操作过于频繁！");
            return "请勿重复提交或者操作过于频繁！";
        }

    }
}

2.创建TestController.java

import com.example.mongodbdemo.interceptor.RequestLimit;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

@Slf4j
@RestController
@RequestMapping("/test")
public class TestController {

    @GetMapping(value = {"/testRepeat"})
    @RequestLimit(time = 1000)
    public String testRepeat(@RequestParam("id") String id) {
        System.out.println("调用testRepeat请求");
        return "ok";
    }

}

---

4.application.yml配置

server:
  port: 8080
spring:
  redis:
    host: 127.0.0.1
    port: 6379

5、访问地址，查看结果

类都创建完成后，启动项目，访问：127.0.0.1:8080/test/testRepeat?id=666 即可，查看控制台输出结果。