mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-03 08:37:37 +00:00
5a66807989
* First take at updating everything to v5 * Patch gRPC gateway to use prysm v5 Fix patch * Update go ssz --------- Co-authored-by: Preston Van Loon <pvanloon@offchainlabs.com>
49 lines
1.2 KiB
Go
49 lines
1.2 KiB
Go
package db
|
|
|
|
import (
|
|
"context"
|
|
"path"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/prysmaticlabs/prysm/v5/cmd"
|
|
"github.com/prysmaticlabs/prysm/v5/io/file"
|
|
"github.com/prysmaticlabs/prysm/v5/validator/db/kv"
|
|
"github.com/urfave/cli/v2"
|
|
)
|
|
|
|
// MigrateUp for a validator database.
|
|
func MigrateUp(cliCtx *cli.Context) error {
|
|
dataDir := cliCtx.String(cmd.DataDirFlag.Name)
|
|
|
|
if !file.Exists(path.Join(dataDir, kv.ProtectionDbFileName)) {
|
|
return errors.New("No validator db found at path, nothing to migrate")
|
|
}
|
|
|
|
ctx := context.Background()
|
|
log.Info("Opening DB")
|
|
validatorDB, err := kv.NewKVStore(ctx, dataDir, &kv.Config{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Info("Running migrations")
|
|
return validatorDB.RunUpMigrations(ctx)
|
|
}
|
|
|
|
// MigrateDown for a validator database.
|
|
func MigrateDown(cliCtx *cli.Context) error {
|
|
dataDir := cliCtx.String(cmd.DataDirFlag.Name)
|
|
|
|
if !file.Exists(path.Join(dataDir, kv.ProtectionDbFileName)) {
|
|
return errors.New("No validator db found at path, nothing to rollback")
|
|
}
|
|
|
|
ctx := context.Background()
|
|
log.Info("Opening DB")
|
|
validatorDB, err := kv.NewKVStore(ctx, dataDir, &kv.Config{})
|
|
if err != nil {
|
|
return err
|
|
}
|
|
log.Info("Running migrations")
|
|
return validatorDB.RunDownMigrations(ctx)
|
|
}
|