Feat: add force auto migrate flag

pull/48/head
zijiren233 2 years ago
parent 45311ab694
commit 08d348961e

@ -15,4 +15,6 @@ var (
GitHubBaseURL string GitHubBaseURL string
DataDir string DataDir string
ForceAutoMigrate bool
) )

@ -40,6 +40,7 @@ func init() {
home = "~" home = "~"
} }
RootCmd.PersistentFlags().StringVar(&flags.DataDir, "data-dir", filepath.Join(home, ".synctv"), "data dir") RootCmd.PersistentFlags().StringVar(&flags.DataDir, "data-dir", filepath.Join(home, ".synctv"), "data dir")
RootCmd.PersistentFlags().BoolVar(&flags.ForceAutoMigrate, "force-auto-migrate", false, "force auto migrate")
} }
func init() { func init() {

@ -4,6 +4,7 @@ import (
"fmt" "fmt"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"github.com/synctv-org/synctv/cmd/flags"
"github.com/synctv-org/synctv/internal/conf" "github.com/synctv-org/synctv/internal/conf"
"github.com/synctv-org/synctv/internal/model" "github.com/synctv-org/synctv/internal/model"
"gorm.io/gorm" "gorm.io/gorm"
@ -55,8 +56,8 @@ func UpgradeDatabase() error {
if err != nil { if err != nil {
return err return err
} }
currentVersion := setting.Value currentVersion = setting.Value
if currentVersion != CurrentVersion { if flags.ForceAutoMigrate || currentVersion != CurrentVersion {
err = autoMigrate(models...) err = autoMigrate(models...)
if err != nil { if err != nil {
return err return err

Loading…
Cancel
Save