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

add migrate options support for restore cmd

Browse source
This commit is contained in:
yflau 2024-06-13 20:00:56 +08:00
parent 48ae099ea0
commit 39b8ef6238
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 migrateNoInsert bool = false
var restoreCmd = &cobra.Command{
Use: "restore",
Short: L("Restore the application data"),
@ -66,7 +67,9 @@ var restoreCmd = &cobra.Command{
}
// Restore models
restoreModels(filepath.Join(dst, "model"))
restoreModels(filepath.Join(dst, "model"), []model.MigrateOption{
model.WithDonotInsertValues(migrateNoInsert),
})
// Restore Data
restoreData(filepath.Join(dst, "data"))
@ -80,6 +83,7 @@ var restoreCmd = &cobra.Command{
func init() {
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) {
@ -104,7 +108,7 @@ func restoreData(basePath string) {
}
}
func restoreModels(basePath string) {
func restoreModels(basePath string, migOpts []model.MigrateOption) {
files, err := ioutil.ReadDir(basePath)
if err != nil {
@ -116,7 +120,7 @@ func restoreModels(basePath string) {
for _, mod := range model.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.Migrate(true)
err := mod.Migrate(true, migOpts...)
if err != nil {
fmt.Println(color.RedString(L("Fatal: %s"), err.Error()))
os.Exit(1)