go-time/rate限流

• https://pkg.go.dev/golang.org/x/time@v0.8.0/rate

1 调用次数限制

1.1 结构

type Sometimes struct {
    First    int           // if non-zero, the first N calls to Do will run f.
    Every    int           // if non-zero, every Nth call to Do will run f.
    Interval time.Duration // if non-zero and Interval has elapsed since f's last run, Do will run f.

    mu    sync.Mutex
    count int       // number of Do calls
    last  time.Time // last time f was run
}

• 五种模式：
  • First：头N次调用f()成功，后续失败。
  • Every：每N次中的第一次调用f()成功，其它的失败。
  • Interval：在周期t内，第一次调用f()成功，其它的失败。
  • Mix：上述三种都有配置，任意一个命中，均会成功调用f()。
  • None：上述三种都不配置，只能调一次f()。
• 并发安全，由Sometimes.mu保证。

1.2 调用

st := Sometimes{
    First: 2,
    Every: 2,
    Interval: 1 * time.Second,
}
st.Do(fn(){})

2 令牌数限制

2.1 结构

type Limiter struct {
    mu     sync.Mutex
    limit  Limit
    burst  int
    tokens float64
    // last is the last time the limiter's tokens field was updated
    last time.Time
    // lastEvent is the latest time of a rate-limited event (past or future)
    lastEvent time.Time
}

type Reservation struct {
    // contains filtered or unexported fields
}

• 并发数量控制：由burst限制最大并发数，即并发时发放的令牌数；tokens为初始令牌数，tokens = burst使初始完毕时能处理burst个令牌。
• 并发安全：由Limiter.mu保证。
• 两种处理方式：阻塞等待WaitN()；丢弃AllowN()；注意申请令牌数N不能超过burst。
• Reservation预约令牌：表示在t时间之后，需要n个令牌去执行操作Act()；用于需要精细化使用令牌的场景、提前计算是否有足够令牌满足后续Act()条件。

2.2 调用

// 第一个参数：生成一个令牌所需的时间；后续计算令牌数
// 第二个参数：初始令牌数/并发令牌发放数
l := rate.NewLimiter(rate.Every(4*time.Second), 2)

// 预约令牌
r := l.ReserveN(time.Now(), 2)