使用动态 DNS 来完成 HTTP 请求

其实针对大多应用场景， DNS 是不会频繁变更的，使用 Nginx 默认的 resolver 配置方式就能解决。

在奇虎 360 企业版的应用场景下，需要支持的系统众多： win 、 centos 、 ubuntu 等，不同的操作系统获取 dns 的方法都不太一样。再加上我们使用 docker ，导致我们在容器内部获取 dns 变得更加难以准确。

如何能够让 Nginx 使用随时可以变化的 DNS 源，成为我们急待解决的问题。

当我们需要在某一个请求内部发起这样一个 http 查询，采用 proxy_pass 是不行的（依赖 resolver 的 dns ，如果 dns 有变化，必须要重新加载配置），并且由于 proxy_pass 不能直接设置 keepconn ，导致每次请求都是短链接，性能损失严重。

使用 resty.http ，目前这个库只支持 ip ： port 的方式定义 url ，其内部实现并没有支持 domain 解析。 resty.http 是支持 set_keepalive 完成长连接，这样我们只需要让他支持 dns 解析就能有完美解决方案了。

local resolver = require "resty.dns.resolver"
local http     = require "resty.http"

function get_domain_ip_by_dns( domain )
  -- 这里写死了 google 的域名服务 ip ，要根据实际情况做调整（例如放到指定配置或数据库中）
  local dns = "8.8.8.8"

  local r, err = resolver:new{
      nameservers = {dns, {dns, 53} },
      retrans = 5,  -- 5 retransmissions on receive timeout
      timeout = 2000,  -- 2 sec
  }

  if not r then
      return nil, "failed to instantiate the resolver: " .. err
  end

  local answers, err = r:query(domain)
  if not answers then
      return nil, "failed to query the DNS server: " .. err
  end

  if answers.errcode then
      return nil, "server returned error code: " .. answers.errcode .. ": " .. answers.errstr
  end

  for i, ans in ipairs(answers) do
    if ans.address then
      return ans.address
    end
  end

  return nil, "not founded"
end

function http_request_with_dns( url, param )
    -- get domain
    local domain = ngx.re.match(url, [[//([\S]+?)/]])
    domain = (domain and 1 == #domain and domain[1]) or nil
    if not domain then
        ngx.log(ngx.ERR, "get the domain fail from url:", url)
        return {status=ngx.HTTP_BAD_REQUEST}
    end

    -- add param
    if not param.headers then
        param.headers = {}
    end
    param.headers.Host = domain

    -- get domain's ip
    local domain_ip, err = get_domain_ip_by_dns(domain)
    if not domain_ip then
        ngx.log(ngx.ERR, "get the domain[", domain ,"] ip by dns failed:", err)
        return {status=ngx.HTTP_SERVICE_UNAVAILABLE}
    end

    -- http request
    local httpc = http.new()
    local temp_url = ngx.re.gsub(url, "//"..domain.."/", string.format("//%s/", domain_ip))

    local res, err = httpc:request_uri(temp_url, param)
    if err then
        return {status=ngx.HTTP_SERVICE_UNAVAILABLE}
    end

    -- httpc:request_uri 内部已经调用了 keepalive ，默认支持长连接
    -- httpc:set_keepalive(1000, 100)
    return res
end

转载请注明：爱开源 » 使用动态 DNS 来完成 HTTP 请求