From a8c8cd767cd5ff3358517901fd8383709ecb5b2a Mon Sep 17 00:00:00 2001
From: dustoair <107600816+dustoair@users.noreply.github.com>
Date: Fri, 22 Jul 2022 18:24:15 +0800
Subject: [PATCH] domain adjust

---
 main.go | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index c25a02d..c21ffe4 100644
--- a/main.go
+++ b/main.go
@@ -20,6 +20,7 @@ func IPHandle(w http.ResponseWriter, req *http.Request) {
 	address := req.FormValue("address")
 	addressType := ""
 	ipDomain := ""
+	domain := ""
 	switch {
 	case address == "":
 		//没传address值 则拿请求ip来查询
@@ -39,6 +40,7 @@ func IPHandle(w http.ResponseWriter, req *http.Request) {
 			addressType = "self"
 		} else {
 			//提供正确域名 则查询ip
+			domain = address
 			ipDomain = ipAddress
 			address = ipAddress
 			addressType = "domain"
@@ -52,7 +54,7 @@ func IPHandle(w http.ResponseWriter, req *http.Request) {
 		ipInfo.Domain = ""
 	case "domain":
 		ipInfo.IP = ipDomain
-		ipInfo.Domain = address
+		ipInfo.Domain = domain
 	case "self":
 		ipInfo.IP = address
 		ipInfo.Domain = ""