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

Merge pull request #621 from yflau/main

Browse source 
add migrate option for yao restore command
This commit is contained in:
Max 2024-06-14 20:25:19 +08:00 • committed by GitHub
commit da89223f9d
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
cmd/restore.go
View file

@ -21,6 +21,7 @@ import (
) )
var restoreForce bool = false var restoreForce bool = false
var migrateNoInsert bool = false
var restoreCmd = &cobra.Command{ var restoreCmd = &cobra.Command{
Use: "restore", Use: "restore",
Short: L("Restore the application data"), Short: L("Restore the application data"),
@ -66,7 +67,9 @@ var restoreCmd = &cobra.Command{
} }
// Restore models // Restore models
restoreModels(filepath.Join(dst, "model")) restoreModels(filepath.Join(dst, "model"), []model.MigrateOption{
model.WithDonotInsertValues(migrateNoInsert),
})
// Restore Data // Restore Data
restoreData(filepath.Join(dst, "data")) restoreData(filepath.Join(dst, "data"))
@ -80,6 +83,7 @@ var restoreCmd = &cobra.Command{
func init() { func init() {
restoreCmd.PersistentFlags().BoolVarP(&restoreForce, "force", "", false, L("Force restore")) restoreCmd.PersistentFlags().BoolVarP(&restoreForce, "force", "", false, L("Force restore"))
restoreCmd.PersistentFlags().BoolVarP(&migrateNoInsert, "migrate-no-insert", "", false, L("Do not insert values when migrating"))
} }
func restoreData(basePath string) { func restoreData(basePath string) {
@ -104,7 +108,7 @@ func restoreData(basePath string) {
} }
} }
func restoreModels(basePath string) { func restoreModels(basePath string, migOpts []model.MigrateOption) {
files, err := ioutil.ReadDir(basePath) files, err := ioutil.ReadDir(basePath)
if err != nil { if err != nil {
@ -116,7 +120,7 @@ func restoreModels(basePath string) {
for _, mod := range model.Models { for _, mod := range model.Models {
fmt.Printf("\r%s", strings.Repeat(" ", 80)) fmt.Printf("\r%s", strings.Repeat(" ", 80))
fmt.Printf(color.GreenString(L("\rUpdate schema model: %s (%s) "), mod.Name, mod.MetaData.Table.Name)) fmt.Printf(color.GreenString(L("\rUpdate schema model: %s (%s) "), mod.Name, mod.MetaData.Table.Name))
err := mod.Migrate(true) err := mod.Migrate(true, migOpts...)
if err != nil { if err != nil {
fmt.Println(color.RedString(L("Fatal: %s"), err.Error())) fmt.Println(color.RedString(L("Fatal: %s"), err.Error()))
os.Exit(1) os.Exit(1)