From 04fdc8387abaaa740055d7871cbe02ae7259a075 Mon Sep 17 00:00:00 2001
From: mogumc <mogumc@outlook.com>
Date: Thu, 14 Aug 2025 11:31:03 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E5=A4=8D=E7=94=A8Client=20=E4=BC=98?=
 =?UTF-8?q?=E5=8C=96=E7=BD=91=E7=BB=9C=E6=80=A7=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 monitoring/unit/ip.go | 72 ++++++++++++++++++++-----------------------
 1 file changed, 33 insertions(+), 39 deletions(-)

diff --git a/monitoring/unit/ip.go b/monitoring/unit/ip.go
index 3e48254..20d5bfd 100644
--- a/monitoring/unit/ip.go
+++ b/monitoring/unit/ip.go
@@ -10,45 +10,43 @@ import (
 	"time"
 )
 
-var userAgent = "curl/8.0.1"
-
-func ipv4Transport() *http.Transport {
-	return &http.Transport{
-		DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
-			d := net.Dialer{
-				Timeout:   15 * time.Second,
-				KeepAlive: 30 * time.Second,
-			}
-			return d.DialContext(ctx, "tcp4", addr) // 锁v4防止出现问题
+var (
+	ipv4HTTPClient = &http.Client{
+		Transport: &http.Transport{
+			DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
+				d := net.Dialer{
+					Timeout:   15 * time.Second,
+					KeepAlive: 30 * time.Second,
+				}
+				return d.DialContext(ctx, "tcp4", addr) // 锁v4防止出现问题
+			},
+			MaxIdleConns:          10,
+			IdleConnTimeout:       30 * time.Second,
+			TLSHandshakeTimeout:   10 * time.Second,
+			ExpectContinueTimeout: 1 * time.Second,
 		},
-		MaxIdleConns:          10,
-		IdleConnTimeout:       30 * time.Second,
-		TLSHandshakeTimeout:   10 * time.Second,
-		ExpectContinueTimeout: 1 * time.Second,
+		Timeout: 15 * time.Second,
 	}
-}
-
-func ipv6Transport() *http.Transport {
-	return &http.Transport{
-		DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
-			d := net.Dialer{
-				Timeout:   15 * time.Second,
-				KeepAlive: 30 * time.Second,
-			}
-			return d.DialContext(ctx, "tcp6", addr) // 锁v6防止出现问题
+	ipv6HTTPClient = &http.Client{
+		Transport: &http.Transport{
+			DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
+				d := net.Dialer{
+					Timeout:   15 * time.Second,
+					KeepAlive: 30 * time.Second,
+				}
+				return d.DialContext(ctx, "tcp6", addr) // 锁v6防止出现问题
+			},
+			MaxIdleConns:          10,
+			IdleConnTimeout:       30 * time.Second,
+			TLSHandshakeTimeout:   10 * time.Second,
+			ExpectContinueTimeout: 1 * time.Second,
 		},
-		MaxIdleConns:          10,
-		IdleConnTimeout:       30 * time.Second,
-		TLSHandshakeTimeout:   10 * time.Second,
-		ExpectContinueTimeout: 1 * time.Second,
+		Timeout: 15 * time.Second,
 	}
-}
+	userAgent = "curl/8.0.1"
+)
 
 func GetIPv4Address() (string, error) {
-	client := &http.Client{
-		Transport: ipv4Transport(),
-		Timeout:   15 * time.Second,
-	}
 
 	webAPIs := []string{
 		"https://www.visa.cn/cdn-cgi/trace",
@@ -67,7 +65,7 @@ func GetIPv4Address() (string, error) {
 			continue
 		}
 		req.Header.Set("User-Agent", userAgent)
-		resp, err := client.Do(req)
+		resp, err := ipv4HTTPClient.Do(req)
 		if err != nil {
 			continue
 		}
@@ -87,10 +85,6 @@ func GetIPv4Address() (string, error) {
 }
 
 func GetIPv6Address() (string, error) {
-	client := &http.Client{
-		Transport: ipv6Transport(), // 对于v6请求只使用v6
-		Timeout:   15 * time.Second,
-	}
 
 	webAPIs := []string{
 		"https://v6.ip.zxinc.org/info.php?type=json",
@@ -106,7 +100,7 @@ func GetIPv6Address() (string, error) {
 			continue
 		}
 		req.Header.Set("User-Agent", userAgent)
-		resp, err := client.Do(req)
+		resp, err := ipv6HTTPClient.Do(req)
 		if err != nil {
 			continue
 		}