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 即可,查看控制台输出结果。