first commit

internal/data/ipdb.go

@@ -0,0 +1,37 @@
+package data
+
+import (
+	"github.com/rs/zerolog/log"
+	"sync"
+	"thuanle.me/ip-info/configs"
+)
+import reader "github.com/oschwald/maxminddb-golang"
+
+type IpDb struct {
+	r *reader.Reader
+}
+
+var (
+	ins  *IpDb
+	once sync.Once
+)
+
+func Ins() *IpDb {
+	once.Do(func() {
+		ins = &IpDb{}
+		_ = ins.Reload()
+	})
+	return ins
+}
+
+func (d *IpDb) Reload() error {
+	r, err := reader.Open(configs.MmdbDbFile)
+	if err != nil {
+		log.Err(err).Msg("Failed to open mmdb")
+		return err
+	}
+	defer r.Close()
+
+	d.r = r
+	return nil
+}

internal/data/query.go

@@ -0,0 +1,23 @@
+package data
+
+import (
+	"fmt"
+	"net"
+)
+
+func (d *IpDb) Query(ipArg string) (map[string]any, *net.IPNet, error) {
+	ip := net.ParseIP(ipArg)
+	if ip == nil {
+		return nil, nil, fmt.Errorf("invalid IP: %s", ipArg)
+	}
+	// Get data of IP.
+	anyData := make(map[string]any)
+	recordNet, ok, err := d.r.LookupNetwork(ip, &anyData)
+	if err != nil {
+		return nil, nil, err
+	}
+	if !ok {
+		return nil, nil, fmt.Errorf("IP not found: %s", ipArg)
+	}
+	return anyData, recordNet, nil
+}