opencode/yao
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[add] migrate command reset & update schema

Browse source
This commit is contained in:
Max 2022-08-22 15:10:07 +08:00
parent 9d19d32fef
commit 3d21a9e19e
2 changed files with 41 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

45
cmd/migrate.go
View file

@ -15,6 +15,7 @@ import (
var name string
var force bool = false
var resetModel bool = false
var migrateCmd = &cobra.Command{
Use: "migrate",
Short: L("Update database schema"),
@ -43,23 +44,57 @@ var migrateCmd = &cobra.Command{
if name != "" {
mod, has := gou.Models[name]
if has {
mod.Migrate(true)
fmt.Printf(color.WhiteString(L("Update schema model: %s (%s) "), mod.Name, mod.MetaData.Table.Name) + "\t")
if !has {
fmt.Println(color.RedString(L("Model: %s does not exits"), name))
return
}
if resetModel {
err := mod.DropTable()
if err != nil {
fmt.Printf(color.RedString(L("FAILURE\n%s"), err.Error()) + "\n")
return
}
}
err := mod.Migrate(false)
if err != nil {
fmt.Printf(color.RedString(L("FAILURE\n%s"), err.Error()) + "\n")
return
}
fmt.Printf(color.GreenString(L("SUCCESS")) + "\n")
return
}
// Do Stuff Here
for _, mod := range gou.Models {
fmt.Println(color.GreenString(L("Update schema model: %s (%s) "), mod.Name, mod.MetaData.Table.Name))
mod.Migrate(true)
fmt.Printf(color.WhiteString(L("Update schema model: %s (%s) "), mod.Name, mod.MetaData.Table.Name) + "\t")
if resetModel {
err := mod.DropTable()
if err != nil {
fmt.Printf(color.RedString(L("FAILURE\n%s"), err.Error()) + "\n")
continue
}
}
err := mod.Migrate(false)
if err != nil {
fmt.Printf(color.RedString(L("FAILURE\n%s"), err.Error()) + "\n")
continue
}
fmt.Printf(color.GreenString(L("SUCCESS")) + "\n")
}
fmt.Println(color.GreenString(L("✨DONE✨")))
// fmt.Println(color.GreenString(L("✨DONE✨")))
},
}
func init() {
migrateCmd.PersistentFlags().StringVarP(&name, "name", "n", "", L("Model name"))
migrateCmd.PersistentFlags().BoolVarP(&force, "force", "", false, L("Force migrate"))
migrateCmd.PersistentFlags().BoolVarP(&resetModel, "reset", "", false, L("Drop the table if exist"))
}

2
cmd/restore.go
View file

@ -116,7 +116,7 @@ func restoreModels(basePath string) {
for _, mod := range gou.Models {
fmt.Printf("\r%s", strings.Repeat(" ", 80))
fmt.Printf(color.GreenString(L("\rUpdate schema model: %s (%s) "), mod.Name, mod.MetaData.Table.Name))
err := mod.ForceCreateSchema()
err := mod.Migrate(true)
if err != nil {
fmt.Println(color.RedString(L("Fatal: %s"), err.Error()))
os.Exit(1)