分布式限流之Redis+Lua实现

最新推荐文章于 2024-07-02 14:08:21 发布

后端老鸟

最新推荐文章于 2024-07-02 14:08:21 发布

阅读量1.2k

点赞数 2

文章标签：分布式 redis lua

本文链接：https://blog.csdn.net/huahao1989/article/details/105911743

版权

本文介绍了如何利用Redis和Lua脚本来实现分布式限流，确保操作的原子性和线程安全性。通过一个SpringBoot项目实例，展示了如何配置Lua脚本、Redis、限流注解以及控制层，以限制时间窗口内的接口请求次数。测试结果表明，当请求超过设定限制时，系统会返回异常。

摘要由CSDN通过智能技术生成

【转载请注明出处】：https://blog.csdn.net/huahao1989/article/details/105911743

分布式限流最关键的是要将限流服务做成原子化，而解决方案可以使用redis+lua或者nginx+lua技术进行实现，通过这两种技术可以实现的高并发和高性能。

首先我们来使用redis+lua实现时间窗内某个接口的请求数限流，实现了该功能后可以改造为限流总并发/请求数和限制总资源数。Lua本身就是一种编程语言，也可以使用它实现复杂的令牌桶或漏桶算法。
因操作是在一个lua脚本中（相当于原子操作），又因Redis是单线程模型，因此是线程安全的。

相比Redis事务来说，Lua脚本有以下优点

减少网络开销：不使用 Lua 的代码需要向 Redis 发送多次请求，而脚本只需一次即可，减少网络传输；
原子操作：Redis 将整个脚本作为一个原子执行，无需担心并发，也就无需事务；
复用：脚本会永久保存 Redis 中，其他客户端可继续使用。

下面使用SpringBoot项目来进行介绍。

准备Lua 脚本

req_ratelimit.lua

local key = "req.rate.limit:" .. KEYS[1]   --限流KEY
local limitCount = tonumber(ARGV[1])       --限流大小
local limitTime = tonumber(ARGV[2])        --限流时间
local current = tonumber(redis.call('get', key) or "0")
if current + 1 > limitCount then --如果超出限流大小
    return 0
else  --请求数+1，并设置1秒过期
    redis.call("INCRBY", key,"1")
    redis.call("expire", key,limitTime)
    return current + 1
end