分布式限流——Redis + Lua脚本实现令牌桶算法

主要思路概括如下：

1. 定义数据结构：

   • 使用Redis存储令牌桶的状态，包括当前令牌数（KEYS[1]）和上一次令牌填充的时间戳（KEYS[1]:last）。

2. 计算新增令牌：

   • 获取当前系统时间与上次令牌填充时间的时间差，并基于令牌生成速率计算在这段时间内应新增的令牌数。
   • 确保新增令牌数不超过桶的总容量。

3. 更新令牌数：

   • 将令牌桶内的令牌数增加至新的值，确保不超过桶的最大容量。

4. 判断是否满足请求：

   • 如果更新后的令牌数足以满足本次请求消耗的令牌数，则扣除相应数量的令牌并返回1，表示请求通过。
   • 否则，返回0，表示请求被限流。

5. 原子性操作：

   • 利用Redis Lua脚本提供的原子性执行能力，确保在多客户端并发访问时，令牌桶状态的读取、更新等操作是线程安全的。

6. 实时性：

   • 脚本设计考虑到随着时间推移动态补充令牌，但实际应用中可能需要更精确的定时任务或者使用Redis的过期时间特性来定期补充令牌。

整个令牌桶算法在Redis和Lua脚本中的实现旨在提供一种简单且高效的手段来进行流量控制，既可以