/api/ip_database.go

 1package api
 2
 3import (
 4	"log"
 5
 6	"github.com/oschwald/geoip2-golang"
 7)
 8
 9type IPDatabase struct {
10	Isp  *geoip2.Reader
11	City *geoip2.Reader
12}
13
14func OpenDatabases() *IPDatabase {
15	db := &IPDatabase{}
16
17	isp, err := geoip2.Open("GeoIP2-ISP.mmdb")
18	if err != nil {
19		log.Fatal(err)
20	}
21
22	city, err := geoip2.Open("GeoLite2-City.mmdb")
23	if err != nil {
24		log.Fatal(err)
25	}
26
27	db.Isp = isp
28	db.City = city
29
30	return db
31}