mirror of
https://github.com/SagerNet/sing.git
synced 2025-04-04 20:37:40 +03:00
33 lines
496 B
Go
33 lines
496 B
Go
package geoip
|
|
|
|
import (
|
|
"net"
|
|
"strings"
|
|
"sync"
|
|
|
|
"github.com/oschwald/geoip2-golang"
|
|
)
|
|
|
|
var (
|
|
mmdb *geoip2.Reader
|
|
loadOnce sync.Once
|
|
loadErr error
|
|
)
|
|
|
|
func LoadMMDB(path string) error {
|
|
if loadErr != nil {
|
|
loadOnce = sync.Once{}
|
|
}
|
|
loadOnce.Do(func() {
|
|
mmdb, loadErr = geoip2.Open(path)
|
|
})
|
|
return loadErr
|
|
}
|
|
|
|
func Match(code string, ip net.IP) bool {
|
|
country, err := mmdb.Country(ip)
|
|
if err != nil {
|
|
return false
|
|
}
|
|
return strings.EqualFold(country.Country.IsoCode, code)
|
|
}
|