You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
64 lines
1.4 KiB
64 lines
1.4 KiB
package ip2region_v2
|
|
|
|
import (
|
|
_ "embed"
|
|
"errors"
|
|
"github.com/dtapps/go-library/utils/gostring"
|
|
"net"
|
|
"strings"
|
|
)
|
|
|
|
// QueryResult 返回
|
|
type QueryResult struct {
|
|
Ip string `json:"ip,omitempty"` // ip
|
|
Country string `json:"country,omitempty"` // 国家
|
|
Province string `json:"province,omitempty"` // 省份
|
|
City string `json:"city,omitempty"` // 城市
|
|
Operator string `json:"operator,omitempty"` // 运营商
|
|
}
|
|
|
|
func (c *Client) Query(ipAddress net.IP) (result QueryResult, err error) {
|
|
|
|
// 备注：并发使用，用整个 xdb 缓存创建的 searcher 对象可以安全用于并发。
|
|
|
|
str, err := c.db.SearchByStr(ipAddress.String())
|
|
if err != nil {
|
|
return QueryResult{}, err
|
|
}
|
|
|
|
split := gostring.Split(str, "|")
|
|
if len(split) <= 0 {
|
|
return QueryResult{}, err
|
|
}
|
|
|
|
result.Ip = ipAddress.String()
|
|
|
|
result.Country = split[0]
|
|
if result.Country == "0" {
|
|
result.Country = ""
|
|
}
|
|
result.Province = split[2]
|
|
if result.Province == "0" {
|
|
result.Province = ""
|
|
}
|
|
result.City = split[3]
|
|
if result.City == "0" {
|
|
result.City = ""
|
|
}
|
|
result.Operator = split[4]
|
|
if result.Operator == "0" {
|
|
result.Operator = ""
|
|
}
|
|
|
|
return result, err
|
|
}
|
|
|
|
// QueryIP ip地址查询对应归属地信息
|
|
func (c *Client) QueryIP(ipAddressStr string) (result QueryResult, err error) {
|
|
arrIpv4 := strings.Split(ipAddressStr, ".")
|
|
if len(arrIpv4) == 4 {
|
|
return c.Query(net.ParseIP(ipAddressStr))
|
|
}
|
|
return QueryResult{}, errors.New("不是IPV4")
|
|
}
|