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

写完了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()
    }

Have Fun ~ Tester ！

• FunTester2021年总结
• FunTester原创大赏
• Groovy语言学习笔记大赏【FunTester】
• getInteger还是getIntValue，这是一个问题
• 固定QPS压测模式探索
• 如何突破职业瓶颈
• 如何选择API测试工具
• 性能框架哪家强—JMeter、K6、locust、FunTester横向对比
• Groovy中的list
• Java&Go三种HTTP客户端性能测试
• 性能测试如何减少本机误差
• 创建Java守护线程
• 分段随机实践—模拟线上流量
• 千万级日志回放引擎设计稿