lua并发限流代码

最新推荐文章于 2025-01-11 07:50:52 发布

原创最新推荐文章于 2025-01-11 07:50:52 发布 · 1.9k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#API限流 #并发限流 #分布式架构限流

微服务架构专栏收录该内容

10 篇文章

订阅专栏

本文介绍了如何在Nginx中使用Lua实现并发限流，通过在nginx.conf配置lua_shared_dict，并展示了具体的Lua限流代码，以保障分布式架构下的API服务稳定性。

首先在nginx.conf 文件中的http节点添加：

lua_shared_dict limit_req_store 10m;

以下是lua并发限流代码：

local limit_req = require "resty.limit.req"

--限制请求速率为500 req/sec
local lim,err = limit_req.new("limit_req_store",5,0)
if not lim then --申请limit_req对象失败
ngx.log(ngx.ERR,"failed to instantiate a resty.limit.req object: ",err)
return ngx.exit(500)
end

-- 下面代码针对每一个单独的请求
-- 使用ip地址作为限流的key
local key = ngx.var.binary_remote_addr
local delay,err = lim:incoming(key,true)
if not delay then
if err == "rejected" then
return ngx.exit(503)
end
ngx.log(ngx.ERR,"failed to limit req: ",err)
return ngx.exit(500)
end


if delay > 0 then
--第二参数（err）保存着超过请求速率的请求数
--例如err等于31，意味着当前速率是 531 req/sec
local excess = err

---- 当前请求超过200 req/sec 但小于 300 req/sec
-- 因此我们sleep一下，保证速率是200 req/sec，请求延迟处理
ngx.sleep(delay) --非阻塞sleep(秒)
end
ngx.say("you success!!!")