NGINX 結合 lua 動態修改upstream

from http://blog.csdn.net/force_eagle/article/details/51966333

 

具體思路是： 

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