dcarrillo/whatismyip
1
0
Fork 0
mirror of https://github.com/dcarrillo/whatismyip.git synced 2025-07-03 08:59:25 +00:00
Code Issues Projects Releases Wiki Activity

Add CI and go-report badges. Add comments to make golint happy

Browse Source
This commit is contained in:
Daniel Carrillo
2021-11-30 17:21:01 +01:00
parent 9b10052cd1
commit 2d330a99b7
8 changed files with 24 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
models/geo.go
View File

@ -7,10 +7,12 @@ import (
"github.com/oschwald/maxminddb-golang"
)
// Record is the interface to be implemented for record operations
type Record interface {
LookUp(ip net.IP)
}
// GeoRecord is the model for City database
type GeoRecord struct {
Country struct {
ISOCode string `maxminddb:"iso_code"`
@ -29,6 +31,7 @@ type GeoRecord struct {
} `maxminddb:"postal"`
}
// ASNRecord is the model for ASN database
type ASNRecord struct {
AutonomousSystemNumber uint `maxminddb:"autonomous_system_number"`
AutonomousSystemOrganization string `maxminddb:"autonomous_system_organization"`
@ -51,11 +54,13 @@ func openMMDB(path string) *maxminddb.Reader {
return db
}
// Setup opens all Geolite2 databases
func Setup(cityPath string, asnPath string) {
db.city = openMMDB(cityPath)
db.asn = openMMDB(asnPath)
}
// CloseDBs unmaps from memory and frees resources to the filesystem
func CloseDBs() {
log.Printf("Closing dbs...")
if err := db.city.Close(); err != nil {
@ -66,6 +71,7 @@ func CloseDBs() {
}
}
// LookUp an IP and get city data
func (record *GeoRecord) LookUp(ip net.IP) error {
err := db.city.Lookup(ip, record)
if err != nil {
@ -75,6 +81,7 @@ func (record *GeoRecord) LookUp(ip net.IP) error {
return nil
}
// LookUp an IP and get ASN data
func (record *ASNRecord) LookUp(ip net.IP) error {
err := db.asn.Lookup(ip, record)
if err != nil {