Merge pull request #10489 from filecoin-project/jen/tvl2
shed: get balances of evm accounts
This commit is contained in:
Jiaying Wang
GitHub
commit
3b0d4f2e45
257
cmd/lotus-shed/fevmanalytics.go
Normal file
257
cmd/lotus-shed/fevmanalytics.go
Normal file
@ -0,0 +1,257 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/ipfs/go-cid"
|
||||
cbor "github.com/ipfs/go-ipld-cbor"
|
||||
"github.com/urfave/cli/v2"
|
||||
"golang.org/x/net/context"
|
||||
"golang.org/x/xerrors"
|
||||
|
||||
"github.com/filecoin-project/go-address"
|
||||
"github.com/filecoin-project/go-state-types/abi"
|
||||
"github.com/filecoin-project/go-state-types/big"
|
||||
|
||||
badgerbs "github.com/filecoin-project/lotus/blockstore/badger"
|
||||
"github.com/filecoin-project/lotus/chain/actors/adt"
|
||||
"github.com/filecoin-project/lotus/chain/actors/builtin"
|
||||
evm2 "github.com/filecoin-project/lotus/chain/actors/builtin/evm"
|
||||
"github.com/filecoin-project/lotus/chain/state"
|
||||
"github.com/filecoin-project/lotus/chain/types"
|
||||
"github.com/filecoin-project/lotus/node/repo"
|
||||
)
|
||||
|
||||
var FevmAnalyticsCmd = &cli.Command{
|
||||
Name: "evm-analytics",
|
||||
Usage: "Get FEVM related metrics",
|
||||
Flags: []cli.Flag{
|
||||
&cli.StringFlag{
|
||||
Name: "repo",
|
||||
Value: "~/.lotus",
|
||||
},
|
||||
},
|
||||
Subcommands: []*cli.Command{
|
||||
FevmBalanceCmd,
|
||||
FevmActorsCmd,
|
||||
},
|
||||
}
|
||||
|
||||
var FevmBalanceCmd = &cli.Command{
|
||||
Name: "evm-balance",
|
||||
Usage: "Balances in eth accounts, evm contracts and placeholders",
|
||||
ArgsUsage: "[state root]",
|
||||
|
||||
Action: func(cctx *cli.Context) error {
|
||||
if cctx.NArg() != 1 {
|
||||
return xerrors.New("only needs state root")
|
||||
}
|
||||
|
||||
if !cctx.Args().Present() {
|
||||
return fmt.Errorf("must pass state root")
|
||||
}
|
||||
|
||||
sroot, err := cid.Decode(cctx.Args().First())
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to parse input: %w", err)
|
||||
}
|
||||
|
||||
fsrepo, err := repo.NewFS(cctx.String("repo"))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
lkrepo, err := fsrepo.Lock(repo.FullNode)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
defer lkrepo.Close() //nolint:errcheck
|
||||
|
||||
path, err := lkrepo.SplitstorePath()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
path = filepath.Join(path, "hot.badger")
|
||||
if err := os.MkdirAll(path, 0755); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
opts, err := repo.BadgerBlockstoreOptions(repo.HotBlockstore, path, lkrepo.Readonly())
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
bs, err := badgerbs.Open(opts)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
cst := cbor.NewCborStore(bs)
|
||||
st, err := state.LoadStateTree(cst, sroot)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
fmt.Println("iterating over all actors")
|
||||
count := 0
|
||||
balanceEvm := abi.NewTokenAmount(0)
|
||||
balanceEthAccount := abi.NewTokenAmount(0)
|
||||
balancePlaceholder := abi.NewTokenAmount(0)
|
||||
|
||||
err = st.ForEach(func(addr address.Address, act *types.Actor) error {
|
||||
if count%200000 == 0 {
|
||||
fmt.Println("processed /n", count)
|
||||
}
|
||||
count++
|
||||
|
||||
if builtin.IsEvmActor(act.Code) {
|
||||
balanceEvm = types.BigAdd(balanceEvm, act.Balance)
|
||||
}
|
||||
|
||||
if builtin.IsEthAccountActor(act.Code) {
|
||||
balanceEthAccount = types.BigAdd(balanceEthAccount, act.Balance)
|
||||
}
|
||||
|
||||
if builtin.IsPlaceholderActor(act.Code) {
|
||||
balancePlaceholder = types.BigAdd(balancePlaceholder, act.Balance)
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
fmt.Println("balances in Eth contracts: ", balanceEvm)
|
||||
fmt.Println("balances in Eth accounts: ", balanceEthAccount)
|
||||
fmt.Println("balances in placeholder: ", balancePlaceholder)
|
||||
fmt.Println("Total balances: ", big.Add(big.Add(balanceEthAccount, balancePlaceholder), balanceEvm))
|
||||
return nil
|
||||
},
|
||||
}
|
||||
|
||||
var FevmActorsCmd = &cli.Command{
|
||||
Name: "evm-actors",
|
||||
Usage: "actors # in eth accounts, evm contracts and placeholders",
|
||||
ArgsUsage: "[state root]",
|
||||
|
||||
Action: func(cctx *cli.Context) error {
|
||||
if cctx.NArg() != 1 {
|
||||
return xerrors.New("only needs state root")
|
||||
}
|
||||
|
||||
if !cctx.Args().Present() {
|
||||
return fmt.Errorf("must pass state root")
|
||||
}
|
||||
|
||||
sroot, err := cid.Decode(cctx.Args().First())
|
||||
if err != nil {
|
||||
return fmt.Errorf("failed to parse input: %w", err)
|
||||
}
|
||||
|
||||
fsrepo, err := repo.NewFS(cctx.String("repo"))
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
lkrepo, err := fsrepo.Lock(repo.FullNode)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
defer lkrepo.Close() //nolint:errcheck
|
||||
|
||||
path, err := lkrepo.SplitstorePath()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
path = filepath.Join(path, "hot.badger")
|
||||
if err := os.MkdirAll(path, 0755); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
opts, err := repo.BadgerBlockstoreOptions(repo.HotBlockstore, path, lkrepo.Readonly())
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
bs, err := badgerbs.Open(opts)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
ctx := context.TODO()
|
||||
cst := cbor.NewCborStore(bs)
|
||||
store := adt.WrapStore(ctx, cst)
|
||||
|
||||
st, err := state.LoadStateTree(cst, sroot)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
fmt.Println("iterating over all actors")
|
||||
count := 0
|
||||
EvmCount := 0
|
||||
EthAccountCount := 0
|
||||
PlaceholderCount := 0
|
||||
ea := []cid.Cid{}
|
||||
|
||||
err = st.ForEach(func(addr address.Address, act *types.Actor) error {
|
||||
if count%200000 == 0 {
|
||||
fmt.Println("processed /n", count)
|
||||
}
|
||||
count++
|
||||
|
||||
if builtin.IsEvmActor(act.Code) {
|
||||
EvmCount++
|
||||
e, err := evm2.Load(store, act)
|
||||
if err != nil {
|
||||
return xerrors.Errorf("fail to load evm actor: %w", err)
|
||||
}
|
||||
bcid, err := e.GetBytecodeCID()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
ea = append(ea, bcid)
|
||||
}
|
||||
|
||||
if builtin.IsEthAccountActor(act.Code) {
|
||||
EthAccountCount++
|
||||
}
|
||||
|
||||
if builtin.IsPlaceholderActor(act.Code) {
|
||||
PlaceholderCount++
|
||||
}
|
||||
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
uniquesa := unique(ea)
|
||||
fmt.Println("# of EVM contracts: ", EvmCount)
|
||||
fmt.Println("# of unqiue EVM contracts: ", len(uniquesa))
|
||||
fmt.Println("b# of Eth accounts: ", EthAccountCount)
|
||||
fmt.Println("# of placeholder: ", PlaceholderCount)
|
||||
return nil
|
||||
},
|
||||
}
|
||||
|
||||
func unique(intSlice []cid.Cid) []cid.Cid {
|
||||
keys := make(map[cid.Cid]bool)
|
||||
list := []cid.Cid{}
|
||||
for _, entry := range intSlice {
|
||||
if _, value := keys[entry]; !value {
|
||||
keys[entry] = true
|
||||
list = append(list, entry)
|
||||
}
|
||||
}
|
||||
return list
|
||||
}
|
||||
@ -85,6 +85,7 @@ func main() {
|
||||
gasTraceCmd,
|
||||
replayOfflineCmd,
|
||||
msgindexCmd,
|
||||
FevmAnalyticsCmd,
|
||||
}
|
||||
|
||||
app := &cli.App{
|
||||
|
||||
Loading…
Reference in New Issue
Block a user