整体思路:

一 具体接口,可以自定义一个注解,配置限流量,然后对需要限流的方法加上注解即可!

二 容器初始化的时候扫描所有所有controller,并找出需要限流的接口方法,获取对应的限流量

三 使用拦截器或者aop,对加上注解的方法进行限流,采用配置的信号量

自定义注解

/**
* 限流注解
*/
@Target(ElementType.METHOD) //作用与方法上
@Retention(RetentionPolicy.RUNTIME) //注解不仅被保存到class文件中,jvm加载class文件之后,仍然存在
@Documented
public @interface ApiRateLimit {
int value(); //控制并发最大数量
}

  

出初始化限流配置,注意:这里设置的如果限流量一样,则两个方法一起限流,比如两个方法限流量都是5,则两个方法总共可以支持最多5个线程访问

实际可以自己调整,加个方法名当key,则可以保证每个方法都独自限流:

/**
* ApplicationContextAware实现类可以获得spring上下文
* 间接获取ApplicationContext中的所有bean,向切面添加所有接口的配置的限流量
*/
@Component
public class InitApiLimit implements ApplicationContextAware {
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
Map<String, Object> beanMap = applicationContext.getBeansWithAnnotation(RestController.class);
System.out.println(beanMap.size());
beanMap.forEach((k,v)->{
Class<?> controllerClass = v.getClass();
System.out.println(controllerClass.toString());
System.out.println(controllerClass.getSuperclass().toString());
//获取所有声明的方法
Method[] allMethods = controllerClass.getSuperclass().getDeclaredMethods();
for (Method method:allMethods){
System.out.println(method.getName());
//判断方法是否使用了限流注解
if (method.isAnnotationPresent(ApiRateLimit.class)){
//获取配置的限流量,实际值可以动态获取,配置key,根据key从配置文件获取
int value = method.getAnnotation(ApiRateLimit.class).value();
String key = String.valueOf(value);
//key作为key.value为具体限流量,传递到切面的map中
ApiLimitAspect.semaphoreMap.put(key,new Semaphore(value));
}
}
System.out.println("----信号量个数:"+ApiLimitAspect.semaphoreMap.size());
});
}
}

  注意:这里有一点需要说明,一旦使用了代理,因为是controller',没有借口,所以是cglib,会创建子类

,此时从容器中获取的是代理的子类,默认是不会有自定义注解的,所以得getSuperClass,从父类,即controller中获取注解信息

编写切面,这里是最主要的,使用jdk自带的信号量:

限流切面

/**
* 限流切面
*/
@Aspect
@Order(value = Ordered.HIGHEST_PRECEDENCE) //最高优先级
@Component
public class ApiLimitAspect {
//存储限流量和方法,必须是static且线程安全,保证所有线程进入都唯一
public static Map<String, Semaphore> semaphoreMap= new ConcurrentHashMap<>();
//拦截所有controller 的所有方法
@Around("execution(* com.hou.serviceorder.controller.*.*(..))")
public Object around(ProceedingJoinPoint joinPoint){
Object result=null;
Semaphore semaphore=null;
Class<?> clz = joinPoint.getTarget().getClass();//获取目标对象
Signature signature = joinPoint.getSignature();//获取增强方法信息
String name = signature.getName();
String limitKey = String.valueOf(getLimitKey(clz, name));
if(limitKey!=null && !"".equals(limitKey)){
semaphore = semaphoreMap.get(limitKey);
try {
semaphore.acquire();
result=joinPoint.proceed();
} catch (Throwable e) {
e.printStackTrace();
} finally {
semaphore.release();
}
}
return result;
} //获取拦截方法配置的限流key,没有返回null
private Integer getLimitKey(Class<?> clz, String methodName){
for (Method method:clz.getDeclaredMethods()){
if(method.getName().equals(methodName)){//找出目标方法
if(method.isAnnotationPresent(ApiRateLimit.class)){//判断是否是限流方法
return method.getAnnotation(ApiRateLimit.class).value();
}
}
}
return null;
}
}

  

使用注解

    @ApiRateLimit(value = 5)
@GetMapping("/name")
public String getOrderName() throws InterruptedException {
System.out.println("-----进入getOrder方法------");
TimeUnit.SECONDS.sleep(2);
return "order";
} @ApiRateLimit(value = 5)
@GetMapping("/order")
public Order getOrder(String id) throws InterruptedException {
System.out.println("-----进入getOrder方法------");
TimeUnit.SECONDS.sleep(2);
return new Order(id,"侯征");
}

  

测试

public class TestSe {

    public static void main(String[] args) {
//测试信号并发量
for (int i = 0; i < 10; i++) {
new Thread(()->{
//访问目标接口
RestTemplate restTemplate = new RestTemplate();
restTemplate.getForObject("http://localhost:8081/order/name",String.class);
}).start();
}
}
}

  会发现最多5个一起打印:

