Compare commits

...

6 Commits

3 changed files with 22 additions and 16 deletions

View File

@ -1,6 +1,12 @@
# QQWry [![Go Reference](https://pkg.go.dev/badge/github.com/xiaoqidun/qqwry.svg)](https://pkg.go.dev/github.com/xiaoqidun/qqwry)
Golang QQWry内存操作线程安全支持缓存的纯真IP查询库
Golang QQWry高性能纯真IP查询库。
# 使用须知
1. 仅支持ipv4查询。
2. city可能是城市也可能是国家。
3. area可能是区域也可能是运营商。
# 使用说明
@ -18,8 +24,8 @@ func main() {
panic(err)
}
// 从内存或缓存查询IP
country, area, err := qqwry.QueryIP("1.1.1.1")
log.Printf("国家:%s区域%s错误%v", country, area, err)
city, area, err := qqwry.QueryIP("1.1.1.1")
log.Printf("城市:%s区域%s错误%v", city, area, err)
}
```

View File

@ -24,8 +24,8 @@ const (
)
type cache struct {
Country string
Area string
City string
Area string
}
func byte3ToUInt32(data []byte) uint32 {
@ -43,9 +43,9 @@ func gb18030Decode(src []byte) string {
}
// QueryIP 从内存或缓存查询IP
func QueryIP(queryIp string) (country string, area string, err error) {
func QueryIP(queryIp string) (city string, area string, err error) {
if v, ok := ipCache.Load(queryIp); ok {
country = v.(cache).Country
city = v.(cache).City
area = v.(cache).Area
return
}
@ -99,19 +99,19 @@ func QueryIP(queryIp string) (country string, area string, err error) {
}
for i := posCA; i < dataLen; i++ {
if data[i] == 0 {
country = string(data[posCA:i])
city = string(data[posCA:i])
break
}
}
if mode != redirectMode2 {
posC += uint32(len(country) + 1)
posC += uint32(len(city) + 1)
}
areaPos = posC
case redirectMode2:
posCA := byte3ToUInt32(data[posM+1 : posM+4])
for i := posCA; i < dataLen; i++ {
if data[i] == 0 {
country = string(data[posCA:i])
city = string(data[posCA:i])
break
}
}
@ -120,11 +120,11 @@ func QueryIP(queryIp string) (country string, area string, err error) {
posCA := offset + 4
for i := posCA; i < dataLen; i++ {
if data[i] == 0 {
country = string(data[posCA:i])
city = string(data[posCA:i])
break
}
}
areaPos = offset + uint32(5+len(country))
areaPos = offset + uint32(5+len(city))
}
areaMode := data[areaPos]
if areaMode == redirectMode1 || areaMode == redirectMode2 {
@ -138,9 +138,9 @@ func QueryIP(queryIp string) (country string, area string, err error) {
}
}
}
country = gb18030Decode([]byte(country))
city = gb18030Decode([]byte(city))
area = gb18030Decode([]byte(area))
ipCache.Store(queryIp, cache{Country: country, Area: area})
ipCache.Store(queryIp, cache{City: city, Area: area})
return
}

View File

@ -12,9 +12,9 @@ func init() {
func TestQueryIP(t *testing.T) {
queryIp := "1.1.1.1"
country, area, err := QueryIP(queryIp)
city, area, err := QueryIP(queryIp)
if err != nil {
t.Fatal(err)
}
t.Log(country, area)
t.Logf("城市:%s区域%s", city, area)
}