forks/go-ethereum
1
0
Fork 1
mirror of https://github.com/ethereum/go-ethereum.git synced 2026-08-20 10:52:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

cmd/geth: make it possible to autopilot removedb

Browse source
This commit is contained in:
Martin Holst Swende 2023-12-21 20:58:28 +01:00
parent cca94792a4
commit 0f734cc7f2
No known key found for this signature in database
GPG key ID: 683B438C05A5DDF0
1 changed files with 27 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
cmd/geth/dbcmd.go
View file

@ -44,10 +44,11 @@ import (
var ( var (
removedbCommand = &cli.Command{ removedbCommand = &cli.Command{
Action: removeDB, Action: removeDB,
Name: "removedb", Name: "removedb",
Usage: "Remove blockchain and state databases", Usage: "Remove blockchain and state databases. If arguments are provided, " +
ArgsUsage: "", "they are interpreted as comma-separated answers to questions.",
ArgsUsage: " <? comma-separated answers>",
Flags: utils.DatabaseFlags, Flags: utils.DatabaseFlags,
Description: ` Description: `
Remove blockchain and state databases`, Remove blockchain and state databases`,
@ -202,6 +203,7 @@ func removeDB(ctx *cli.Context) error {
var ( var (
rootDir = stack.ResolvePath("chaindata") rootDir = stack.ResolvePath("chaindata")
ancientDir = config.Eth.DatabaseFreezer ancientDir = config.Eth.DatabaseFreezer
answers = strings.Split(ctx.Args().Get(0), ",")
) )
switch { switch {
case ancientDir == "": case ancientDir == "":
@ -209,13 +211,21 @@ func removeDB(ctx *cli.Context) error {
case !filepath.IsAbs(ancientDir): case !filepath.IsAbs(ancientDir):
ancientDir = config.Node.ResolvePath(ancientDir) ancientDir = config.Node.ResolvePath(ancientDir)
} }
var oracle = func() string {
var ret string
if len(answers) > 0 {
ret = answers[0]
answers = answers[1:]
}
return ret
}
// Delete state data // Delete state data
statePaths := []string{rootDir, filepath.Join(ancientDir, rawdb.StateFreezerName)} statePaths := []string{rootDir, filepath.Join(ancientDir, rawdb.StateFreezerName)}
confirmAndRemoveDB(statePaths, "state data") confirmAndRemoveDB(statePaths, "state data", oracle)
// Delete ancient chain // Delete ancient chain
chainPaths := []string{filepath.Join(ancientDir, rawdb.ChainFreezerName)} chainPaths := []string{filepath.Join(ancientDir, rawdb.ChainFreezerName)}
confirmAndRemoveDB(chainPaths, "ancient chain") confirmAndRemoveDB(chainPaths, "ancient chain", oracle)
return nil return nil
} }
@ -238,14 +248,23 @@ func removeFolder(dir string) {
// confirmAndRemoveDB prompts the user for a last confirmation and removes the // confirmAndRemoveDB prompts the user for a last confirmation and removes the
// list of folders if accepted. // list of folders if accepted.
func confirmAndRemoveDB(paths []string, kind string) { func confirmAndRemoveDB(paths []string, kind string, oracle func() string) {
var (
confirm bool
err error
)
msg := fmt.Sprintf("Location(s) of '%s': \n", kind) msg := fmt.Sprintf("Location(s) of '%s': \n", kind)
for _, path := range paths { for _, path := range paths {
msg += fmt.Sprintf("\t- %s\n", path) msg += fmt.Sprintf("\t- %s\n", path)
} }
fmt.Println(msg) fmt.Println(msg)
confirm, err := prompt.Stdin.PromptConfirm(fmt.Sprintf("Remove '%s'?", kind)) if answer := oracle(); answer != "" {
fmt.Printf("Remove '%s'? [y/n] %s\n", kind, answer)
confirm = (answer == "y")
} else {
confirm, err = prompt.Stdin.PromptConfirm(fmt.Sprintf("Remove '%s'?", kind))
}
switch { switch {
case err != nil: case err != nil:
utils.Fatalf("%v", err) utils.Fatalf("%v", err)