Update.

Author: xOS

model/config.go

@@ -99,6 +99,9 @@ type Config struct {
 	DatabaseType     string // 数据库类型，可选：sqlite、badger
 	DatabaseLocation string // 数据库文件路径
 
+	// GeoIP 配置
+	GeoIPDB string `yaml:"geoipdb" koanf:"geoipdb"` // 自定义 GeoIP 数据库路径（MMDB 格式），留空则使用内嵌数据库
+
 	k        *koanf.Koanf
 	filePath string
 }

pkg/geoip/geoip.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"log"
 	"net"
+	"os"
 	"strings"
 	"sync"
 
@@ -14,11 +15,7 @@ import (
 //go:embed geoip.db
 var geoDBFS embed.FS
 
-var (
-	dbData []byte
-	err    error
-)
-
+// IPInfo 表示 IP 地理位置信息
 type IPInfo struct {
 	Country       string `maxminddb:"country"`
 	CountryName   string `maxminddb:"country_name"`
@@ -28,30 +25,104 @@ type IPInfo struct {
 
 var (
 	db     *maxminddb.Reader
+	dbMu   sync.RWMutex
 	dbOnce sync.Once
+	dbPath string // 当前加载的外部数据库路径
 )
 
-func init() {
-	dbData, err = geoDBFS.ReadFile("geoip.db")
-	if err != nil {
-		log.Printf("NG>> Failed to open geoip database: %v", err)
+// Init 初始化 GeoIP 数据库
+// 如果 path 不为空，优先从外部文件加载；否则使用内嵌数据库
+func Init(path string) error {
+	dbMu.Lock()
+	defer dbMu.Unlock()
+
+	// 关闭旧的数据库
+	if db != nil {
+		db.Close()
+		db = nil
 	}
-}
 
-// initDB 初始化GeoIP数据库，只执行一次
-func initDB() {
-	if dbData != nil {
-		var err error
-		db, err = maxminddb.FromBytes(dbData)
+	// 尝试从外部 MMDB 文件加载
+	if path != "" {
+		reader, err := loadFromFile(path)
 		if err != nil {
-			log.Printf("NG>> Failed to initialize geoip database: %v", err)
+			log.Printf("NG>> 无法加载外部 GeoIP 数据库 %s: %v，回退到内嵌数据库", path, err)
+		} else {
+			db = reader
+			dbPath = path
+			log.Printf("NG>> 已加载外部 GeoIP 数据库: %s", path)
+			return nil
 		}
 	}
+
+	// 回退到内嵌数据库
+	reader, err := loadEmbedded()
+	if err != nil {
+		return fmt.Errorf("无法加载内嵌 GeoIP 数据库: %w", err)
+	}
+	db = reader
+	dbPath = ""
+	log.Printf("NG>> 已加载内嵌 GeoIP 数据库")
+	return nil
+}
+
+// Reload 重新加载 GeoIP 数据库（用于配置热更新）
+func Reload(path string) error {
+	return Init(path)
+}
+
+// loadFromFile 从外部 MMDB 文件加载数据库
+func loadFromFile(path string) (*maxminddb.Reader, error) {
+	if _, err := os.Stat(path); err != nil {
+		return nil, fmt.Errorf("文件不存在: %w", err)
+	}
+	reader, err := maxminddb.Open(path)
+	if err != nil {
+		return nil, fmt.Errorf("打开 MMDB 文件失败: %w", err)
+	}
+	return reader, nil
+}
+
+// loadEmbedded 从内嵌资源加载数据库
+func loadEmbedded() (*maxminddb.Reader, error) {
+	data, err := geoDBFS.ReadFile("geoip.db")
+	if err != nil {
+		return nil, fmt.Errorf("读取内嵌数据失败: %w", err)
+	}
+	reader, err := maxminddb.FromBytes(data)
+	if err != nil {
+		return nil, fmt.Errorf("解析内嵌数据失败: %w", err)
+	}
+	return reader, nil
+}
+
+// ensureInit 确保数据库已初始化（兼容未显式调用 Init 的场景）
+func ensureInit() {
+	dbOnce.Do(func() {
+		dbMu.RLock()
+		initialized := db != nil
+		dbMu.RUnlock()
+		if !initialized {
+			if err := Init(""); err != nil {
+				log.Printf("NG>> GeoIP 自动初始化失败: %v", err)
+			}
+		}
+	})
 }
 
+// GetDBPath 返回当前加载的数据库路径，空字符串表示使用内嵌数据库
+func GetDBPath() string {
+	dbMu.RLock()
+	defer dbMu.RUnlock()
+	return dbPath
+}
+
+// Lookup 查询 IP 对应的国家/地区代码
 func Lookup(ip net.IP, record *IPInfo) (string, error) {
-	// 确保数据库只初始化一次
-	dbOnce.Do(initDB)
+	ensureInit()
+
+	dbMu.RLock()
+	defer dbMu.RUnlock()
 
 	if db == nil {
 		return "", fmt.Errorf("geoip database not available")

service/singleton/singleton.go

@@ -22,6 +22,7 @@ import (
 
 	"github.com/xos/serverstatus/db"
 	"github.com/xos/serverstatus/model"
+	"github.com/xos/serverstatus/pkg/geoip"
 	"github.com/xos/serverstatus/pkg/utils"
 )
 
@@ -238,6 +239,12 @@ func LoadSingleton() {
 		}
 	}()
 
+	// 初始化 GeoIP 数据库
+	log.Println("正在初始化 GeoIP 数据库...")
+	if err := geoip.Init(Conf.GeoIPDB); err != nil {
+		log.Printf("GeoIP 数据库初始化失败: %v", err)
+	}
+
 	// 加载通知服务
 	log.Println("正在加载通知服务...")
 	func() {