使用AOP和Semaphore对项目中具体的某一个接口进行限流的更多相关文章

  1. 转:C4项目中验证用户登录一个特性就搞定

    转:C4项目中验证用户登录一个特性就搞定   在开发过程中,需要用户登陆才能访问指定的页面这种功能,微软已经提供了这个特性.     // 摘要:    //     表示一个特性,该特性用于限制调用 ...

  2. java web项目(spring项目)中集成webservice ,实现对外开放接口

    什么是WebService?webService小示例 点此了解 下面进入正题: Javaweb项目(spring项目)中集成webservice ,实现对外开放接口步骤: 准备: 采用与spring ...

  3. 一个项目中:只能存在一个 WebMvcConfigurationSupport (静态文件失效之坑)

    一个项目中:只能存在一个 WebMvcConfigurationSupport 在一个项目中WebMvcConfigurationSupport只能存在一个,多个的时候,只有一个会生效. 静态文件访问 ...

  4. 《 .NET并发编程实战》一书中的节流为什么不翻译成限流

    有读者问,为什么< .NET并发编程实战>一书中的节流为什么不翻译成限流? 这个问题问得十分好!毕竟“限流”这个词名气很大,耳熟能详,知名度比“节流”大多了. 首先,节流的原词Thrott ...

  5. Consul+Ocelot+Polly在.NetCore中使用(.NET5)-Ocelot+Polly缓存、限流、熔断、降级

    相关文章 Consul+Ocelot+Polly在.NetCore中使用(.NET5)-Consul服务注册,服务发现 Consul+Ocelot+Polly在.NetCore中使用(.NET5)-网 ...

  6. iOS中AOP与Method Swizzling 项目中的应用

    引子:项目中需要对按钮点击事件进行统计分析,现在项目中就是在按钮的响应代码中添加点击事件,非常繁琐.所以使用了AOP(面向切面编程),将统计的业务逻辑统一抽离出来. 项目中添加的开源库:https:/ ...

  7. C#.NET常见问题(FAQ)-程序如何把窗体文件从从一个项目中复制到另一个项目

    一个窗体有三个文件,全部拷贝到新的项目中   在新的项目中点击显示所有文件,然后右击导入的文件,点击包括在项目中,会自动修改颜色(此时还没有被识别为窗体)   重启这个项目,三个文件已经被识别出来了 ...

  8. C#程序如何把窗体文件从从一个项目中复制到另一个项目

    一个窗体有三个文件,全部拷贝到新的项目中   在新的项目中点击显示所有文件,然后右击导入的文件,点击包括在项目中,会自动修改颜色(此时还没有被识别为窗体)   重启这个项目,三个文件已经被识别出来了 ...

  9. 在.net core web api项目中安装swagger展示api接口(相当于生成api文档)

    1,  建立或打开项目后,在“程序包管理器控制台”中执行以下命令添加包引用: Install-Package Swashbuckle.AspNetCore 2,在项目中打开Startup.cs文件,找 ...

随机推荐

  1. Java中的变量、数据类型和运算符

    1. java语言是一种强类型的语言,对各种数据类型都有明确的区分,而计算机使用内存来记忆大量运算时需要使用的数据,而当声明一个变量时,即在内存中划分一块空间存储数据,而变量类型决定划分内存空间的大小 ...

  2. [译文] 为什么你在 C# 里总是应该使用 "var" 关键字

    [译文] Why You Should Always Use the 'var' Keyword in C# (为什么你总是应该在 C# 里使用 "var" 关键字) Using ...

  3. 前端——BOM与DOM

    目录 前戏 window对象 window的子对象 navigator对象(了解即可) screen对象(了解即可) history对象(了解即可) location对象 弹出框 计时相关 DOM H ...

  4. Linux学习笔记(一):什么是挂载?mount的用处在哪?

    关于挂载的作用一直不是很清楚,今天在阅读教材时看见了mount这个命令,发现它的用处很隐晦但非常强大.奈何教材说的不明朗,因此在网上整合了一些优秀的解释,看完之后豁然开朗. 1.提一句Windows下 ...

  5. 01_elementUI tree 插件 去图标

    1:elementUI饿了吗前端ui框架,结合vue开发过程中,是不是对tree组件很头疼呢?是不是想自定义图标或者去掉所有图标只留末级checkbox呢? 实现很简单添加几行css代码完美搞定!!! ...

  6. 05_jquery 操作table使tr(数据)整行上移下移

    1:ajax请求数据到页面 function GetWorkSpaceList() { GetServerData("get", GetEnterpriseUrl() + &quo ...

  7. linux下安装cmake方法(1)---下载压缩包

    OpenCV 2.2以后的版本需要使用Cmake生成makefile文件,因此需要先安装cmake:还有其它一些软件都需要先安装cmake 1.在linux环境下打开网页浏览器,输入网址:http:/ ...

  8. 【转】python及其工具包安装基本流程

    昨天晚上在家里的旧电脑上安装了<利用python进行数据分析>的部分环境,遇到若干问题,在此予以记录. 部分细节转:http://blog.csdn.net/huanbia/article ...

  9. 《C++Primer》第五版习题解答--第四章【学习笔记】

    [C++Primer]第五版习题解答--第四章[学习笔记] ps:答案是个人在学习过程中书写,可能存在错漏之处,仅作参考. 作者:cosefy Date: 2020/1/11 第四章:表达式 练习4. ...

  10. 【Spark 内核】 Spark 内核解析-上

    Spark内核泛指Spark的核心运行机制,包括Spark核心组件的运行机制.Spark任务调度机制.Spark内存管理机制.Spark核心功能的运行原理等,熟练掌握Spark内核原理,能够帮助我们更 ...