location / {  
    # 如果該location 下存在靜態(tài)資源文件可以做一個(gè)判斷        
    #if ($request_uri ~ .*\.(html|htm|jpg|js|css)) {  
    # access_by_lua_file /usr/local/lua/access_limit.lua;     
    #}  
      
    access_by_lua_file /usr/local/lua/access_limit.lua; # 加上了這條配置，則會(huì)根據(jù) access_limit.lua 的規(guī)則進(jìn)行限流  
    alias /usr/local/web/;  
    index  index.html index.htm;  
}  

-- 可以實(shí)現(xiàn)自動(dòng)將訪問頻次過高的IP地址加入黑名單封禁一段時(shí)間  
  
--連接池超時(shí)回收毫秒  
local pool_max_idle_time = 10000  
--連接池大小  
local pool_size = 100  
--redis 連接超時(shí)時(shí)間  
local redis_connection_timeout = 100  
--redis host  
local redis_host = "your redis host ip"  
--redis port  
local redis_port = "your redis port"  
--redis auth  
local redis_auth = "your redis authpassword";  
--封禁IP時(shí)間（秒）  
local ip_block_time= 120  
--指定ip訪問頻率時(shí)間段（秒）  
local ip_time_out = 1  
--指定ip訪問頻率計(jì)數(shù)最大值（次）  
local ip_max_count = 3  
  
  
--  錯(cuò)誤日志記錄  
local function errlog(msg, ex)  
    ngx.log(ngx.ERR, msg, ex)  
end  
  
-- 釋放連接池  
local function close_redis(red)  
    if not red then  
        return  
    end  
    local ok, err = red:set_keepalive(pool_max_idle_time, pool_size)  
    if not ok then  
        ngx.say("redis connct err:",err)  
        return red:close()  
    end  
end  
  
  
--連接redis  
local redis = require "resty.redis"  
local client = redis:new()  
local ok, err = client:connect(redis_host, redis_port)  
-- 連接失敗返回服務(wù)器錯(cuò)誤  
if not ok then  
    close_redis(client)  
    ngx.exit(ngx.HTTP_INTERNAL_SERVER_ERROR)  
end  
--設(shè)置超時(shí)時(shí)間  
client:set_timeout(redis_connection_timeout)  
  
-- 優(yōu)化驗(yàn)證密碼操作 代表連接在連接池使用的次數(shù)，如果為0代表未使用，不為0代表復(fù)用 在只有為0時(shí)才進(jìn)行密碼校驗(yàn)  
local connCount, err = client:get_reused_times()  
-- 新建連接，需要認(rèn)證密碼  
if  0 == connCount then  
    local ok, err = client:auth(redis_auth)  
    if not ok then  
        errlog("failed to auth: ", err)  
        return  
    end  
    --從連接池中獲取連接，無需再次認(rèn)證密碼  
elseif err then  
    errlog("failed to get reused times: ", err)  
    return  
end  
  
-- 獲取請(qǐng)求ip  
local function getIp()  
    local clientIP = ngx.req.get_headers()["X-Real-IP"]  
    if clientIP == nil then  
        clientIP = ngx.req.get_headers()["x_forwarded_for"]  
    end  
    if clientIP == nil then  
        clientIP = ngx.var.remote_addr  
    end  
    return clientIP  
end  
  
local cliendIp = getIp();  
  
local incrKey = "limit:count:"..cliendIp  
local blockKey = "limit:block:"..cliendIp  
  
--查詢ip是否被禁止訪問，如果存在則返回403錯(cuò)誤代碼  
local is_block,err = client:get(blockKey)  
if tonumber(is_block) == 1 then  
    ngx.exit(ngx.HTTP_FORBIDDEN)  
    close_redis(client)  
end  
  
local ip_count, err = client:incr(incrKey)  
if tonumber(ip_count) == 1 then  
    client:expire(incrKey,ip_time_out)  
end  
--如果超過單位時(shí)間限制的訪問次數(shù)，則添加限制訪問標(biāo)識(shí)，限制時(shí)間為ip_block_time  
if tonumber(ip_count) > tonumber(ip_max_count) then  
    client:set(blockKey,1)  
    client:expire(blockKey,ip_block_time)  
end  
  
close_redis(client)