From 765f7d9c34fb2fab9d230352a459504fd29474c6 Mon Sep 17 00:00:00 2001
From: naiba <hi@nai.ba>
Date: Sat, 19 Oct 2024 15:37:12 +0800
Subject: [PATCH] =?UTF-8?q?=E9=BB=98=E8=AE=A4=E4=BD=BF=E7=94=A8=20passthro?=
 =?UTF-8?q?ugh=20resolver=20close=20#75?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: NewbieOrange <7200314+NewbieOrange@users.noreply.github.com>
---
 cmd/agent/main.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/cmd/agent/main.go b/cmd/agent/main.go
index 192df9d..8a919dc 100644
--- a/cmd/agent/main.go
+++ b/cmd/agent/main.go
@@ -29,6 +29,7 @@ import (
 	"google.golang.org/grpc"
 	"google.golang.org/grpc/credentials"
 	"google.golang.org/grpc/credentials/insecure"
+	"google.golang.org/grpc/resolver"
 
 	"github.com/nezhahq/agent/model"
 	fm "github.com/nezhahq/agent/pkg/fm"
@@ -63,7 +64,7 @@ var (
 	arch        string
 	client      pb.NezhaServiceClient
 	initialized bool
-	resolver    = &net.Resolver{PreferGo: true}
+	dnsResolver = &net.Resolver{PreferGo: true}
 )
 
 var agentCmd = &cobra.Command{
@@ -99,6 +100,7 @@ const (
 )
 
 func init() {
+	resolver.SetDefaultScheme("passthrough")
 	net.DefaultResolver.PreferGo = true // 使用 Go 内置的 DNS 解析器解析域名
 	net.DefaultResolver.Dial = func(ctx context.Context, network, address string) (net.Conn, error) {
 		d := net.Dialer{
@@ -862,7 +864,7 @@ func generateQueue(start int, size int) []int {
 
 func lookupIP(hostOrIp string) (string, error) {
 	if net.ParseIP(hostOrIp) == nil {
-		ips, err := resolver.LookupIPAddr(context.Background(), hostOrIp)
+		ips, err := dnsResolver.LookupIPAddr(context.Background(), hostOrIp)
 		if err != nil {
 			return "", err
 		}