前边我们学习了阿里的限流工具sentinel。她是有一个@SentinelResource注解可以使用的,但是呢,使用@SentinelResource注解需要链接sentinel控制台,在控制台中创建对应的规则。
再在对应的方法中使用@SentinelResource注解来配置功能。
但是呢,我这里有一个小小的尴尬,我目前使用的是springboot项目,而非springCloud,也可能是我配置的问题,连接sentinel控制台始终没有成功。
所以我就没有办法使用@SentinelResource注解了,但是我还不想使用其他方式来对方法进行限流,那怎么办呢?
很简单,我们之前学过自定义注解这个东西啊。
一:定义一个注解
java">package com.modules.customannotations.myAnnotation;import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;@Target(ElementType.METHOD)
@Retention(RetentionPolicy.RUNTIME)
public @interface MySentinelResource
{// 可以定义一些属性,如果不需要,可以留空String resource();int number() default 1;// 这个可以不传参数
}
二:定义注解对应的切面类
java">package com.modules.customannotations.annotationAspect;import com.alibaba.csp.sentinel.Entry;
import com.alibaba.csp.sentinel.SphU;
import com.alibaba.csp.sentinel.slots.block.RuleConstant;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRule;
import com.alibaba.csp.sentinel.slots.block.flow.FlowRuleManager;
import com.modules.customannotations.myAnnotation.MySentinelResource;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;import java.util.ArrayList;
import java.util.List;@Aspect // 声明该类为一个注解类;
@Component
public class MySentinelResourceAspect
{private final static Logger logger = LoggerFactory.getLogger(MyCustomAnnotationAspect.MyCustomAspect.class);// 自定义注解参数/*** 资源名称*/private String resource;/*** 限流数量*/private int number;// 以自定义 @MySentinelResource 注解为切点 --- @annotation里配置的 @MySentinelResource的自定义注解的全路径名@Pointcut("@annotation(com.modules.customannotations.myAnnotation.MySentinelResource)")public void mySentinelResourcePointcut() {}/*** 在切点之前,织入相关代码;* @param joinPoint* @param mySentinelResource*/@Before("@annotation(mySentinelResource)")public void beforeMethod(JoinPoint joinPoint, MySentinelResource mySentinelResource) throws Exception{Entry ignored = null;try{// 获取注解的参数this.resource = mySentinelResource.resource();this.number = mySentinelResource.number();/* 1.创建存放限流规则的集合 */List<FlowRule> rules = new ArrayList<>();/* 2.创建限流规则 */FlowRule rule = new FlowRule();/* 定义资源,表示 Sentinel 会对哪个资源生效 "AddUser" */rule.setResource(resource);/* 定义限流的类型(此处使用 QPS 作为限流类型) */rule.setGrade(RuleConstant.FLOW_GRADE_QPS);/* 定义 QPS 每秒通过的请求数 1 */rule.setCount(number);/* 3.将限流规则存放到集合中 */rules.add(rule);/* 4.加载限流规则 */FlowRuleManager.loadRules(rules);// 设置一个资源名称为 Helloignored = SphU.entry(resource);}catch(Exception e){ // 被限流之后就会进入到这里来// 方法执行前的逻辑,终止程序执行可以通过抛出异常或其他方式实现throw new RuntimeException("别急,等一会在请求!");}finally{ // 兜底方法,销毁资源// 销毁资源if (ignored != null){ignored.exit();}}}/*** 环绕,可以在切入点前后织入代码,并且可以自由的控制何时执行切点;* @param point* @return* @throws Throwable*/@Around("mySentinelResourcePointcut()")private Object testAop(ProceedingJoinPoint point) throws Throwable{// 获取方法返回的数据Object obj = point.proceed();//System.out.println("obj:"+obj);return obj;}
}
这里我在做的时候钻牛角尖了,我在@Before注解定义的方法中对资源进行限流,但是呢,有一个小问题,这个方法是没有返回值的,也不能终止程序,这可怎么办呢?
后来我一寻思,这不傻了吗,我可以抛出异常啊,抛出异常程序不就停止了吗。
在结合我前边定义的全局异常处理类,直接这不就圆满了吗。
调用:
java">@MySentinelResource(resource = "getData", number = 1) // 自定义注解:sentinel限流
public Map<String, Object> getData(@RequestParam(defaultValue = "1") Integer page, @RequestParam(defaultValue = "") String search)
{Map<String, Object> result = indexService.getData(page, search);
}
到这里,使用sentinel自定义注解对请求限流就完成了。
有好的建议,请在下方输入你的评论。