init: initial commit

initialize/gorm.go

@@ -0,0 +1,53 @@
+package initialize
+
+import (
+	"os"
+	"sundynix-go/global"
+	"sundynix-go/model/plant"
+	"sundynix-go/model/system"
+
+	"go.uber.org/zap"
+	"gorm.io/gorm"
+)
+
+// Gorm 根据全局配置中的数据库类型返回对应的 *gorm.DB 实例。
+// 该函数通过检查 global.Config.System.DbType 的值来决定使用哪种数据库连接。
+//
+// 返回值:
+//   - *gorm.DB: 返回对应数据库类型的 *gorm.DB 实例
+func Gorm() *gorm.DB {
+	switch global.Config.System.DbType {
+	case "mysql":
+		return GormMysql() // 返回 MySQL 数据库的 *gorm.DB 实例
+	case "pgsql":
+		return GromPgsql() // 返回 PostgreSQL 数据库的 *gorm.DB 实例
+	case "sqlite":
+		return GormSqlite() // 返回 SQLite 数据库的 *gorm.DB 实例
+	default:
+		return GormMysql() // 默认返回 MySQL 数据库的 *gorm.DB 实例
+	}
+}
+
+// MigrateTable 创建数据库表结构。
+func MigrateTable() {
+	db := global.DB
+	err := db.AutoMigrate(
+		system.User{},
+		system.Client{},
+		system.Role{},
+		system.Menu{},
+		system.SysOperationRecord{},
+		system.Oss{},
+
+		plant.MyPlant{},    //我的植物
+		plant.CarePlan{},   //植物养护计划
+		plant.CareTask{},   //植物养护任务
+		plant.CareRecord{}, //植物养护记录
+
+	)
+	if err != nil {
+		global.Logger.Error("Migrate table failed,err:", zap.Error(err))
+		os.Exit(0)
+	}
+	global.Logger.Info("Migrate table success")
+}