nginx+lua+redis防刷，lua代码

问题：

当机器过多时，在每台机器的nginx上用nginx自带防刷模块，往往限制太松。


思路：

多台机器，通过nginx-lua模块，连接redis，以ip（记得nginx安装real-ip模块，取到x-forword-for字段对应的真实用户ip）来更新访问次数，并根据redis设置的阀值进行比较，决定是否限流，不考虑并发更新丢值情况，因为访问次数足够时，总能到达阀值。


优化：

1、redis发送请求，需要查访问次数和阀值，发两次请求浪费资源，可进行合并。
通过eval指令，执行redis-lua脚本，一次性返回两个字段。

2、每次发送redis执行的lua脚本，传输字符较多，浪费带宽。
通过evalsha执行脚本对应的校验码。若redis执行过一个lua脚本后，会记录脚本，并生成对应的校验码，可通过evalsha指令，参数为对应的校验码，即可执行脚本。

3、（不采用）建立nginx到redis的连接池，防止每次都建立链接。但考虑到以下两点，所以放弃使用，如有错误和方案，麻烦指正。
（1）、连接池不应该很大，所以在并发量很大的时候，大多数请求还是要从新建立连接。
（2）、要维护连接池，保证同一时刻没有两个请求共用一个链接，造成连接关闭的异常。没有一个变量，用来标识调用不同的连接。

lua代码(可直接使用，亲测可用)：

首先需要在nginx上输入下面的配置：
lua_shared_dict ngx_shared_redis 1m;
建立一个从nginx启动后便存在的共享变量，1m大小（可以小点），名字叫redis。用于存储生成的redis-lua叫的校验码，用ngx.shared.redis:get("redis")方式来获取。

--关闭连接
local function close_redis(red)
    if not red then
        return
    end
    red:close()
end
--主要处理函数
local function get_limit()
    --加载模块
    local red = require("resty.redis"):new()
    red:set_timeout(100)
    local ok, err = red:connect("127.0.0.1", 6379)
    if not ok then
        ngx.log(ngx.ERR, "redis_connect:"..err)
        return close_redis(red)
    end
    --根据ip作为key，来防刷
    local key = ngx.var.remote_addr
    --判断之前是否已经执行过redis-lua脚本，且有对应的校验码值用于调用
    if not ngx.shared.redis:get("redis") then
        --若之前未执行过，第一次执行
        local script = table.concat({
            --手动拼接为一行，两个参数，第一个为key即ip，第二个为限制阀值key
            "local val = redis.call('get',KEYS[1]) ",
            "if val then ",
                "val = redis.call('incr',KEYS[1]) ",
                "return {val,redis.call('get', KEYS[2])} ",
            "end ",
            "redis.call('set',KEYS[1],1) ",
            "redis.call('expire',KEYS[1],60) ",
            "return {0,nil}"
        })
        --在redis加载对应的lua脚本
        local sha1, err = red:script("load", script)
        if not sha1 then
            ngx.log(ngx.ERR, "load_script:"..err)
            return close_redis(red)
        end
        --拿到生成的校验码，更新对应的字段
        ngx.shared.redis:set("redis", sha1)
    end
    --根据校验码，执行脚本，cart_limit为限制阀值key
    local resp, err = red:evalsha(ngx.shared.redis:get("redis"), 2, key, "cart_limit");
    if not resp then
        --若执行失败，一种情况为，redis清空了脚本缓存，此情况下，退出并删除存储的校验码，等下次执行，再更新
        ngx.log(ngx.ERR, "not_resp:"..err)
        ngx.shared.redis:delete("redis")
        return close_redis(red)
    end
    --限制值若redis没有，默认为1200，若有则判断是否大于120
    local limit = 1200
    if resp[2] then
        local new_limit = tonumber(resp[2])
        limit = new_limit > 120 and new_limit or 1200
    end
    --若次数超过限制，则拦截
    if resp[1] > limit then
        ngx.exit(ngx.HTTP_FORBIDDEN)
    end
    close_redis(red)
end
--用pcall,调用脚本，类似try-catch，如有错误，打印脚本
if not pcall(get_limit) then
    ngx.log(ngx.ERR, "lua error")
end

以上若有错误，欢迎指正，谢谢！