一、流程

  1. 定时任务线程按照100/ms往redis的set结构中添加令牌100个,1s添加10次共计1000个令牌
    1. 当桶中满了后丢弃增加的令牌
    2. 令牌桶的大小设置1000,也就是每秒控制并发1000个
    3. 令牌可以用数字表示
    4. 首先 scard命令判断令牌桶当前的大小,然后添加令牌用sadd命令,两个操作必须用lua脚本保证原子性
  2. 请求每次都从set中获取令牌
    1. 获取到令牌请求通过
    2. 获取不到阻塞、重试、拒绝
    3. 获取1个令牌就从set中删除几个令牌,用spop命令

二、缺点

  1. 需要定时任务维护桶中的令牌,增加了复杂度
  2. 突然到来的尖峰流量会让并发最大到2倍也就是2000(假如当前桶是满的,突发流量100ms内取走了1000个令牌,然后继续等令牌桶产生令牌,等到1s结束后,添加令牌9次共产生900个令牌,加上第一次取走的1000个令牌,总的1s达到1900个令牌,也就是接近2倍窗口的并发量)

三、思考

  1. 令牌桶可以用其它组件代替:MySQL,Zookeeper,MQ等,但是都没redis简单高效
  2. 往令牌桶添加令牌的动作是另外一个线程,这个操作能不能做成多线程保证可用性
  3. 往令牌桶添加令牌的动作再进一步优化,能不能不需要线程就可以 详见:分布式限流-【3】- Redis-触发式-令牌桶_泰山与水-CSDN博客
  4. 往令牌桶添加令牌的动作可以增加动态调整功能,运维人员可以手动调整添加的速率,或者按照某种策略实现令牌的添加
Logo

更多推荐