涨薪5K必学高并发核心编程，限流原理与实战，分布式计数器限流

local redisExecutor = require("luaScript.redis.RedisOperator");--一个统一的模块对象local _Module = {}_Module.__index = _Module--方法：创建一个新的实例function _Module.new(self, key) local object = { red = nil } setmetatable(object, self) --创建自定义的redis操作对象 local red = redisExecutor:new(); red:open(); object.red = red; object.key = "count_rate_limit:" .. key; return objectend--方法：判断是否能通过流量控制--返回值为true表示通过流量控制，返回值为false表示被限制function _Module.acquire(self) local redis = self.red; local current = redis:getValue(self.key); --判断是否大于限制次数 local limited = current and current ~= ngx.null and tonumber(current) > ; --限流的次数 --被限流 if limited then redis:incrValue(self.key); return false; end if not current or current == ngx.null then redis:setValue(self.key, ); redis:expire(self.key, ); --限流的时间范围 else redis:incrValue(self.key); end return true;end--方法：取得访问次数，供演示使用function _Module.getCount(self) local current = self.red:getValue(self.key); if current and current ~= ngx.null then return tonumber(current); end return ;end--方法：归还redis连接function _Module.close(self) self.red:close();endreturn _Module以上代码位于练习工程LuaDemoProject的src/luaScript/module/ratelimit/文件夹下，文件名称为RedisKeyRateLimiter.lua。然后介绍access_auth_nginx限流脚本，该脚本使用前面定义的RedisKeyRateLimiter.lua通用访问计算器脚本，完成针对同一个IP的限流操作，具体的代码如下：---此脚本的环境：nginx内部---启动调试--local mobdebug = require("luaScript.initial.mobdebug");--mobdebug.start();--导入自定义的计数器模块local RedisKeyRateLimiter = require("luaScript.module.ratelimit.RedisKeyRateLimiter");定义出错的输出对象--定义出错的JSON输出对象local errorOut = { resp_code = -1, resp_msg = "限流出错", datas = {} };--取得用户的iplocal shortKey = ngx.var.remote_addr;--没有限流关键字段，提示错误if not shortKey or shortKey == ngx.null then errorOut.resp_msg = "shortKey不能为空" ngx.say(cjson.encode(errorOut)); return ;end--拼接计数的redis keylocal key = "ip:" .. shortKey;local limiter = RedisKeyRateLimiter:new(key);local passed = limiter:acquire();--如果通过流量控制if passed then ngx.var.count = limiter:getCount(); --注意，在这里直接输出会导致content阶段的指令被跳过 --ngx.say( "目前的访问总数：",limiter:getCount(),"<br>");end--回收redis连接limiter:close();--如果没有流量控制，就终止nginx的处理流程if not passed then errorOut.resp_msg = "抱歉，被限流了"; ngx.say(cjson.encode(errorOut)); ngx.exit(ngx.HTTP_UNAUTHORIZED);endreturn ;

count_rate_limit:ip:192.168.233.1

location = /access/demo/nginx/lua { set $count ; access_by_lua_file luaScript/module/ratelimit/access_auth_nginx.lua; content_by_lua_block { ngx.say( "目前的访问总数：",ngx.var.count,"<br>"); ngx.say("hello world!"); }}

http://nginx.server/access/demo/nginx/lua?seckillGoodId=1

---此脚本的环境：redis内部，不是运行在Nginx内部--返回表示被限流，返回其他表示统计的次数local cacheKey = KEYS[]local data = redis.call("incr", cacheKey)local count=tonumber(data)--首次访问，设置过期时间if count ==  then redis.call("expire", cacheKey, ) --设置超时时间秒endif count >  then --设置超过的限制为人表示需要限流 return ; --表示需要限流end--redis.debug(redis.call("get", cacheKey))return count;

[root@localhost ~]#cd /work/develop/LuaDemoProject/src/luaScript/module/ratelimit/[root@localhost ratelimit]#/usr/local/redis/bin/redis-cli script load "$(cat redis_rate_limiter.lua)""2c95b6bc3be1aa662cfee3bdbd6f00e8115ac657"

---此脚本的环境：nginx内部local RedisKeyRateLimiter = require("luaScript.module.ratelimit.RedisKeyRateLimiter");--定义出错的JSON输出对象local errorOut = { resp_code = -, resp_msg = "限流出错", datas = {} };--读取get参数local args = ngx.req.get_uri_args()--取得用户的iplocal shortKey = ngx.var.remote_addr;--没有限流关键字段，提示错误if not shortKey or shortKey == ngx.null then errorOut.resp_msg = "shortKey不能为空" ngx.say(cjson.encode(errorOut)); return ;end--拼接计数的redis keylocal key = "count_rate_limit:ip:" .. shortKey;local limiter = RedisKeyRateLimiter:new(key);local passed = limiter:acquire();--如果通过流量控制if passed then ngx.var.count = limiter:getCount(); --注意，在这里直接输出会导致content阶段的指令被跳过 --ngx.say( "目前的访问总数：",limiter:getCount(),"<br>");end--回收redis连接limiter:close();如果没有流量控制就终止的处理流程--如果没有流量控制，就终止Nginx的处理流程if not passed then errorOut.resp_msg = "抱歉，被限流了"; ngx.say(cjson.encode(errorOut)); ngx.exit(ngx.HTTP_UNAUTHORIZED);endreturn ;

 location = /access/demo/evalsha/lua { set $count ; access_by_lua_file luaScript/module/ratelimit/access_auth_evalsha.lua; content_by_lua_block { ngx.say( "目前的访问总数：",ngx.var.count,"<br>"); ngx.say("hello world!"); }}

http://nginx.server/access/demo/evalsha/lua