Honor query timeout in udp/tcp resolvers

folbricht · Jul 23, 2023 · e24bfa3 · e24bfa3
1 parent 75fca5a
commit e24bfa3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/dnsclient.go b/dnsclient.go
@@ -42,17 +42,17 @@ func NewDNSClient(id, endpoint, network string, opt DNSClientOptions) (*DNSClien
 	if opt.LocalAddr != nil {
 		switch network {
 		case "tcp":
-			dialer = &net.Dialer{LocalAddr: &net.TCPAddr{IP: opt.LocalAddr}}
+			dialer = &net.Dialer{LocalAddr: &net.TCPAddr{IP: opt.LocalAddr}, Timeout: opt.QueryTimeout}
 		case "udp":
-			dialer = &net.Dialer{LocalAddr: &net.UDPAddr{IP: opt.LocalAddr}}
+			dialer = &net.Dialer{LocalAddr: &net.UDPAddr{IP: opt.LocalAddr}, Timeout: opt.QueryTimeout}
 		}
 	}
-
 	client := &dns.Client{
 		Net:       network,
 		Dialer:    dialer,
 		TLSConfig: &tls.Config{},
 		UDPSize:   4096,
+		Timeout:   opt.QueryTimeout,
 	}
 	return &DNSClient{
 		id:       id,