mirror of
https://github.com/dcarrillo/whatismyip.git
synced 2024-09-07 03:07:32 +00:00
86 lines
1.7 KiB
Go
86 lines
1.7 KiB
Go
|
package models
|
||
|
|
||
|
import (
|
||
|
"log"
|
||
|
"net"
|
||
|
|
||
|
"github.com/oschwald/maxminddb-golang"
|
||
|
)
|
||
|
|
||
|
type Record interface {
|
||
|
LookUp(ip net.IP)
|
||
|
}
|
||
|
|
||
|
type GeoRecord struct {
|
||
|
Country struct {
|
||
|
ISOCode string `maxminddb:"iso_code"`
|
||
|
Names map[string]string `maxminddb:"names"`
|
||
|
} `maxminddb:"country"`
|
||
|
City struct {
|
||
|
Names map[string]string `maxminddb:"names"`
|
||
|
} `maxminddb:"city"`
|
||
|
Location struct {
|
||
|
Latitude float64 `maxminddb:"latitude"`
|
||
|
Longitude float64 `maxminddb:"longitude"`
|
||
|
TimeZone string `maxminddb:"time_zone"`
|
||
|
} `maxminddb:"location"`
|
||
|
Postal struct {
|
||
|
Code string `maxminddb:"code"`
|
||
|
} `maxminddb:"postal"`
|
||
|
}
|
||
|
|
||
|
type ASNRecord struct {
|
||
|
AutonomousSystemNumber uint `maxminddb:"autonomous_system_number"`
|
||
|
AutonomousSystemOrganization string `maxminddb:"autonomous_system_organization"`
|
||
|
}
|
||
|
|
||
|
type geodb struct {
|
||
|
city *maxminddb.Reader
|
||
|
asn *maxminddb.Reader
|
||
|
}
|
||
|
|
||
|
var db geodb
|
||
|
|
||
|
func openMMDB(path string) *maxminddb.Reader {
|
||
|
db, err := maxminddb.Open(path)
|
||
|
if err != nil {
|
||
|
log.Fatal(err)
|
||
|
}
|
||
|
log.Printf("Database %s has been loaded\n", path)
|
||
|
|
||
|
return db
|
||
|
}
|
||
|
|
||
|
func Setup(cityPath string, asnPath string) {
|
||
|
db.city = openMMDB(cityPath)
|
||
|
db.asn = openMMDB(asnPath)
|
||
|
}
|
||
|
|
||
|
func CloseDBs() {
|
||
|
log.Printf("Closing dbs...")
|
||
|
if err := db.city.Close(); err != nil {
|
||
|
log.Printf("Error closing city db: %s", err)
|
||
|
}
|
||
|
if err := db.asn.Close(); err != nil {
|
||
|
log.Printf("Error closing ASN db: %s", err)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func (record *GeoRecord) LookUp(ip net.IP) error {
|
||
|
err := db.city.Lookup(ip, record)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (record *ASNRecord) LookUp(ip net.IP) error {
|
||
|
err := db.asn.Lookup(ip, record)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return nil
|
||
|
}
|