一、流程
- 定时任务线程按照100/ms往redis的set结构中添加令牌100个,1s添加10次共计1000个令牌
- 当桶中满了后丢弃增加的令牌
- 令牌桶的大小设置1000,也就是每秒控制并发1000个
- 令牌可以用数字表示
- 首先 scard命令判断令牌桶当前的大小,然后添加令牌用sadd命令,两个操作必须用lua脚本保证原子性
- 请求每次都从set中获取令牌
- 获取到令牌请求通过
- 获取不到阻塞、重试、拒绝
- 获取1个令牌就从set中删除几个令牌,用spop命令
二、缺点
- 需要定时任务维护桶中的令牌,增加了复杂度
- 突然到来的尖峰流量会让并发最大到2倍也就是2000(假如当前桶是满的,突发流量100ms内取走了1000个令牌,然后继续等令牌桶产生令牌,等到1s结束后,添加令牌9次共产生900个令牌,加上第一次取走的1000个令牌,总的1s达到1900个令牌,也就是接近2倍窗口的并发量)
三、思考
- 令牌桶可以用其它组件代替:MySQL,Zookeeper,MQ等,但是都没redis简单高效
- 往令牌桶添加令牌的动作是另外一个线程,这个操作能不能做成多线程保证可用性
- 往令牌桶添加令牌的动作再进一步优化,能不能不需要线程就可以 详见:分布式限流-【3】- Redis-触发式-令牌桶_泰山与水-CSDN博客
- 往令牌桶添加令牌的动作可以增加动态调整功能,运维人员可以手动调整添加的速率,或者按照某种策略实现令牌的添加