use mysql

models/init.go

@@ -10,6 +10,8 @@ import (
 	"fmt"
 	"go_blog/config"
 	"go_blog/pkg/util"
+	"log/slog"
+	"os"
 	"time"
 
 	"gorm.io/driver/mysql"
@@ -34,14 +36,21 @@ func InitDatabase(conf *config.Config) {
 			TablePrefix: conf.DataBase.Prefix,
 		},
 	})
-	if err != nil {
-		panic("数据库连接失败: " + err.Error())
-	}
 
+	if err != nil {
+		slog.Error("数据库连接失败", "error", err)
+		os.Exit(1)
+	}
+	sqlDB, _ := db.DB()
+	defer func() {
+		if err := sqlDB.Close(); err != nil {
+			slog.Error("数据库关闭异常", "error", err)
+		}
+	}()
 	DB = db
 	// 3. 自动迁移数据模型
-	DB.AutoMigrate(&Account{})
-	DB.AutoMigrate(&Content{})
+	db.AutoMigrate(&Account{})
+	db.AutoMigrate(&Content{})
 	// if err := db.AutoMigrate(&models.Article{}, &models.User{}); err != nil {
 	// 	panic("数据库迁移失败: " + err.Error())
 	// }