From ccbd959007fa6654675f54a57d92ef8d0c96cd81 Mon Sep 17 00:00:00 2001
From: Toby <tobyxdd@gmail.com>
Date: Thu, 16 Dec 2021 14:06:13 -0800
Subject: [PATCH] chore: setResolver

---
 cmd/client.go | 9 +--------
 cmd/server.go | 9 +--------
 2 files changed, 2 insertions(+), 16 deletions(-)

diff --git a/cmd/client.go b/cmd/client.go
index abf5d14..5c4c220 100644
--- a/cmd/client.go
+++ b/cmd/client.go
@@ -1,7 +1,6 @@
 package main
 
 import (
-	"context"
 	"crypto/tls"
 	"crypto/x509"
 	"io"
@@ -30,13 +29,7 @@ func client(config *clientConfig) {
 	logrus.WithField("config", config.String()).Info("Client configuration loaded")
 	// Resolver
 	if len(config.Resolver) > 0 {
-		net.DefaultResolver = &net.Resolver{
-			PreferGo: true,
-			Dial: func(ctx context.Context, network, address string) (net.Conn, error) {
-				d := net.Dialer{}
-				return d.DialContext(ctx, "udp", config.Resolver)
-			},
-		}
+		setResolver(config.Resolver)
 	}
 	// TLS
 	tlsConfig := &tls.Config{
diff --git a/cmd/server.go b/cmd/server.go
index a441f97..fa0347a 100644
--- a/cmd/server.go
+++ b/cmd/server.go
@@ -1,7 +1,6 @@
 package main
 
 import (
-	"context"
 	"crypto/tls"
 	"github.com/lucas-clemente/quic-go"
 	"github.com/lucas-clemente/quic-go/congestion"
@@ -25,13 +24,7 @@ func server(config *serverConfig) {
 	logrus.WithField("config", config.String()).Info("Server configuration loaded")
 	// Resolver
 	if len(config.Resolver) > 0 {
-		net.DefaultResolver = &net.Resolver{
-			PreferGo: true,
-			Dial: func(ctx context.Context, network, address string) (net.Conn, error) {
-				d := net.Dialer{}
-				return d.DialContext(ctx, "udp", config.Resolver)
-			},
-		}
+		setResolver(config.Resolver)
 	}
 	// Load TLS config
 	var tlsConfig *tls.Config