java 一个注解实现限流

简介: java 一个注解实现限流

在Java中,可以使用自定义注解来实现限流功能。

```java
import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;
@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.METHOD)
public @interface RateLimit {
    int value() default 1;
}
```


上面的代码定义了一个名为`RateLimit`的注解,通过给方法添加这个注解,可以限制方法的访问频率。


然后,创建一个切面类来拦截被`RateLimit`注解标注的方法:

```java
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.TimeUnit;
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
@Aspect
public class RateLimitAspect {
    private ConcurrentHashMap<String, Integer> counterMap = new ConcurrentHashMap<>();
    @Around("@annotation(rateLimitAnnotation)")
    public Object rateLimit(ProceedingJoinPoint joinPoint, RateLimit rateLimitAnnotation) throws Throwable {
        String methodName = joinPoint.getSignature().getName();
        // 检查方法名对应的计数器是否存在
        counterMap.putIfAbsent(methodName, 0);
        int counter = counterMap.get(methodName);
        int limit = rateLimitAnnotation.value();
        if (counter >= limit) {
            throw new RuntimeException("Rate limit exceeded");
        }
        // 执行目标方法
        Object result = joinPoint.proceed();
        // 计数器加1
        counterMap.put(methodName, counter + 1);
        return result;
    }
}
```


`RateLimitAspect`类是一个切面类,通过`@Around`注解实现对被`RateLimit`注解标注的方法的拦截。


最后,使用这个注解和切面示例:

```java
public class Main {
    @RateLimit(5) // 每秒限制调用次数为5次
    public void doSomething() {
        System.out.println("Doing something...");
    }
    public static void main(String[] args) {
        // 创建切面
        RateLimitAspect aspect = new RateLimitAspect();
        // 创建代理对象
        Main mainObj = (Main) aspect.aspectOf(Main.class);
        // 调用被限流的方法
        for (int i = 0; i < 10; i++) {
            try {
                mainObj.doSomething();
            } catch (RuntimeException e) {
                System.out.println("Rate limit exceeded");
            }
            // 每次调用后暂停一秒钟
            try {
                TimeUnit.SECONDS.sleep(1);
            } catch (InterruptedException e) {
                e.printStackTrace();
            }
        }
    }
}
```


上面的例子中,`Main`类中的`doSomething`方法被`@RateLimit(5)`注解标注,表示该方法在1秒内最多只能被调用5次。通过切面拦截,如果超过次数限制则会抛出`RuntimeException`异常。


在`main`方法中,我们创建了`RateLimitAspect`切面实例,并通过`aspectOf`方法获得代理对象。然后,我们调用被限流的`doSomething`方法,可以看到在一秒内只能调用5次,超出限制后会打印"Rate limit exceeded"。


这是一个简单的基于注解的限流实现,你可以根据自己的需求进行扩展和适应。

目录
相关文章
|
1月前
|
XML Java 编译器
Java注解的底层源码剖析与技术认识
Java注解(Annotation)是Java 5引入的一种新特性,它提供了一种在代码中添加元数据(Metadata)的方式。注解本身并不是代码的一部分,它们不会直接影响代码的执行,但可以在编译、类加载和运行时被读取和处理。注解为开发者提供了一种以非侵入性的方式为代码提供额外信息的手段,这些信息可以用于生成文档、编译时检查、运行时处理等。
65 7
|
13天前
|
NoSQL 算法 Java
Java Redis多限流
通过本文的介绍,我们详细讲解了如何在Java中使用Redis实现三种不同的限流策略:固定窗口限流、滑动窗口限流和令牌桶算法。每种限流策略都有其适用的场景和特点,根据具体需求选择合适的限流策略可以有效保护系统资源和提高服务的稳定性。
37 18
|
3月前
|
XML Java 编译器
Java学习十六—掌握注解:让编程更简单
Java 注解(Annotation)是一种特殊的语法结构,可以在代码中嵌入元数据。它们不直接影响代码的运行,但可以通过工具和框架提供额外的信息,帮助在编译、部署或运行时进行处理。
111 43
Java学习十六—掌握注解:让编程更简单
|
1月前
|
Java 编译器 数据库
Java 中的注解(Annotations):代码中的 “元数据” 魔法
Java注解是代码中的“元数据”标签,不直接参与业务逻辑,但在编译或运行时提供重要信息。本文介绍了注解的基础语法、内置注解的应用场景,以及如何自定义注解和结合AOP技术实现方法执行日志记录,展示了注解在提升代码质量、简化开发流程和增强程序功能方面的强大作用。
87 5
|
2月前
|
Java 开发者 Spring
[Java]自定义注解
本文介绍了Java中的四个元注解(@Target、@Retention、@Documented、@Inherited)及其使用方法,并详细讲解了自定义注解的定义和使用细节。文章还提到了Spring框架中的@AliasFor注解,通过示例帮助读者更好地理解和应用这些注解。文中强调了注解的生命周期、继承性和文档化特性,适合初学者和进阶开发者参考。
78 14
|
2月前
|
Java 编译器
Java进阶之标准注解
Java进阶之标准注解
46 0
|
3月前
|
JSON Java 数据库
java 常用注解大全、注解笔记
关于Java常用注解的大全和笔记,涵盖了实体类、JSON处理、HTTP请求映射等多个方面的注解使用。
58 0
java 常用注解大全、注解笔记
|
4月前
|
Java 编译器 程序员
Java注解,元注解,自定义注解的使用
本文讲解了Java中注解的概念和作用,包括基本注解的用法(@Override, @Deprecated, @SuppressWarnings, @SafeVarargs, @FunctionalInterface),Java提供的元注解(@Retention, @Target, @Documented, @Inherited),以及如何自定义注解并通过反射获取注解信息。
Java注解,元注解,自定义注解的使用
|
4月前
|
算法 Java
Java项目不使用框架如何实现限流?
Java项目不使用框架如何实现限流?
45 2
|
3月前
|
IDE Java 编译器
java的反射与注解
java的反射与注解
31 0