使用balancer_by_lua实现动态转发

lua_shared_dict resolvConf 32k;
lua_shared_dict dnsCache 1m;

init_by_lua_block {
    local cjson = require "cjson.safe"
    local dnsutils = require "resty.dns.utils"

    local config = ngx.shared.resolvConf
    local resolvConf = dnsutils.parseResolvConf()

    for name, value in pairs(resolvConf) do
        config:set(name, cjson.encode(value))
    end
}

upstream backend_svc {
    server 0.0.0.1 down; # 占位server

    balancer_by_lua_block {
        local balancer = require "ngx.balancer"
        balancer.set_timeouts(1, 0.5, 0.5)  -- 后端的连接、读、写超时时间
        balancer.set_more_tries(1)          -- 连接失败后最多重试1次
        local ip = ngx.ctx.backend_ip
        local ok, err = balancer.set_current_peer(ip, 8101)
        if not ok then
            ngx.log(ngx.ERR, "failed to set current_peer: ", err)
            return ngx.exit(500)
        end
    }

    keepalive 10;
}

server {
    listen       8101;
    server_name  _;

    location / {
        rewrite_by_lua_file '/rewrite_handle.lua';
        #默认值为0：重试次数不受限制
        proxy_next_upstream_tries 2;
        proxy_pass http://backend_svc;
    }
}

local cjson = require "cjson"
local resolver = require "resty.dns.resolver"
local nkeys = require "table.nkeys"
local json_decode = cjson.decode
local ngx = ngx
local tostring = tostring

local uri = ngx.var.uri
local method = ngx.req.get_method()
local resolvConf = ngx.shared.resolvConf
local dnsCache = ngx.shared.dnsCache
local ttl = 60

-- 获取resolvConf中search(即域名后缀)
local function getDomainSuffix()
    local domain_suffix = resolvConf:get("search")
    if domain_suffix then
        local t_domain_suffix = json_decode(domain_suffix)
        if nkeys(t_domain_suffix) > 0 then
            return t_domain_suffix[1]
        end
    end
end

local function getServiceIP(host)
    local ipAddr = dnsCache:get(host)
    if ipAddr then
        return ipAddr
    else
        local nameserver = resolvConf:get("nameserver")
        if nameserver then
            local r, err = resolver:new{
                nameservers = json_decode(nameserver),
                no_random = true -- always start with first nameserver
            }

            if not r then
                ngx.log(ngx.ERR, "failed to instantiate the resolver: ", err)
                return
            end

            local domain_suffix = getDomainSuffix()
            local fullHost = host .. "." .. domain_suffix

            local answers, err = r:query(fullHost, {
                qtype = r.TYPE_A
            })
            if not answers then
                ngx.log(ngx.ERR, "failed to query the DNS server: ", err)
                return
            end

            if answers.errcode then
                ngx.log(ngx.ERR, "server returned error code: ", answers.errcode, ": ", answers.errstr, ": ", fullHost)
                return
            end

            if nkeys(answers) > 0 then
                ipAddr = answers[1].address
                dnsCache:set(host, ipAddr, ttl)
                return ipAddr
            end

            return
        end
    end
end

if method == "POST" then
    local headers = ngx.req.get_headers()
    local serverId = headers["server"]
    if serverId then
        local serviceName = "default"
        if uri == "/mail" then
            serviceName = "mail"
        end

        local backend_addr = "zone" .. tostring(serverId) .. "-" .. serviceName
        local backend_ip = getServiceIP(backend_addr)
        if not backend_ip then
            return ngx.exit(502)
        end
        ngx.ctx.backend_ip = backend_ip
    else
        return ngx.exit(403)
    end
else
    return ngx.exit(405)
end