48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
|
package domain
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"fmt"
|
|||
|
|
"net"
|
|||
|
|
"strings"
|
|||
|
|
|
|||
|
|
"github.com/oschwald/maxminddb-golang"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// GeoIP2Domain 抽象了 GeoIP2 国家数据库的领域逻辑。
|
|||
|
|
type GeoIP2Domain struct {
|
|||
|
|
db *maxminddb.Reader
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
type countryRecord struct {
|
|||
|
|
Country struct {
|
|||
|
|
ISOCode string `maxminddb:"iso_code"`
|
|||
|
|
} `maxminddb:"country"`
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// NewGeoIP2Domain 创建一个新的 GeoIP2 国家数据库领域对象。需要 maxminddb 文件的路径。
|
|||
|
|
func NewGeoIP2Domain(path string) (*GeoIP2Domain, error) {
|
|||
|
|
db, err := maxminddb.Open(path)
|
|||
|
|
if err != nil {
|
|||
|
|
return nil, fmt.Errorf("无法打开 geoip2 国家数据库 %s: %w", path, err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return &GeoIP2Domain{db}, nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// GetCountryCodeForIP 返回给定 IP 的两字母国家代码（ISO 3166-1）。
|
|||
|
|
func (d *GeoIP2Domain) GetCountryCodeForIP(ip net.IP) (string, error) {
|
|||
|
|
record := &countryRecord{}
|
|||
|
|
|
|||
|
|
err := d.db.Lookup(ip, record)
|
|||
|
|
if err != nil {
|
|||
|
|
return "", fmt.Errorf("无法查找 IP %s 的国家: %w", ip, err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return strings.ToLower(record.Country.ISOCode), nil
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Close 需要被调用以正确关闭内部的 maxminddb 数据库。
|
|||
|
|
func (d *GeoIP2Domain) Close() {
|
|||
|
|
d.db.Close()
|
|||
|
|
}
|