fix: cli数据库迁移

tnb-labs · Dec 29, 2024 · cff8925 · cff8925
1 parent 40d6b2c
commit cff8925
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 4 deletions.
diff --git a/cmd/cli/wire_gen.go b/cmd/cli/wire_gen.go
diff --git a/internal/app/cli.go b/internal/app/cli.go
@@ -4,20 +4,28 @@ import (
 	"context"
 	"os"
 
+	"github.com/go-gormigrate/gormigrate/v2"
 	"github.com/urfave/cli/v3"
 )
 
 type Cli struct {
-	cmd *cli.Command
+	cmd      *cli.Command
+	migrator *gormigrate.Gormigrate
 }
 
-func NewCli(cmd *cli.Command) *Cli {
+func NewCli(cmd *cli.Command, migrator *gormigrate.Gormigrate) *Cli {
 	IsCli = true
 	return &Cli{
-		cmd: cmd,
+		cmd:      cmd,
+		migrator: migrator,
 	}
 }
 
 func (r *Cli) Run() error {
+	// migrate database
+	if err := r.migrator.Migrate(); err != nil {
+		return err
+	}
+
 	return r.cmd.Run(context.Background(), os.Args)
 }