mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-12 18:00:49 +01:00
67 lines
2.1 KiB
Go
67 lines
2.1 KiB
Go
|
package cmd
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"flag"
|
||
|
"fmt"
|
||
|
"os"
|
||
|
"path"
|
||
|
|
||
|
"github.com/netbirdio/netbird/management/server"
|
||
|
"github.com/netbirdio/netbird/util"
|
||
|
log "github.com/sirupsen/logrus"
|
||
|
"github.com/spf13/cobra"
|
||
|
)
|
||
|
|
||
|
var shortDown = "Rollback SQLite store to JSON file store. Please make a backup of the SQLite file before running this command."
|
||
|
|
||
|
var downCmd = &cobra.Command{
|
||
|
Use: "downgrade [--datadir directory] [--log-file console]",
|
||
|
Aliases: []string{"down"},
|
||
|
Short: shortDown,
|
||
|
Long: shortDown +
|
||
|
"\n\n" +
|
||
|
"This command reads the content of {datadir}/store.db and migrates it to {datadir}/store.json that can be used by File store driver.",
|
||
|
RunE: func(cmd *cobra.Command, args []string) error {
|
||
|
flag.Parse()
|
||
|
err := util.InitLog(logLevel, logFile)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("failed initializing log %v", err)
|
||
|
}
|
||
|
|
||
|
sqliteStorePath := path.Join(mgmtDataDir, "store.db")
|
||
|
if _, err := os.Stat(sqliteStorePath); errors.Is(err, os.ErrNotExist) {
|
||
|
return fmt.Errorf("%s doesn't exist, couldn't continue the operation", sqliteStorePath)
|
||
|
}
|
||
|
|
||
|
fileStorePath := path.Join(mgmtDataDir, "store.json")
|
||
|
if _, err := os.Stat(fileStorePath); err == nil {
|
||
|
return fmt.Errorf("%s already exists, couldn't continue the operation", fileStorePath)
|
||
|
}
|
||
|
|
||
|
sqlstore, err := server.NewSqliteStore(mgmtDataDir, nil)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("failed creating file store: %s: %v", mgmtDataDir, err)
|
||
|
}
|
||
|
|
||
|
sqliteStoreAccounts := len(sqlstore.GetAllAccounts())
|
||
|
log.Infof("%d account will be migrated from sqlite store %s to file store %s",
|
||
|
sqliteStoreAccounts, sqliteStorePath, fileStorePath)
|
||
|
|
||
|
store, err := server.NewFilestoreFromSqliteStore(sqlstore, mgmtDataDir, nil)
|
||
|
if err != nil {
|
||
|
return fmt.Errorf("failed creating file store: %s: %v", mgmtDataDir, err)
|
||
|
}
|
||
|
|
||
|
fsStoreAccounts := len(store.GetAllAccounts())
|
||
|
if fsStoreAccounts != sqliteStoreAccounts {
|
||
|
return fmt.Errorf("failed to migrate accounts from sqlite to file[]. Expected accounts: %d, got: %d",
|
||
|
sqliteStoreAccounts, fsStoreAccounts)
|
||
|
}
|
||
|
|
||
|
log.Info("Migration finished successfully")
|
||
|
|
||
|
return nil
|
||
|
},
|
||
|
}
|