分布式限流,依赖redis
实现1个按秒限流的限流器,
知识点:自定义注解,切面,注解的使用
源码
1.创建自定义注解 RateLimit
首先,我们定义一个自定义注解 RateLimit,它包含 code 和 limit 属性。
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 {
String code();
int limit();
}
2.创建切面处理注解逻辑
接下来,我们使用 Spring AOP 来创建一个切面,处理 RateLimit 注解的逻辑。在这个示例中,我们假设你使用的是 Spring 框架。
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.Around;
import org.aspectj.lang.annotation.Aspect;
import org.springframework.stereotype.Component;
import redis.clients.jedis.Jedis;
@Aspect
@Component
public class RateLimitAspect {
private Jedis jedis;
public RateLimitAspect() {
// 初始化 Jedis 客户端
this.jedis = new Jedis("localhost", 6379); // 请根据你的 Redis 配置进行调整
}
@Around("@annotation(rateLimit)")
public Object around(ProceedingJoinPoint joinPoint, RateLimit rateLimit) throws Throwable {
String key = rateLimit.code();
int limit = rateLimit.limit();
long currentCount = jedis.incr(key);
if (currentCount == 1) {
// 设置过期时间为 1 秒
jedis.expire(key, 1);
}
if (currentCount > limit) {
// 超过限流限制
throw new RuntimeException("Rate limit exceeded");
}
return joinPoint.proceed();
}
}
3.使用 RateLimit 注解
现在你可以在你的控制器或服务方法上使用 RateLimit 注解来进行限流。
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
@RestController
@RequestMapping("/api")
public class MyController {
@RateLimit(code = "my-api", limit = 5)
@GetMapping("/limited")
public String limitedEndpoint() {
return "This endpoint is rate limited!";
}
}
4.配置 Spring 应用
确保你的 Spring 应用配置了 AOP 支持。你可以在 Spring Boot 应用中通过在主类上添加 @EnableAspectJAutoProxy 注解来启用 AOP。
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.context.annotation.EnableAspectJAutoProxy;
@SpringBootApplication
@EnableAspectJAutoProxy
public class RateLimitApplication {
public static void main(String[] args) {
SpringApplication.run(RateLimitApplication.class, args);
}
}
5. 依赖项
确保你的 pom.xml 文件中包含必要的依赖项:
<dependencies>
<!-- Spring Boot Starter Web -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<!-- Spring Boot Starter AOP -->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-aop</artifactId>
</dependency>
<!-- Jedis -->
<dependency>
<groupId>redis.clients</groupId>
<artifactId>jedis</artifactId>
<version>4.2.3</version>
</dependency>
</dependencies>
总结
标签:springboot,redis,springframework,RateLimit,限流,org,import,annotation From: https://www.cnblogs.com/PythonOrg/p/18394431通过以上步骤,你已经实现了一个自定义注解 RateLimit,并使用 Jedis 实现了 Redis 分布式限流功能。这个示例展示了如何使用 Spring AOP 来处理注解逻辑,并将限流逻辑应用到特定的控制器方法上。你可以根据需要进一步扩展和优化这个实现。