restructure

migrate.go

@@ -0,0 +1,103 @@
+package atom
+
+import (
+	"log"
+	"sort"
+
+	"github.com/rogeecn/atom/container"
+	"github.com/rogeecn/atom/contracts"
+
+	"github.com/go-gormigrate/gormigrate/v2"
+	"github.com/spf13/cobra"
+	"go.uber.org/dig"
+	"gorm.io/gorm"
+)
+
+func WithMigration(rootCmd *cobra.Command) *cobra.Command {
+	rootCmd.AddCommand(migrateCmd)
+	migrateCmd.AddCommand(migrateUpCmd)
+	migrateCmd.AddCommand(migrateDownCmd)
+
+	migrateCmd.PersistentFlags().StringVar(&migrateToId, "to", "", "migration to id")
+
+	return rootCmd
+}
+
+// migrateCmd represents the migrate command
+var migrateCmd = &cobra.Command{
+	Use:   "migrate",
+	Short: "migrate database tables",
+	Long:  `migrate database tables`,
+}
+
+var migrateToId string
+
+// MigrationInfo http service container
+type MigrationInfo struct {
+	dig.In
+
+	DB         *gorm.DB
+	Migrations []contracts.Migration `group:"migrations"`
+}
+
+// migrateUpCmd represents the migrateUp command
+var migrateUpCmd = &cobra.Command{
+	Use:   "up",
+	Short: "migrate up database tables",
+	Long:  `migrate up database tables`,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		return container.Container.Invoke(func(mi MigrationInfo) error {
+			m := gormigrate.New(mi.DB, gormigrate.DefaultOptions, sortedMigrations(mi.Migrations))
+			if len(migrateToId) > 0 {
+				log.Printf("migrate up to [%s]\n", migrateToId)
+				return m.MigrateTo(migrateToId)
+			}
+			return m.Migrate()
+		})
+	},
+	PostRun: func(cmd *cobra.Command, args []string) {
+		log.Println("BINGO! migrate up done")
+	},
+}
+
+// migrateDownCmd represents the migrateDown command
+var migrateDownCmd = &cobra.Command{
+	Use:   "down",
+	Short: "migrate down database tables",
+	Long:  `migrate down database tables`,
+	RunE: func(cmd *cobra.Command, args []string) error {
+		return container.Container.Invoke(func(mi MigrationInfo) error {
+			m := gormigrate.New(mi.DB, gormigrate.DefaultOptions, sortedMigrations(mi.Migrations))
+
+			if len(migrateToId) > 0 {
+				log.Printf("migrate down to [%s]\n", migrateToId)
+				return m.RollbackTo(migrateToId)
+			}
+			return m.RollbackLast()
+		})
+	},
+	PostRun: func(cmd *cobra.Command, args []string) {
+		log.Println("BINGO! migrate down done")
+	},
+}
+
+func sortedMigrations(ms []contracts.Migration) []*gormigrate.Migration {
+	migrationKeys := []string{}
+	migrationMaps := make(map[string]*gormigrate.Migration)
+	for _, m := range ms {
+		migrationKeys = append(migrationKeys, m.ID())
+		migrationMaps[m.ID()] = &gormigrate.Migration{
+			ID:       m.ID(),
+			Migrate:  m.Up,
+			Rollback: m.Down,
+		}
+	}
+	sort.Strings(migrationKeys)
+
+	migrations := []*gormigrate.Migration{}
+	for _, key := range migrationKeys {
+		migrations = append(migrations, migrationMaps[key])
+	}
+
+	return migrations
+}