lua并发限流代码

诸葛本不亮

于 2018-10-16 22:11:26 发布

阅读量1.9k

点赞数

CC 4.0 BY-SA版权

分类专栏：微服务架构文章标签： API限流并发限流分布式架构限流

本文链接：https://blog.youkuaiyun.com/david_pfw/article/details/83098454

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

10 篇文章

订阅专栏

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

首先在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!!!")