From 82e2ba192892cc605d80e5e644b578a6f7233263 Mon Sep 17 00:00:00 2001
From: wangsiyuan <2392948297@qq.com>
Date: Fri, 16 Jun 2023 16:07:05 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20main.go?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.go | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/main.go b/main.go
index 13311f3..80bb825 100644
--- a/main.go
+++ b/main.go
@@ -7,22 +7,31 @@ import (
 	_ "github.com/go-sql-driver/mysql"
 	"log"
 	"net/http"
+	"strings"
 )
 
 func main() {
 	//modules.ImportDatabaseToHex()
 	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
 		//测试用ip地址
-		ip := "103.47.100.182"
+		//ip := "103.47.100.182"
 		//获取访问浏览器ip
-		//ip := strings.Split(r.RemoteAddr, ":")[0]
+		ip := strings.Split(r.RemoteAddr, ":")[0]
 		//fmt.Println("客户端IP地址:", ip)
+
+		if ip == "" {
+			w.Write([]byte("Invalid IP"))
+		}
+		IP_info = modules.GetIPInformation(ip)
 		log.Println("客户端ip：", ip)
+
 		//使用mmdb查询ip
-		//IpIformation := modules.MmdbQueryIP(ip)
+		IpIformation := modules.MmdbQueryIP(ip)
+
 		//使用MySql查询ip
-		dsn := modules.Readconfig()
-		IpIformation := modules.MysqlQuery(ip, dsn)
+		//dsn := modules.Readconfig()
+		//IpIformation := modules.MysqlQuery(ip, dsn)
+
 		log.Println("tag:main", IpIformation)
 		jsonData, err := json.Marshal(IpIformation)
 		if err != nil {