令牌桶限流算法实现
大家好，我是程序员小李。今天在面试中，我被问到一个经典问题：如何实现令牌桶限流算法？这在高并发系统里很常见，比如保护API不被刷爆。咱们就以口述方式聊聊这个实现，就像在面试现场一样。先分享个福利：2025年Java面试宝典，网盘地址：https://pan.baidu.com/s/1RUVf75gmDVsg8MQp4yRChg?pwd=9b3g 提取码: 9b3g（蓝色字体）。更多资源，可以访问面试鸭返利网获取。
什么是令牌桶限流算法？
令牌桶限流算法是一种流量控制机制，核心是模拟一个桶里放令牌。想象一下，桶里定期生成令牌，每个请求来的时候，必须拿到一个令牌才能执行，否则就被拒绝。这能防止突发流量打垮系统。比如，在微服务架构里，用它保护数据库查询。令牌桶限流算法的优势是灵活：它允许短时间内的突发请求，但长期看，平均速率是可控的。面试官常问这个，因为它比固定窗口算法更平滑。
上图展示了令牌桶的基本结构，桶里的令牌代表可用的请求配额。
令牌桶限流算法的原理
令牌桶限流算法的原理很简单：桶有固定容量，比如100个令牌。系统每秒向桶里添加一定数量的令牌（比如10个）。当请求到达时，如果桶里有令牌，就取走一个并放行请求；如果桶空了，请求就被限流拒绝。这实现的关键是“令牌生成”和“令牌消耗”的平衡。令牌桶限流算法能处理突发，是因为桶里积累的令牌可以一次性用掉。比如，如果桶满时来了50个请求，它们能全通过；但之后如果令牌生成慢，后续请求就得排队。
在分布式系统中，令牌桶限流算法常结合Redis或Guava库实现。面试时，我会强调它的适用场景：API网关、消息队列限流等。令牌桶限流算法避免了“饥饿”问题，因为令牌是按时间均匀添加的。
如何实现令牌桶限流算法？
实现令牌桶限流算法，分几步走。第一，初始化桶：设置桶容量和令牌添加速率。比如，容量100，每秒加10个令牌。第二，处理请求：每次请求来时，检查当前桶里的令牌数。如果有令牌，就减1并放行；如果没有，就拒绝或等待。第三，定时添加令牌：用一个后台线程或定时任务，每秒向桶里加指定数量的令牌，但不能超过容量上限。
令牌桶限流算法的实现要注意线程安全，因为多个请求可能同时操作桶。我常用原子类或锁来同步。比如，在Java里，AtomicInteger能高效更新令牌数。令牌桶限流算法不需要复杂代码，核心是维护一个计数器。面试中，我会口述伪代码：定义一个类，有addToken和tryAcquire方法。addToken定时调用，tryAcquire在请求时检查。
这张图对比了令牌桶和其他算法，显示令牌桶如何处理突发流量。
令牌桶限流算法的优化点
令牌桶限流算法的实现可以优化。例如，用漏桶算法结合，处理更平滑的流量。或者，在分布式环境用Redis的INCR命令管理令牌，避免单点瓶颈。令牌桶限流算法还能动态调整速率：监控系统负载，自动增加或减少令牌添加速度。面试时，我提到过实际案例，比如在电商大促时用令牌桶保护支付接口。
最后，如果你在准备面试，需要购买面试鸭会员获取更多题库，可以通过面试鸭返利网找到我，返利25元，超划算！令牌桶限流算法的掌握能帮你轻松应对系统设计题。
总结令牌桶的应用
令牌桶限流算法在实战中很实用，比如限流框架如Sentinel就内置了它。面试中，我会强调它的优点：简单、高效、可扩展。令牌桶限流算法实现起来不复杂，但能显著提升系统稳定性。记住，多练习是关键，资源在开头网盘里。
优化后的令牌桶在云原生环境中的应用示意图。
希望这篇口述帮你搞定面试！访问面试鸭返利网获取更多支持。令牌桶限流算法是必考项，多练几次就熟了。
