Sentinel系列之令牌桶代码分析

发布时间 2023-10-22 09:17:57作者: kingsleylam
  • 原理

    令牌桶(Token Bucket)限流算法,系统会以一个恒定的速度往桶里放入令牌,请求在处理前需要先从桶里获取一个令牌。如果桶里没有令牌可取,则拒绝服务。

    image

  • 代码实现

    热点参数限流是基于令牌桶算法实现的。

    这个模式的难点在于,如何一边往令牌桶投放令牌,一边获取令牌。一般会想到使用两个线程,但Sentinel的实现方式是让请求来触发投放令牌的动作,按周期投放令牌。

    代码在com.alibaba.csp.sentinel.slots.block.flow.param.ParamFlowChecker,具体流程如下:
    image