nginx+lua 限制接口访问次数

最近看了一些nginx+lua 的东西，尝试实现了一下限流脚本，只包含最根本的功能。

代码如下

access_nginx_check.lua

-- 一个按照 url + 参数 进行 在固定时间内访问次数次数限制的lua 脚本，
--  此处仅仅实现 按照url 10s 内限制 2次访问，记录每次访问次数（无用仅仅用于核对代码逻辑可去除），获取参数函数已有，后续需要完善

--package.path = package.path ..';..\\?.lua';
--dofile ("./log.lua")
--是否开启检测
local check = "on"    
--限制最大访问量
local count = 2
--限制时间范围
local seconds = 10
--获取参数的值 
function getParam()
	if "GET" == request_method then
	    args = ngx.req.get_uri_args()
	elseif "POST" == request_method then
	    ngx.req.read_body()
	    args = ngx.req.get_post_args()
	end
	return args
end
--写入日志
function writerLog(str)
	local  time = os.time()
	local  date = os.date("%Y%m%d",time)

	local file = io.open("/var/www/lua/logTest/"..tostring(date).."_log.log","a")
	--ngx.say("/var/www/lua/log/"..tostring(date).."_log.log")
	--assert(file)
	file:write(str.."\n")
	file:close();

end
--对uri 进行限制主要逻辑
function access_user_uri_check()
	-- body
	if check=="on" then
		local access_uri = ngx.var.host..ngx.var.uri
		--local param = getParam()
		local key = access_uri
--提前在nginx中声明的变量
		local limit = ngx.shared.limit
		local req 
		if limit then
			req ,_= limit:get(key)
		end
		if req then
			if req>=count then
--超过限制之后返回错误
				ngx.say("您在最近:"..seconds.."秒,访问："..access_uri.."次数超过:"..count.."次，已经触发保护机制，请稍后访问")
				ngx.exit(403)
			end
--对key对应的数据进行累加
			limit:incr(key,1)
			
		else
--没有值设置初始为1
			limit:set(key,1,seconds)
		end
--记录log
		writerLog(key..":"..limit:get(key))

	end

end
--错误调试函数
function myerrorhandler( err )
   ngx.say( "ERROR:", err )
end
--测试代码，可以自动执行
status = xpcall( access_user_uri_check, myerrorhandler )
--access_user_uri_check()

对应nginx 配置

#声明变量 对应lua 脚本中 的ngx.shared.limit
lua_shared_dict limit 10m;
server{
    listen 80;
    server_name localhost.accesslua.com;
    root /var/www/html/test;
    charset utf-8;
    location ~\.php{
#为了调试lua脚本，直接将信息抛出到浏览器，否则会自动下载
        default_type 'text/html';
#嵌入的lua脚本
        access_by_lua_file "/var/www/lua/access_nginx_check.lua";
#php 的正常调用
        fastcgi_pass 192.168.33.11:9000;
        fastcgi_index index.php;
        fastcgi_param  SCRIPT_FILENAME  $document_root$fastcgi_script_name;
        include        fastcgi_params;
    }
#记录log
    access_log  /usr/local/var/log/lua.accesslua-php5.access.log  main;

}

演示效果