add

backend/country/src/db/postgres.go

@@ -0,0 +1,54 @@
+package db
+
+import (
+	"database/sql"
+	"fmt"
+	"os"
+	"time"
+	
+	_ "github.com/lib/pq"
+	"go.uber.org/zap"
+)
+
+var DB *sql.DB
+
+// 初始化数据库连接
+func Init() {
+	// 从环境变量获取数据库配置
+	dbHost := os.Getenv("DB_HOST")
+	dbPort := os.Getenv("DB_PORT")
+	dbUser := os.Getenv("DB_USER")
+	dbPassword := os.Getenv("DB_PASSWORD")
+	dbName := os.Getenv("DB_NAME")
+	zap.L().Info(
+		"💡 读取数据库配置",
+		zap.String("host", dbHost),
+		zap.String("port", dbPort),
+		zap.String("user", dbUser),
+		zap.String("dbname", dbName),
+	)
+
+	// 构建数据库连接字符串
+	connStr := fmt.Sprintf(
+		"host=%s port=%s user=%s password=%s dbname=%s sslmode=disable",
+		dbHost, dbPort, dbUser, dbPassword, dbName,
+	)
+
+	var err error
+	DB, err = sql.Open("postgres", connStr)
+	if err != nil {
+		zap.L().Panic("❌ 无法连接数据库", zap.Error(err))
+	}
+	
+	// 设置连接池参数
+    DB.SetMaxOpenConns(100)       // 最大打开连接数
+    DB.SetMaxIdleConns(20)        // 最大空闲连接数
+    DB.SetConnMaxLifetime(time.Hour) // 连接最大存活时间
+
+	// 验证数据库连接
+	if err := DB.Ping(); err != nil {
+		zap.L().Panic("❌ 数据库连接失败", zap.Error(err))
+	}
+
+	zap.L().Info("✅ 数据库连接验证成功")
+}