migrate.go 702 B

123456789101112131415161718192021222324252627282930313233343536373839
  1. // Author: simon (ynwdlxm@163.com)
  2. // Date: 2025/5/28 16:26
  3. // Desc:
  4. package cmd
  5. import (
  6. "github.com/spf13/cobra"
  7. "github.com/runningwater/gohub/database/migrations"
  8. "github.com/runningwater/gohub/pkg/migrate"
  9. )
  10. var CmdMigrate = &cobra.Command{
  11. Use: "migrate",
  12. Short: "Run database migrations",
  13. }
  14. var CmdMigrateUp = &cobra.Command{
  15. Use: "up",
  16. Short: "Run up migrations",
  17. Run: runUp,
  18. }
  19. // 初始化命令
  20. func init() {
  21. CmdMigrate.AddCommand(CmdMigrateUp)
  22. }
  23. func runUp(cmd *cobra.Command, args []string) {
  24. migrator().Up()
  25. }
  26. func migrator() *migrate.Migrator {
  27. // 注册 database/migrations 目录下的所有迁移文件
  28. migrations.Initialize()
  29. return migrate.NewMigrator()
  30. }