使用令牌桶算法通过redis实现限流

令牌桶算法是一种常用的限流算法,它可以平滑地控制请求的处理速率。在 Java 中结合 Redis 实现令牌桶算法,可以利用 Redis 的原子操作来保证多节点环境下的限流效果。

一 实现思路

  1. 初始化令牌桶:在 Redis 中存储令牌桶的相关信息,如令牌数量、令牌桶容量、上一次更新时间等。
  2. 添加令牌:根据时间间隔,计算需要添加到令牌桶中的令牌数量,并更新令牌桶的令牌数量。
  3. 获取令牌:尝试从令牌桶中获取指定数量的令牌,如果令牌数量足够,则返回成功;否则返回失败。

二 Redis Lua 脚本

编写一个 Lua 脚本来处理令牌桶的逻辑。lua脚本会在 Redis 执行时保证原子性。

Lua 复制代码
-- 最大令牌个数(qps)
local limit = tonumber(ARGV[1]) 
-- 令牌生成速率
local rate = tonumber(ARGV[2]) 
-- 当前时间秒*1000=毫秒
local now = redis.call('TIME')[1] * 1000
-- 剩余可用的令牌个数
local tokens = tonumber(redis.call('GET', KEYS[1]) or 0)
-- 上次访问时间戳(单位毫秒)
local lastTimestamp = tonumber(redis.call('GET', KEYS[2]) or 0)
-- 本次访问时间戳-上次访问时间戳
local elapsedTime = now - lastTimestamp
-- 计算本次访问与上次访问之间需要放入的令牌个数=上次剩余的令牌数+本次访问与上次访问之间差的令牌数
local newTokens = tokens + elapsedTime * rate / 1000
-- 如果需要放入的令牌个数超过限制的最大令牌个数,则直接使用最大令牌个数
if newTokens > limit then
	newTokens = limit 
end 
-- 如果需要放入的令牌个数小于1,则意味着无法获取到锁
if newTokens < 1 then
	return 0 
else
-- 获取到锁,则更新剩余可获取的锁个数-1,更新本次访问时间戳
	redis.call('SET', KEYS[1], newTokens - 1)
	redis.call('SET', KEYS[2], now)
	return 1 
end

三 Java 实现

java 复制代码
public class RedisRateLimiter {

  private JedisPool jedisPool;

  private String LUA_SCRIPT = "上面的lua脚本替换到此处";

  public RedisRateLimiter(JedisPool jedisPool) {
    this.jedisPool = jedisPool;
  }

  public boolean tryAcquire(String key, int rate, int capacity) {
        try (Jedis jedis = jedisPool.getResource()) {
      Object result = jedis.eval(LUA_SCRIPT,
          Arrays.asList(key, key + ":timestamp"), Arrays.asList(capacity, rate));
      long response = (Long) result;
      return response != 0;
    } catch (Exception e) {
        return false;
    }
  }
}
相关推荐
ohMyGod_12340 分钟前
React16,17,18,19新特性更新对比
前端·javascript·react.js
@大迁世界42 分钟前
第1章 React组件开发基础
前端·javascript·react.js·前端框架·ecmascript
Hilaku1 小时前
从一个实战项目,看懂 `new DataTransfer()` 的三大妙用
前端·javascript·jquery
爱分享的程序员1 小时前
前端面试专栏-算法篇:20. 贪心算法与动态规划入门
前端·javascript·node.js
我想说一句1 小时前
事件委托与合成事件:前端性能优化的"偷懒"艺术
前端·javascript
爱泡脚的鸡腿1 小时前
Web第二次笔记
前端·javascript
Dream耀1 小时前
React合成事件揭秘:高效事件处理的幕后机制
前端·javascript
洋流1 小时前
0基础进大厂,第12天:ES6语法基础篇
javascript
I'm写代码1 小时前
el-tree树形结构笔记
javascript·vue.js·笔记
到底起什么网名才能不重名1 小时前
使用各种CSS美化网页
前端·css·vscode·bootstrap·html