一份电子商务网站建设规划书,松原市住房和城乡建设局网站,海南网站定制,在家做网站维护兼职文章目录 如何实现单机限流谷歌的 RateLimiter 介绍使用 AOP 实现单机限流Controller切面 如何实现单机限流
谷歌的 RateLimiter 介绍
使用谷歌的 guava 包中 RateLimiter 类来实现#xff0c;先来介绍一个这个类#xff1a;
RateLimiter 是一个频率限制器#xff0c;通过… 文章目录 如何实现单机限流谷歌的 RateLimiter 介绍使用 AOP 实现单机限流Controller切面 如何实现单机限流
谷歌的 RateLimiter 介绍
使用谷歌的 guava 包中 RateLimiter 类来实现先来介绍一个这个类
RateLimiter 是一个频率限制器通过配置频率来发放许可如果 1 秒内可以访问十次那么这十次许可发送的间隔是完全相同的并发使用是安全的RateLimiter 还可以去配置先处于一个预热器每秒增加发放的许可知道达到稳定的频率RateLimiter 不影响请求本身的节流而是影响下一次请求的节流比如当前任务如果占用许可较多到达 RateLimiter 之后会立即占用当下一个请求到达 RateLimiter 时就会经历节流因为上一个请求已经占用大量的许可。
一个小示例用法如果想要发送一组数据我们限制他在 5kb 每秒
// 给每一个字节发放 1 个许可限制在 5kb 每秒的话只需要每秒发放 5000 个许可即可
final RateLimiter rateLimiter RateLimiter.create(5000.0);
void submitPacket(byte[] packet) {rateLimiter.acquire(packet.length);networkService.send(packet);
}使用 AOP 实现单机限流
实现步骤
定义切面拦截 Controller 层方法创建一个 RateLimiter定义访问频率当执行方法时发放一个许可如果拿不到许可直接拦截 Controller 层方法的执行返回一个访问频繁的提示
Controller
RestController
RequestMapping(/rate)
public class RateController {GetMappingpublic String testRate() {return 测试 Rate ;}
}切面
Component
Aspect
public class ServiceLogAspect {/*** 对 controller 限流*/Pointcut(execution(* com.javagpt.back.controller.*.*(..)))public void rateLimitPointCut() {}private static final RateLimiter rateLimiter RateLimiter.create(10);SneakyThrows // 使用之后不需要抛出异常lombok会自动在编译时加上try/catchAround(rateLimitPointCut())public Object rateLimit(ProceedingJoinPoint joinPoint) {double rate rateLimiter.getRate();System.out.println(rate);if (rateLimiter.tryAcquire()) {return joinPoint.proceed();} else {// 如果超出限流次数拦截方法的执行注意这里返回的对象要和 Controller 方法的返回对象类型相同否则会报错return 访问太过频繁;}}}
