快捷導(dǎo)航

nginx+lua+redis實(shí)現(xiàn)限流的示例代碼

更新時(shí)間：2025年12月09日 10:13:31 作者：果殼~

本文主要介紹了nginx+lua+redis實(shí)現(xiàn)限流的示例代碼,包括HttpRedis模塊、HttpRedis2Module模塊和lua-resty-redis庫(kù),具有一定的參考價(jià)值,感興趣的可以了解一下

概述

nginx、lua訪問(wèn)redis的三種方式

HttpRedis模塊
指令少，功能單一，適合簡(jiǎn)單緩存。只支持get,select命令。
HttpRedis2Module模塊
功能強(qiáng)大，比較靈活
lua-resty-redis庫(kù)
OpenResty提供的API。適合復(fù)雜業(yè)務(wù)，節(jié)省內(nèi)存。
以上3個(gè)模塊OpenResty都有集成

OpenResty: 基于nginx開(kāi)源版本的一個(gè)擴(kuò)展版本。集成了大量的精良的lua庫(kù)。所以接下來(lái)我們就使用OpenResty來(lái)實(shí)現(xiàn)相應(yīng)的功能

OpenResty的安裝

# 安裝wget 如果沒(méi)有的話
yum install wget
# 下載資源庫(kù) 這樣yum就可以直接安裝了
# 得到 openresty.repo
cd /etc/yum.repos.d/
wget https://openresty.org/package/centos/openresty.repo
# 安裝openresty 安裝目錄:/usr/local/openresty
yum install openresty

編寫(xiě)nginx配置

cd /usr/local/openresty/nginx/conf
vim nginx-lua.conf

openresty 提供了幾種方式來(lái)配置lua腳本需要先了解一下

content_by_lua 'ngx.say("hello my openrestry")' : 可以直接在配置文件中寫(xiě)入單行的lua腳本的字符串。
content_by_lua_block :可以在nginx配置文件中配置多行的 lua腳本代碼塊

content_by_lua_block {
	ngx.say("hello");
	ngx.say("block");
}

content_by_lua_file /usr/local/openresty/nginx/lua/lua-test.lua: 可以配置lua腳本的文件路徑
log_by_lua_file /usr/local/openresty/nginx/lua/lua-log-test.lua : 可以配置lua腳本的日志文件

# nginx-lua.conf
worker_processes 1;
error_log logs/error.log debug;
events {
	worker_connections 1024;
}

http {
	include mime.types;
	default_type application/octet-stream;
	sendfile on;
	# 連接的超時(shí)時(shí)間
	keepalive_timeout 65;
	server {
		listen 8080;
		location / {
			default_type text/html;
			# 此處可以寫(xiě)lua腳本的文件路徑
			content_by_lua_file /usr/local/openresty/nginx/lua/ip_limit_log.lua;
		}
	}
}

-- ip_limit_access.lua
ngx.say("ip limit lua");

先編寫(xiě)一個(gè)簡(jiǎn)單的腳本測(cè)試看看是否配置成功。
重啟nginx并加載指定配置

# 查看nginx是否啟動(dòng)
ps -ef | grep nginx
# 停止nginx
/usr/local/openresty/nginx/sbin/nginx -s stop
# 啟動(dòng)nginx  -p指定工作目錄  -c 指定配置文件
/usr/local/openresty/nginx/sbin/nginx -p /usr/local/openresty/nginx/ -c /usr/local/openresty/nginx/conf/nginx-lua.conf
# 使用curl訪問(wèn)地址 測(cè)試成功
[root@localhost nginx]# curl http://localhost
ip limit lua

nginx_lua_redis限流

通過(guò)以上測(cè)試，nginx配置lua腳本已經(jīng)通過(guò)接下來(lái)就可以開(kāi)始實(shí)現(xiàn)限流功能了。

整體思路

編寫(xiě)nginx配置

編寫(xiě)配置ngin-ip-limit.conf

# ngin-ip-limit.conf
worker_processes 1;
error_log logs/error.log debug;
events {
	worker_connections 1024;
}
http {
	include mime.types;
	default_type application/octet-stream;
	server {
		listen 80;
		localtion / {
			default_type text/html;
			# 配置lua腳本文件路徑
			access_by_lua_file /usr/local/openresty/nginx/lua/ip_limit_access.lua;
			# 配置lua日志腳本路徑
			log_by_lua_file /usr/local/openresty/nginx/lua/ip_limit_log.lua;
			# 需要準(zhǔn)備一個(gè)被代理的服務(wù)
			proxy_pass http://localhost:8080/;
		}
	}
}

編寫(xiě)lua日志腳本

-- ip_limit_log.lua
local ip = ngx.var.remote_addr;
ngx.log(ngx.INFO, "request ip is:"..ip);

編寫(xiě)lua限流腳本

需求：系統(tǒng)每秒限流2個(gè)請(qǐng)求，如果超過(guò)閾值(每秒2個(gè)請(qǐng)求)，則系統(tǒng)限制10秒內(nèi)，不能被訪問(wèn)

-- ip_limit_access.lua
ngx.log(ngx.INFO, "ip limit log");

local redis = require "resty.redis";
local red = redis:new();

-- 連接redis
red:connect("127.0.0.1",6379);

-- 判斷是否限流
limit = red:get("limit");
if limit == '1' then
  return ngx.exit(503);
end
-- 次數(shù)加1
inc = red:incr("testLimit");
if inc <= 2 then
  -- 設(shè)置過(guò)期時(shí)間 1秒
  red:expire("testLimit",1);
else
  -- 超過(guò)閾值 limit設(shè)置成1 并設(shè)置過(guò)期時(shí)間10秒
  red:set("limit",1);
  red:expire("limit", 10);
end