Go语言自定义DNS解析器实践

FunTester

发布于 2022-04-01 09:39:44

2.4K0

发布于 2022-04-01 09:39:44

文章被收录于专栏：FunTester

写完了Java自定义DNS解析器实践和Java自定义DNS解析器负载均衡实现之后，自然也需要对Go语言的测试拓展相同的功能，走了一些弯路，最终目的还是实现了。今天分享一下Go语言HTTP接口测试自定义DNS解析的实现。这里只用http库作为演示，fasthttp以后有机会再尝试分享。

设置net.Dialer

这里先分享一下net.Dialer的设置方式。net.Dialer翻译为拨号器，我的理解是HTTP连接的建立类，类似于Java语言HttpClient库里面的org.apache.http.impl.conn.PoolingHttpClientConnectionManager部分功能。（HttpClient4.5x以后推荐这个）。

// clients 初始化请求客户端
// @Description:
// @return fhttp.Client
func clients() http.Client {
 dialer := &net.Dialer{
  Timeout: 1 * time.Second,
 }

 return http.Client{
  Timeout: time.Duration(5) * time.Second, //超时时间
  Transport: &http.Transport{
   MaxIdleConnsPerHost:   200,   //单个路由最大空闲连接数
   MaxConnsPerHost:       10000, //单个路由最大连接数
   IdleConnTimeout:       90 * time.Second,
   TLSHandshakeTimeout:   10 * time.Second,
   ExpectContinueTimeout: 1 * time.Second,
   DialContext:           dialer.DialContext,
  },
 }
}

奇怪的知识点

在本次学习的过程中，发现了Go语言的net/http库还支持了另外一个有趣的功能，就是绑定DNS服务IP，这个有时候也能部分解决将固定域名的请求发送到固定机器的需求。

简单设置的方法如下：

 dialer := &net.Dialer{
  Timeout: 1 * time.Second,
 }
 dialer.Resolver = &net.Resolver{
  Dial: func(ctx context.Context, network, address string) (net.Conn, error) {
   return dialer.DialContext(ctx, "tcp", "114.114.114.114:53") // 通过tcp请求nameserver解析域名
  },
 }

其他设置项同上。

自定义net.Dialer

在http.Transport创建参数中，有一个DialContext参数就是指定用于创建未加密 TCP 连接的拨号函数。参数类型是func(ctx context.Context, network, addr string) (net.Conn, error)方法，这里我习惯称之为闭包。我们只要实现这个方法即可。

下面这个例子我设置了两个IP来测试负载均衡（下期出文字版和视频版）。

DialContext: func(ctx context.Context, network, address string) (net.Conn, error) {
    host, port, err := net.SplitHostPort(address)
    if err != nil {
     return nil, err
    }
    //通过自定义nameserver获取域名解析的IP
    //ips, _ := dialer.Resolver.LookupHost(ctx, host)
    //for _, s := range ips {
    // log.Println(s)
    //}

    // 创建链接
    if host == "fun.tester" {
     ip := "127.0.0.1"
     log.Println(ip)
     conn, err := dialer.DialContext(ctx, network, ip+":"+port)
     if err == nil {
      return conn, nil
     }
    }
    return dialer.DialContext(ctx, network, address)
   },

中间有一部分通过自定义的nameserver获取到域名解析结果IP的过程，注释掉了，留着以后用。

测试

测试用例如下：

// TestFaast
// @Description: 测试自定义DNS解析功能
// @param t
func TestFaast(t *testing.T) {
 url := "http://fun.tester:12345/test"
 get := fhttp.Get(url, nil)
 //for i := 0; i < 10; i++ {
 // //go log.Println(string(fhttp.Response(get)))
 // go func() {
 //  log.Println(string(fhttp.Response(get)))
 // }()
 //}
 response := fhttp.Response(get)
 log.Println(string(response))
}

控制台输出：

=== RUN   TestFaast
2022/02/07 15:34:47 127.0.0.1
2022/02/07 15:34:48 Have Fun ~ Tester ！
--- PASS: TestFaast (0.31s)
PASS

测试服务

测试服务基于moco_FunTester框架写的简单服务，脚本如下：

    static void main(String[] args) {
        def util = new ArgsUtil(args)
        def server = getServerNoLog(util.getIntOrdefault(0, 12345))
        server.response(delay(textRes("Have Fun ~ Tester ！"), 100))
        def run = run(server)
        waitForKey("fan")
        run.stop()
    }

本文参与腾讯云自媒体同步曝光计划，分享自微信公众号。

原始发表：2022-02-17，如有侵权请联系 cloudcommunity@tencent.com 删除

dns