104 lines
2.6 KiB
Go
104 lines
2.6 KiB
Go
package cmd
|
|
|
|
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
|
|
}
|