通过通道查找DNS服务器列表中最快响应的服务器

积累知识，胜过积蓄金银！毕竟在Golang开发的过程中，会遇到各种各样的问题，往往都是一些细节知识点还没有掌握好而导致的，因此基础知识点的积累是很重要的。下面本文《通过通道查找DNS服务器列表中最快响应的服务器》，就带大家讲解一下知识点，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我正在尝试使用通道从一组服务器中查找对 dns 请求响应最快的 dns 服务器。

这是我的代码。我想知道这是否是正确的方法或者可以优化吗？

package main

import (
    "context"
    "fmt"
    "net"
    "time"
)

func main() {

    ch := make(chan string, 1)
    resolvers := []string{
        "1.1.1.1",
        "1.0.0.1",
        "8.8.4.4",
        "9.9.9.9",
        "8.8.8.8",
        "4.2.2.2",
    }

    for _, server := range resolvers {
        go func(server string, ch chan string) {
            // dns client
            r := &net.Resolver{
                PreferGo: true,
                Dial: func(ctx context.Context, network, address string) (net.Conn, error) {
                    d := net.Dialer{
                        Timeout: time.Millisecond * time.Duration(10000),
                    }
                    return d.DialContext(ctx, network, server+":53")
                },
            }
            _, _ = r.LookupHost(context.Background(), "www.google.com")
            ch <- server

        }(server, ch)
    }
    fmt.Printf("fastest is %s", <-ch)

}

正确答案

您可以按如下方式对请求进行计时并打印确切的计时。

package main

import (
    "context"
    "fmt"
    "net"
    "time"
)

type benchmark struct {
    server   string
    duration time.duration
}

func main() {
    ch := make(chan benchmark, 1)
    resolvers := []string{
        "1.1.1.1",
        "1.0.0.1",
        "8.8.4.4",
        "9.9.9.9",
        "8.8.8.8",
        "4.2.2.2",
    }

    for _, server := range resolvers {
        go func(server string, ch chan<- benchmark) {
            // dns client
            r := &net.resolver{
                prefergo: true,
                dial: func(ctx context.context, network, address string) (net.conn, error) {
                    d := net.dialer{
                        timeout: time.millisecond * time.duration(10000),
                    }
                    return d.dialcontext(ctx, network, server+":53")
                },
            }
            t1 := time.now()
            defer func() {
                ch <- benchmark{server: server, duration: time.since(t1)}
            }()
            _, _ = r.lookuphost(context.background(), "www.google.com")
        }(server, ch)
    }
    for _, _ = range resolvers {
        b := <-ch
        fmt.printf("server %s took %s\n", b.server, b.duration)
    }
}

defer 函数将在 lookup 函数返回后立即执行。有关 defer 的更多信息，您可能会喜欢我的博客。 https://marcofranssen.nl/the-use-of-defer-in-go

这在我的系统计时上显示如下。

$ go run main.go
server 9.9.9.9 took 49.836914ms
server 8.8.4.4 took 49.954711ms
server 4.2.2.2 took 49.996936ms
server 1.1.1.1 took 49.933049ms
server 8.8.8.8 took 49.94578ms
server 1.0.0.1 took 49.997141ms

你可以例如将结果放入新的基准切片中，您甚至可以根据持续时间对其进行排序。

另一项改进是多次运行此基准测试，然后取每个服务器的平均 pr 中值，以便随着时间的推移获得更准确的结果。

好了，本文到此结束，带大家了解了《通过通道查找DNS服务器列表中最快响应的服务器》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！