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 1

cmd/geth: add dbDeleteAll

Browse source
This commit is contained in:
weiihann 2024-03-12 13:04:46 +08:00
parent ebf9e11af2
commit 32d7c88cc7
1 changed files with 75 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

75
cmd/geth/dbcmd.go
View file

@ -72,6 +72,7 @@ Remove blockchain and state databases`,
dbCompactCmd,
dbGetCmd,
dbDeleteCmd,
dbDeleteAllCmd,
dbPutCmd,
dbGetSlotsCmd,
dbDumpFreezerIndex,
@ -141,6 +142,17 @@ corruption if it is aborted during execution'!`,
utils.SyncModeFlag,
}, utils.NetworkFlags, utils.DatabaseFlags),
Description: `This command deletes the specified database key from the database.
WARNING: This is a low-level operation which may cause database corruption!`,
}
dbDeleteAllCmd = &cli.Command{
Action: dbDeleteAll,
Name: "delete-all",
Usage: "Delete all database keys based on a prefix (WARNING: may corrupt your database)",
ArgsUsage: "<prefix>",
Flags: flags.Merge([]cli.Flag{
utils.SyncModeFlag,
}, utils.NetworkFlags, utils.DatabaseFlags),
Description: `This command deletes all keys from the database based on a prefix.
WARNING: This is a low-level operation which may cause database corruption!`,
}
dbPutCmd = &cli.Command{
@ -472,6 +484,69 @@ func dbDelete(ctx *cli.Context) error {
return nil
}
// dbDeleteAll deletes all keys based on a prefix key from the database
func dbDeleteAll(ctx *cli.Context) error {
if ctx.NArg() != 1 {
return fmt.Errorf("required arguments: %v", ctx.Command.ArgsUsage)
}
prefixKey, err := common.ParseHexOrString(ctx.Args().Get(0))
if err != nil {
return fmt.Errorf("failed to hex-decode 'prefix': %v", err)
}
stack, _ := makeConfigNode(ctx)
defer stack.Close()
db := utils.MakeChainDatabase(ctx, stack, false)
defer db.Close()
it := db.NewIterator(prefixKey, nil)
defer it.Release()
var (
start = time.Now()
logged = time.Now()
batch = db.NewBatch()
count int64
key []byte
)
for it.Next() {
key = it.Key()
if err = batch.Delete(key); err != nil {
log.Error("Delete operation returned an error", "key", fmt.Sprintf("%#x", key), "error", err)
return err
}
if batch.ValueSize() > ethdb.IdealBatchSize {
if err = batch.Write(); err != nil {
log.Error("Write batch operation returned an error", "error", err)
return err
}
batch.Reset()
}
count++
if time.Since(logged) > 8*time.Second {
log.Info("Deleting keys", "count", count, "elapsed", common.PrettyDuration(time.Since(start)))
logged = time.Now()
}
}
if batch.ValueSize() > 0 {
if err = batch.Write(); err != nil {
log.Error("Write batch operation returned an error", "error", err)
return err
}
batch.Reset()
}
log.Info(fmt.Sprintf("Deleted all keys at prefix '%v'", prefixKey), "count", count, "elapsed", common.PrettyDuration(time.Since(start)))
return nil
}
// dbPut overwrite a value in the database
func dbPut(ctx *cli.Context) error {
if ctx.NArg() != 2 {