记录一次分布式限流器实现

最新推荐文章于 2025-05-07 10:10:33 发布

原创

最新推荐文章于 2025-05-07 10:10:33 发布 · 753 阅读

13 ·

CC 4.0 BY-SA版权

文章标签：

#java #lua #sentinel #guava

限流工具调研

常用限流方法

令牌桶：针对业务有短时间较高并发场景
漏桶：针对限定入口速率，保证下游服务安全
滑动窗口：针对不会出现极大的突增流量的场景，平滑并发情况，对于短时间的高并发（恶意调用）情况能够及时响应
计数器：有毛刺现象，不推荐，滑动窗口为其升级版

常用限流产品

Guava RateLimiter。优：简单易用缺：单机
Sentinel。优：大厂背书，社区活跃，提供分布式方案，提供控制台，支持接口配置
Spring Cloud Gateway。
Hystrix。缺：不再维护
Resilience4j。取代Hystrix
阿里云AHAS。优：简单易用，提供控制台收费

背景

Sentinel是一款很优秀的限流产品，公司内部使用也非常广泛。但由于申请Sentinel资源比较麻烦，所以先暂时自己实现了简单的分布式限流算法。在实际项目中华选择了更贴合业务场景的滑动窗口算法。

手写限流工具

令牌桶

场景：商品秒杀场景，例如秒杀100台电脑，秒杀开始前向内存中预热110个token，拿到token的请求才能执行后面的流程

-- 限流目标
local key = KEYS[1]
-- 上一次令牌生成时间
local time_key = KEYS[2]
local capacity = tonumber(ARGV[1])
local qps = tonumber(ARGV[2])
local cur_time = tonumber(ARGV[3])
local one_week = 60 * 60 * 24 * 7
-- 获取桶中令牌
local token = redis.call('get', key)
if token then
    token = tonumber(token)
else
    token = capacity
end
-- 计算时间段生成的令牌
local last_time = redis.call('get', time_key)
if not last_time then
    last_time = cur_time
else
    last_time = tonumber(last_time)
end
<