|
|
package GeoIPUtil
|
|
|
|
|
|
import (
|
|
|
"dsDataex/Utils/CacheUtil"
|
|
|
"dsDataex/Utils/RedisUtil"
|
|
|
"github.com/yinheli/qqwry"
|
|
|
"time"
|
|
|
)
|
|
|
|
|
|
var DB *qqwry.QQwry
|
|
|
|
|
|
var Province2260 map[string]string
|
|
|
var City2260 map[string][]string
|
|
|
|
|
|
func init() {
|
|
|
|
|
|
//打开 纯真 IP离线库
|
|
|
DB = qqwry.NewQQwry("GeoLite2/qqwry.dat")
|
|
|
|
|
|
//行政区划缓存
|
|
|
var sql = "SELECT area_code from t_dataex_gbt2260 where area_type=2 "
|
|
|
|
|
|
var list, _, _ = CacheUtil.Page(sql, 5000, 0)
|
|
|
|
|
|
Province2260 = make(map[string]string)
|
|
|
|
|
|
for no := 0; no < len(list); no++ {
|
|
|
Province2260[list[no]["area_code"].(string)] = list[no]["area_name"].(string)
|
|
|
}
|
|
|
|
|
|
City2260 = make(map[string][]string)
|
|
|
|
|
|
for no := 0; no < len(list); no++ {
|
|
|
City2260[list[no]["area_name"].(string)] = []string{list[no]["area_code"].(string), list[no]["area_name"].(string), "", ""}
|
|
|
}
|
|
|
|
|
|
sql = "SELECT area_code from t_dataex_gbt2260 where area_type=3 "
|
|
|
|
|
|
list, _, _ = CacheUtil.Page(sql, 5000, 0)
|
|
|
for no := 0; no < len(list); no++ {
|
|
|
var provinceCode = list[no]["area_code"].(string)[0:2] + "0000"
|
|
|
var provinceName = Province2260[provinceCode]
|
|
|
|
|
|
City2260[provinceName+list[no]["area_name"].(string)] = []string{provinceCode, provinceName, list[no]["area_code"].(string), list[no]["area_name"].(string)}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func GetGeo4IP(temp string) []string {
|
|
|
|
|
|
//add by zhangjun 2020-08-04 使用 Redis 缓存，缓解 IP文件并发读问题！！！
|
|
|
var city,err =RedisUtil.RedisClient.Get("Dataex_IP:"+temp).Result()
|
|
|
|
|
|
if err==nil && city != "" {
|
|
|
return City2260[ city ]
|
|
|
} else {
|
|
|
DB.Find(temp)
|
|
|
|
|
|
RedisUtil.RedisClient.Set("Dataex_IP:"+temp, DB.Country, 10*time.Hour)
|
|
|
|
|
|
//fmt.Println("qqwry :",DB.Country,DB.City)
|
|
|
return City2260[DB.Country]
|
|
|
}
|
|
|
}
|
