整体思路:

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

二 容器初始化的时候扫描所有所有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. Excel读取方式优化(浅谈对规律的认知)

    相信大家都接触过对Excel的读取,今天突发奇想,想将自己的一小段经历分享出来.灵活识别列名并将其存到对象数组中. 固定形式的Excel列的读取: 源于我第一次操作Excel,将列名对应成table中 ...

  2. tomcat 配置项目前缀(推荐方式四)

    一. 显示配置 Context 的 path   需要在 server.xml 文件中手动配置. <Host name="localhost" appBase="w ...

  3. ACID特性及幻读的理解

    事务是关系型数据库的重要特性.它是一个包含了一条或多条SQL语句的逻辑原子单元.一个事务包含的SQL要么全部提交,要么全部回滚. Oracle 官方文档中对事务的描述如下: A transaction ...

  4. schedule of 2016-09-26~2016-10-02(Monday~Sunday)——1st semester of 2nd Grade

    2016/9/26 Monday 1.make ppt for this afternoon's group meeting 2.ask teacher Xiqi&Liu some probl ...

  5. spring boot学习笔记(1)

    一.构建springboot项目 可以通过idea的springboot initialization来创建, idea的功能很强大,可以自己选择需要集成的插件. 完整的项目结构是这样的: DemoA ...

  6. MySQL之性能优化

    查看执行计划explain     1.1 Explain命令:它可以对select语句进行分析,并输出select执行的详细信息,以对开发人员针对性优化  1.2 Explain的用法:在selec ...

  7. 【Think In Java笔记】第1章 对象导论

    1. 对象导论 OOP 面向对象编程 C.Basic等语言所在的抽象仍要求在解决问题时基于计算机的解决,而不是基于所解决问题的结构来考虑. 要建立起问题空间的元素和解空间的对象之间一一映射的关系 万物 ...

  8. lldb调试器知多少

    lldb调试器简介   lldb 是一个有着 REPL 的特性和 C++ .Python 插件的开源调试器.lldb调试器的由来是伴随着Xcode的版本升级而来. Xcode4.3之前使用的默认调试器 ...

  9. Spring Boot2 系列教程(一) | 如何使用 IDEA 构建 Spring Boot 工程

    微信公众号:一个优秀的废人 如有问题或建议,请后台留言,我会尽力解决你的问题. Search 前言 新年立了个 flag,好好运营这个公众号.具体来说,就是每周要写两篇文章在这个号发表.刚立的 fla ...

  10. 搞定SpringBoot多数据源(2):动态数据源

    目录 1. 引言 2. 动态数据源流程说明 3. 实现动态数据源 3.1 说明及数据源配置 3.1.1 包结构说明 3.1.2 数据库连接信息配置 3.1.3 数据源配置 3.2 动态数据源设置 3. ...