【转】基于nginx + lua实现的反向代理动态更新

大家都知道，nginx是当前应用非常广泛的web服务器，热度因为他的高并发高性能高可靠性，且轻量级！牛逼的不行，不多说这些。

今天要介绍的是，如何基于nginx和lua脚本，也就是在openresty的环境下，实现动态的反向代理逻辑，有一个开关控制。开关控制反向代理工作在nginx原生的upstream的模式，还是工作在lua控制的动态代理模式。动态代理的服务器，通过http请求实现灵活的操作，向lua_shared_dict定义的全局变量里面写入或者删除动态代理的服务器信息。

环境信息如下：

1 [root@bogon sbin]# ./nginx -V
2 nginx version: openresty/1.11.2.2
3 built by gcc 4.8.5 20150623 (Red Hat 4.8.5-11) (GCC) 
4 built with OpenSSL 1.0.1e-fips 11 Feb 2013
5 TLS SNI support enabled
6 configure arguments: --prefix=/usr/local/openresty/nginx --with-cc-opt=-O2 --add-module=../ngx_devel_kit-0.3.0 --add-module=../echo-nginx-module-0.60 --add-module=../xss-nginx-module-0.05 --add-module=../ngx_coolkit-0.2rc3 --add-module=../set-misc-nginx-module-0.31 --add-module=../form-input-nginx-module-0.12 --add-module=../encrypted-session-nginx-module-0.06 --add-module=../srcache-nginx-module-0.31 --add-module=../ngx_lua-0.10.7 --add-module=../ngx_lua_upstream-0.06 --add-module=../headers-more-nginx-module-0.32 --add-module=../array-var-nginx-module-0.05 --add-module=../memc-nginx-module-0.17 --add-module=../redis2-nginx-module-0.13 --add-module=../redis-nginx-module-0.3.7 --add-module=../rds-json-nginx-module-0.14 --add-module=../rds-csv-nginx-module-0.07 --with-ld-opt=-Wl,-rpath,/usr/local/openresty/luajit/lib --add-module=/opt/nginx-rtmp-module-master --with-http_ssl_module

先设计一下功能模块，在nginx.conf文件里面，有如下几个location：

location / {。。。}
业务逻辑的入口

location /config {。。。}
动态反向代理开关控制入口

location /add_ups {。。。}
添加反向代理的服务器信息进入lua_shared_dict定义的全局表单

location /stop_ups {。。。}
从lua_shared_dict定义的全局表单里面删除掉不再参与反向代理的服务器信息

location /check_ups {。。。}
查看当前lua_shared_dict定义的全局表单里面有那些服务器信息

为了验证这个设计，我在本地开发机器上，将同一个RDConsumer应用部署在3个不同的端口下，对应于nginx里面的upstream块：

upstream robot_ups {
    server 10.90.9.20:8090;
    server 10.90.9.20:8081;
    server 10.90.9.20:9080;
}

在这里，强调一下，这里，我们验证的动态反向代理，是轮询的方式，当然，根据需要可以设计成符合各自业务的逻辑。

下面，对这个设计的全部过程进行详细介绍。若有需要，可以用起来或者转走，觉得好，还可以点个赞，或者加个关注！

1. nginx.conf的配置文件

#user  nobody;
worker_processes  4;

error_log  logs/error.log;
error_log  logs/error.log  notice;
error_log  logs/error.log  info;

#pid        logs/nginx.pid;


events {
    worker_connections  1024;
    use   epoll;
}

http {
    lua_shared_dict dyn_ups_zone 10m;

    include       mime.types;
    default_type  application/octet-stream;

    #log_format  main  '$remote_addr - $remote_user [$time_local] "$request" '
    #                  '$status $body_bytes_sent "$http_referer" '
    #                  '"$http_user_agent" "$http_x_forwarded_for"';

    #access_log  logs/access.log  main;

    sendfile        on;
    #tcp_nopush     on;

    #keepalive_timeout  0;
    keepalive_timeout  65;

    #gzip  on;
 
    upstream robot_ups {
        server 10.90.9.20:8090;
        server 10.90.9.20:8081;
        server 10.90.9.20:9080;
    }

    server {
        listen       80;
        server_name  localhost;

        #charset koi8-r;
        
        location / {
            set_by_lua_file $cur_ups /opt/shihuc/luahome/ablb/bussups.lua
            proxy_next_upstream off;  
            proxy_set_header Host $host:$server_port;
            proxy_set_header Remote_Addr $remote_addr;
            proxy_set_header X-Real-IP $remote_addr;
            proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
            proxy_pass http://$cur_ups;  
        } 

        location /config {
            default_type  text/plain;
            content_by_lua_block {
                local foo = ngx.req.get_uri_args()["foo"]
                if foo == nil then
                    ngx.say("usage: /config?foo=off, or /config?foo=on")
                    return;
                end
                ngx.log(ngx.INFO, "config ab deploy feature: ", foo);
                ngx.say("config ab deploy feature: ", foo);
                    
                ngx.shared.dyn_ups_zone:set("robotfoo", foo);
            }
        }

        location /stop_ups {
            default_type  text/plain;
            content_by_lua_file /opt/shihuc/luahome/ablb/stopups.lua;
        }

        location /add_ups {
            default_type  text/plain;
            content_by_lua_block {
                local add_s = ngx.req.get_uri_args()["ups"]  
                if add_s == nil then 
                    ngx.say("usage: /add_ups?ups=x.x.x.x")  
                    return "no add_s"
                end
                ngx.log(ngx.INFO, "add upstream server: ", add_s);
                local dynupszone = ngx.shared.dyn_ups_zone;
                local ups = dynupszone:get("robotups");
                if ups == nil then
                   ngx.say("first init global dict dyn_ups_zone");
                   ups = add_s
                else
                   local unders = "-";
                   ups = ups..unders
                   ups = ups..add_s
                end
                local succ, err, forcible = dynupszone:set("robotups", ups);
                ngx.say("succ: ",succ, ", err: ", err, ", forcible: ", forcible);
                ngx.say(dynupszone.get(dynupszone, "robotups"))
            }
        }
        
        location /check_ups {
            default_type  text/plain;
            content_by_lua_block {
                local dynupszone = ngx.shared.dyn_ups_zone;
                local ups = dynupszone:get("robotups");
                ngx.say(ups);
            }
        }

        error_page   500 502 503 504  /50x.html;
        location = /50x.html {
            root   html;
        }
    }
}