Fix: nil pointer when use GeoIP2 to find IPv6 #43

zu1k · Apr 10, 2021 · 27600d4 · 27600d4
1 parent 10b36ef
commit 27600d4
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/internal/app/parse.go b/internal/app/parse.go
@@ -19,7 +19,7 @@ var (
 	geoip geoip2.GeoIP
 )
 
-// init ip db content
+// InitIPDB init ip db content
 func InitIPDB(ipdbtype ipdb.IPDBType) {
 	db = make([]ipdb.IPDB, 1)
 	switch ipdbtype {
@@ -41,7 +41,7 @@ func ParseIPs(ips []string) {
 	if len(db) > 1 {
 		db1 = db[1]
 	} else {
-		db1 = nil
+		db1 = db[0]
 	}
 	for _, ip := range ips {
 		v := tools.ValidIP(ip)
@@ -83,7 +83,7 @@ func ReplaceIPInString(str string) (result string) {
 	if len(db) > 1 {
 		db1 = db[1]
 	} else {
-		db1 = nil
+		db1 = db[0]
 	}
 
 	result = str