mirror of
https://github.com/xiaoqidun/qqwry.git
synced 2023-06-27 14:38:22 +08:00
Compare commits
6 Commits
3caa55c1c7
...
514f7fcd37
Author | SHA1 | Date | |
---|---|---|---|
514f7fcd37 | |||
17b359bd3d | |||
c1803c9752 | |||
ebb890b8d1 | |||
021ab50216 | |||
a95d612bbe |
12
README.md
12
README.md
@ -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)
|
||||
}
|
||||
```
|
||||
|
||||
|
22
qqwry.go
22
qqwry.go
@ -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
|
||||
}
|
||||
|
||||
|
@ -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)
|
||||
}
|
||||
|
Loading…
x
Reference in New Issue
Block a user