Initial commit

cmd/server/migrate.go

@@ -0,0 +1,35 @@
+package main
+
+import (
+	"context"
+	"ent-scaffold/internal/db"
+	"ent-scaffold/internal/global"
+
+	"github.com/spf13/cobra"
+	"go.uber.org/zap"
+)
+
+var migrateCmd = &cobra.Command{
+	Use:   "migrate",
+	Short: "执行数据库 Schema 自动迁移",
+	Run: func(cmd *cobra.Command, args []string) {
+		err := db.Init(global.Config.Database.Driver, global.Config.Database.Dsn())
+		if err != nil {
+			global.Log.Fatal("连接数据库失败", zap.Error(err))
+		}
+		defer db.Close()
+		global.Log.Info("开始执行数据库 Schema 自动迁移...")
+
+		ctx := context.Background()
+		// 直接使用 db.Client 跨包调用
+		if err := db.Client.Schema.Create(ctx); err != nil {
+			global.Log.Fatal("Schema 迁移失败", zap.Error(err))
+		}
+		
+		global.Log.Info("数据库 Schema 迁移执行完毕！")
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(migrateCmd)
+}