初始化

server/config/db.go

@@ -0,0 +1,64 @@
+package config
+
+import (
+	"fmt"
+	"log"
+	"time"
+
+	"database/sql"
+
+	_ "github.com/go-sql-driver/mysql"
+)
+
+// DBConfig 数据库配置
+var DBConfig = struct {
+	Username string
+	Password string
+	Host     string
+	Port     string
+	DBName   string
+}{
+	Username: "root",
+	Password: "root",
+	Host:     "127.0.0.1",
+	Port:     "3306",
+	DBName:   "nl_blog",
+}
+
+// DB 全局数据库连接池
+var DB *sql.DB
+
+// InitDB 初始化数据库连接
+func InitDB() {
+	// 构建DSN
+	dsn := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8mb4&parseTime=True&loc=Local",
+		DBConfig.Username, DBConfig.Password, DBConfig.Host, DBConfig.Port, DBConfig.DBName)
+
+	// 打开数据库连接
+	var err error
+	DB, err = sql.Open("mysql", dsn)
+	if err != nil {
+		log.Fatalf("Failed to open database connection: %v", err)
+	}
+
+	// 配置连接池
+	DB.SetMaxOpenConns(25)                  // 最大打开连接数
+	DB.SetMaxIdleConns(5)                   // 最大空闲连接数
+	DB.SetConnMaxLifetime(5 * time.Minute)  // 连接最大生命周期
+	DB.SetConnMaxIdleTime(30 * time.Second) // 连接最大空闲时间
+
+	// 测试连接
+	if err := DB.Ping(); err != nil {
+		log.Fatalf("Failed to ping database: %v", err)
+	}
+
+	log.Println("Database connection established successfully!")
+}
+
+// CloseDB 关闭数据库连接
+func CloseDB() {
+	if DB != nil {
+		DB.Close()
+		log.Println("Database connection closed!")
+	}
+}