From 892ba586edf27a006eaa8eb40e879ad371e1b81e Mon Sep 17 00:00:00 2001
From: zu1k <i@lgf.im>
Date: Sun, 16 Aug 2020 12:16:54 +0800
Subject: [PATCH] fix for go test

---
 app/global.go | 2 +-
 main.go       | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/app/global.go b/app/global.go
index 85b8541..d0e52b6 100644
--- a/app/global.go
+++ b/app/global.go
@@ -7,6 +7,6 @@ var (
 	ProjectName = "proxypool"
 )
 
-func init() {
+func InitGeoIpDB() {
 	GeoIp = proxy.NewGeoIP("assets/GeoLite2-City.mmdb")
 }
diff --git a/main.go b/main.go
index ceac19e..32b4d49 100644
--- a/main.go
+++ b/main.go
@@ -23,6 +23,8 @@ func main() {
 			fmt.Println(err)
 		}
 	}
+	app.InitGeoIpDB()
+
 	go app.Cron()
 	fmt.Println("Do the first crawl...")
 	app.CrawlGo()