first commit

models/geo.go

@@ -0,0 +1,85 @@
+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
+}