从0到1实现rpc之限流
12 Apr 2024 |- 在服务提供者provider端添加限流逻辑
限流:指定时间内请求数超过指定阈值时就抛出异常。
在ProviderInvoker的调用过程中,添加限流逻辑:
- 使用滑动窗口SlidingTimeWindow统计30s的请求数;
- 每个服务service对应一个滑动窗口;
- 对限流模块使用同步锁;
- 判断请求数是否超过阈值,如果是则抛出异常;
- 记录请求次数。
- 添加测试案例
循环请求指定服务,每秒请求一次,限流配置参数为30s内超过20个请求就会被限流。
观察日志,在30s内连续请求20次后第21次的请求就被限流了。
源码:
https://github.com/midnight2104/midnight-rpc/tree/lesson12