mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-14 12:05:18 +00:00
84 lines
2.2 KiB
Go
84 lines
2.2 KiB
Go
|
package iavlviewer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strconv"
|
||
|
|
|
||
|
|
dbm "github.com/cometbft/cometbft-db"
|
||
|
|
"github.com/cosmos/cosmos-sdk/client"
|
||
|
|
"github.com/cosmos/cosmos-sdk/server"
|
||
|
|
ethermintserver "github.com/evmos/ethermint/server"
|
||
|
|
"github.com/spf13/cobra"
|
||
|
|
|
||
|
|
"github.com/cosmos/iavl"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
DefaultCacheSize int = 10000
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewCmd(opts ethermintserver.StartOptions) *cobra.Command {
|
||
|
|
cmd := &cobra.Command{
|
||
|
|
Use: "iavlviewer <data|hash|shape|versions> <prefix> [version number]",
|
||
|
|
Short: "Output various data, hashes, and calculations for an iavl tree",
|
||
|
|
}
|
||
|
|
|
||
|
|
cmd.AddCommand(newDataCmd(opts))
|
||
|
|
cmd.AddCommand(newHashCmd(opts))
|
||
|
|
cmd.AddCommand(newShapeCmd(opts))
|
||
|
|
cmd.AddCommand(newVersionsCmd(opts))
|
||
|
|
|
||
|
|
return cmd
|
||
|
|
}
|
||
|
|
|
||
|
|
func parseVersion(arg string) (int, error) {
|
||
|
|
version, err := strconv.Atoi(arg)
|
||
|
|
if err != nil {
|
||
|
|
return 0, fmt.Errorf("invalid version number: '%s'", arg)
|
||
|
|
}
|
||
|
|
return version, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func openPrefixTree(opts ethermintserver.StartOptions, cmd *cobra.Command, prefix string, version int) (*iavl.MutableTree, error) {
|
||
|
|
clientCtx := client.GetClientContextFromCmd(cmd)
|
||
|
|
ctx := server.GetServerContextFromCmd(cmd)
|
||
|
|
ctx.Config.SetRoot(clientCtx.HomeDir)
|
||
|
|
|
||
|
|
db, err := opts.DBOpener(ctx.Viper, clientCtx.HomeDir, server.GetAppDBBackend(ctx.Viper))
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("failed to open database at %s: %s", clientCtx.HomeDir, err)
|
||
|
|
}
|
||
|
|
defer func() {
|
||
|
|
if err := db.Close(); err != nil {
|
||
|
|
ctx.Logger.Error("error closing db", "error", err.Error())
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
|
||
|
|
tree, err := readTree(db, version, []byte(prefix))
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("failed to read tree with prefix %s: %s", prefix, err)
|
||
|
|
}
|
||
|
|
return tree, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// ReadTree loads an iavl tree from the directory
|
||
|
|
// If version is 0, load latest, otherwise, load named version
|
||
|
|
// The prefix represents which iavl tree you want to read. The iaviwer will always set a prefix.
|
||
|
|
func readTree(db dbm.DB, version int, prefix []byte) (*iavl.MutableTree, error) {
|
||
|
|
if len(prefix) != 0 {
|
||
|
|
db = dbm.NewPrefixDB(db, prefix)
|
||
|
|
}
|
||
|
|
|
||
|
|
tree, err := iavl.NewMutableTree(db, DefaultCacheSize, false)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
ver, err := tree.LoadVersion(int64(version))
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
fmt.Printf("Latest version: %d\n", ver)
|
||
|
|
fmt.Printf("Got version: %d\n", version)
|
||
|
|
return tree, err
|
||
|
|
}
|