NGINX 结合 lua 动态修改upstream

关注

原创最新推荐文章于 2026-04-17 16:13:45 发布 2.6w 阅读 AI 写同款· GEO 优化 ›

Nginx负载均衡之动态更新upstream 由于该指令只支持 worker 级别的操作，为使得 Nginx 的所有 worker 都生效，此处通过编写 Lua 脚本与 lua-resty-upstream-healthcheck 模块做了简单的集成，利用 lua-resty-upstream-healthcheck 模块的共享内存机制将节点状态同步给其他工作进程，实现对 upstream 的节点状态的控制。在 Nginx 的商业版本中，提供了 ngx_http_api_module 模块，可以通过 API 动态添加或移除上游服务器组中的节点。阅读详情

具体思路是：

1 > 利用lua中 "lua_shared_dict" 指令开辟一个共享内存空间;

2> 通过API动态根据key值&参数修改 upstream (这里使用 host 作为key);

3> 利用 proxy_pass 可使用变量特性及lua指令 "set_by_lua" 动态修改当前 upstream 变量即可；

以下是利用 qq.com 作为示例:

http {
	lua_shared_dict _ups_zone 64m; # 定义upstream共享内存空间
	...
	upstream qq_backend{
		server 14.17.42.40;
		server 14.17.32.211;
		server 59.37.96.63;
	}

	server {
		listen       80;
		server_name  www.qq.com *.qq.com;
		access_log  off;

		# 更新 upstream API 接口
		location = /ups_update {
			content_by_lua '
				local ups = ngx.req.get_uri_args()["ups"]
				if ups == nil then
					ngx.say("usage: /ups_update?ups=x.x.x.x")
					return
				end
				local host = ngx.var.http_host
				local ups_from = ngx.shared._ups_zone:get(host);
				ngx.log(ngx.WARN, host, " update upstream from ", ups_from, " to ", ups)
				ngx.shared._ups_zone:set(host, ups);
				ngx.say(host, " update upstream from ", ups_from, " to ", ups)
			';
		}
		location / {
			# 动态设置当前 upstream, 未设置则使用默认 upstream
			set_by_lua $cur_ups '
				local ups = ngx.shared._ups_zone:get(ngx.var.http_host)
				if ups ~= nil then
					ngx.log(ngx.ERR, "get [", ups, "] from ngx.shared._ups_zone")
					return ups
				end
				--ngx.log(ngx.INFO, "use default upstream");
				return "qq_backend";
			'
			proxy_next_upstream off;
			proxy_set_header    Host $host;
			proxy_http_version  1.1;
			proxy_set_header    Connection  "";
			proxy_pass $scheme://$cur_ups;
		}
	}
}

测试:

cur -x 127.0.0.1:80 www.qq.com  # 使用默认upstream
# 修改为140.206.160.207 & 61.135.157.156
curl -x 127.0.0.1:80 www.qq.com/ups_update?ups=140.206.160.207
curl -x 127.0.0.1:80 www.qq.com/ups_update?ups=61.135.157.156

这里只是一个简单的思路，并未考虑到 upstream 持久化，如需要可参考以下这篇文章, 使用 redis 来实现:

Dynamic nginx upstreams with Lua and Redis

收录于