openresty 限流

准备环境

准备lua script

mkdir lua

创建lua/test.lua,并写入内容

local redis = require('resty.redis')
local red = redis.new()
red:set_timeout(1000)

local function close(red)
    if not red then
        return
    end
    local ok, err = red:set_keepalive(10000, 100)
    if not ok then
        ngx.log(ngx.ERR, "failed to set keepalive: ", err)
    end
end

local ok, err = red:connect("127.0.0.1",6379)
if not ok then
    return close(red)
end

这边限制每秒2次(方便验证)
inc  = red:incr("testlimit")
if inc >2 then
    red:expire("testlimit",10)
    ngx.exit(ngx.HTTP_FORBIDDEN)
end

if inc <2 then
    red:expire("testlimit",1)
end
close(red)

准备html/index.html,空文件即可

修改conf/nginx.conf

http {
    server {
        listen 8080;
        location / {
            default_type text/html;
access_by_lua_file "lua/test.lua";
        }
    }
}

重新加载配置

ps aux|grep nginx
想master进程发送信号
sudo kill -s SIGHUP pid

验证

连续执行3次

curl -v http://localhost:8080

可以看到403了
查看redis

redis-cli
ttl testlimit可以看到过期时间 

总结

这里只是粗粒度的限流,如果要细粒度的限流,比如ip(可以判断X-Real-IP(),X-FORWARD-FOR(从 ngx.req.get_headers()中获取),以及ngx.var.remote_addr)