4. 令牌桶算法详解：算法原理、参数配置与催收API适用场景

令牌桶算法是催收触达系统中应用最广泛的限流算法之一。它通过一个“桶”来平滑突发流量，既允许短时间内的请求爆发，又能保证长期的平均速率，非常适合催收系统中“白天高并发、夜间低负载”的业务特征。

4.1 算法原理

令牌桶的核心思想是：系统以恒定速率向一个桶中放入令牌，每个请求必须从桶中获取一个令牌才能被处理。如果桶中没有令牌，请求将被拒绝或排队等待。

4.2 参数配置与计算

令牌桶有两个核心参数，它们的配置直接决定了限流效果：

4.3 在催收API中的适用场景

催收系统的流量模式具有明显的“潮汐效应”，令牌桶算法在以下场景中表现优异：

• 批量催收任务启动：每天 9:00、14:00、20:00 等固定时间点，系统会批量发送催收通知。令牌桶的容量参数允许这些突发请求快速通过，而速率参数保证后续请求平滑。
• 短信/外呼网关保护：第三方短信或外呼平台通常有 QPS 限制。令牌桶可以精确匹配上游速率，避免因超限导致封禁或额外费用。
• 多级限流配合：在 API 网关层使用令牌桶做全局限流，在业务层使用滑动窗口做精细化控制，形成双层防护。
• 夜间低负载时段：催收系统夜间请求量极低，令牌桶会自动积累令牌，为次日早高峰做好准备。

4.4 代码实现

以下是一个基于 Go 语言的令牌桶实现，可直接嵌入催收 API 网关中间件：

package limiter

import (
    "sync"
    "time"
)

// TokenBucket 令牌桶限流器
type TokenBucket struct {
    rate       float64       // 令牌生成速率（个/秒）
    capacity   float64       // 桶容量
    tokens     float64       // 当前令牌数
    lastTime   time.Time     // 上次更新令牌的时间
    mu         sync.Mutex
}

// NewTokenBucket 创建一个新的令牌桶
func NewTokenBucket(rate, capacity float64) *TokenBucket {
    return &TokenBucket{
        rate:     rate,
        capacity: capacity,
        tokens:   capacity, // 初始时桶是满的
        lastTime: time.Now(),
    }
}

// Allow 判断是否允许一个请求通过
func (tb *TokenBucket) Allow() bool {
    tb.mu.Lock()
    defer tb.mu.Unlock()

    now := time.Now()
    // 1. 计算时间差，补充令牌
    elapsed := now.Sub(tb.lastTime).Seconds()
    tb.tokens += elapsed * tb.rate
    if tb.tokens > tb.capacity {
        tb.tokens = tb.capacity
    }
    tb.lastTime = now

    // 2. 判断是否有令牌可用
    if tb.tokens >= 1 {
        tb.tokens--
        return true
    }
    return false
}

// AllowN 判断是否允许 N 个请求通过（批量判断）
func (tb *TokenBucket) AllowN(n int) bool {
    tb.mu.Lock()
    defer tb.mu.Unlock()

    now := time.Now()
    elapsed := now.Sub(tb.lastTime).Seconds()
    tb.tokens += elapsed * tb.rate
    if tb.tokens > tb.capacity {
        tb.tokens = tb.capacity
    }
    tb.lastTime = now

    if tb.tokens >= float64(n) {
        tb.tokens -= float64(n)
        return true
    }
    return false
}

// GetTokenCount 获取当前令牌数（用于监控）
func (tb *TokenBucket) GetTokenCount() float64 {
    tb.mu.Lock()
    defer tb.mu.Unlock()
    return tb.tokens
}

4.5 催收场景配置示例

假设某催收系统对接的短信网关限制为：每秒最多 200 条，且允许短时间突发 500 条。则令牌桶配置如下：

从图中可以看出，在 09:00 批量催收任务启动时，令牌桶允许 450 个请求（超过速率线）通过，这是利用了桶中积累的令牌。随后流量被平滑到速率线以下，保证了系统的稳定性。