cmd: add hash trie node prune tool

cmd/geth/dbcmd.go

@@ -75,6 +75,7 @@ Remove blockchain and state databases`,
 			// dbMigrateFreezerCmd,
 			dbCheckStateContentCmd,
 			dbHash2PathCmd,
+			dbPruneHashTrieCmd,
 			dbTrieGetCmd,
 		},
 	}
@@ -109,6 +110,17 @@ a data corruption.`,
 		Usage:       "Convert Hash-Base to Path-Base trie node.",
 		Description: `This command iterates the entire trie node database and convert the hash-base node to path-base node.`,
 	}
+	dbPruneHashTrieCmd = &cli.Command{
+		Action:    pruneHashTrie,
+		Name:      "prune-hash-trie",
+		ArgsUsage: "",
+		Flags: []cli.Flag{
+			utils.DataDirFlag,
+			utils.SyncModeFlag,
+		},
+		Usage:       "Prune all the hash trie node in diskdb",
+		Description: `This command iterates the entrie kv in leveldb and delete all the hash trie node.`,
+	}
 	dbStatCmd = &cli.Command{
 		Action: dbStats,
 		Name:   "stats",
@@ -357,6 +369,20 @@ func hash2Path(ctx *cli.Context) error {
 	return nil
 }
 
+func pruneHashTrie(ctx *cli.Context) error {
+	if ctx.NArg() != 0 {
+		return fmt.Errorf("required none argument")
+	}
+
+	stack, _ := makeConfigNode(ctx)
+	defer stack.Close()
+
+	db := utils.MakeChainDatabase(ctx, stack, false, false)
+	defer db.Close()
+
+	return rawdb.PruneHashTrieNodeInDataBase(db)
+}
+
 func inspect(ctx *cli.Context) error {
 	var (
 		prefix []byte

core/rawdb/database.go

@@ -582,6 +582,28 @@ func AncientInspect(db ethdb.Database) error {
 	return nil
 }
 
+func PruneHashTrieNodeInDataBase(db ethdb.Database) error {
+	it := db.NewIterator([]byte{}, []byte{})
+	defer it.Release()
+
+	total_num := 0
+	for it.Next() {
+		var key = it.Key()
+		switch {
+		case IsLegacyTrieNode(key, it.Value()):
+			db.Delete(key)
+			total_num++
+			if total_num%100000 == 0 {
+				fmt.Printf("Complete progress: %v hash-base trie nodes\n", total_num)
+			}
+		default:
+			continue
+		}
+	}
+	fmt.Printf("Complete progress: %v hash-base trie nodes\n", total_num)
+	return nil
+}
+
 // InspectDatabase traverses the entire database and checks the size
 // of all different categories of data.
 func InspectDatabase(db ethdb.Database, keyPrefix, keyStart []byte) error {