package main import ( "context" "fmt" "io" "math/big" big2 "github.com/filecoin-project/go-state-types/big" "github.com/filecoin-project/lotus/chain/consensus/filcns" "github.com/filecoin-project/go-address" "github.com/filecoin-project/go-state-types/abi" "github.com/filecoin-project/lotus/chain/actors/builtin/miner" "github.com/filecoin-project/lotus/chain/actors/builtin/power" "github.com/filecoin-project/lotus/chain/state" "github.com/filecoin-project/lotus/chain/store" "github.com/filecoin-project/lotus/chain/types" "github.com/filecoin-project/lotus/node/repo" builtin4 "github.com/filecoin-project/specs-actors/v4/actors/builtin" "github.com/filecoin-project/specs-actors/v4/actors/util/adt" "github.com/ipfs/go-cid" cbor "github.com/ipfs/go-ipld-cbor" "github.com/urfave/cli/v2" "golang.org/x/xerrors" ) var minerTypesCmd = &cli.Command{ Name: "miner-types", Usage: "Scrape state to report on how many miners of each WindowPoStProofType exist", Flags: []cli.Flag{ &cli.StringFlag{ Name: "repo", Value: "~/.lotus", }, }, Action: func(cctx *cli.Context) error { ctx := context.TODO() 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 bs, err := lkrepo.Blockstore(ctx, repo.UniversalBlockstore) if err != nil { return fmt.Errorf("failed to open blockstore: %w", err) } defer func() { if c, ok := bs.(io.Closer); ok { if err := c.Close(); err != nil { log.Warnf("failed to close blockstore: %s", err) } } }() mds, err := lkrepo.Datastore(context.Background(), "/metadata") if err != nil { return err } cs := store.NewChainStore(bs, bs, mds, filcns.Weight, nil) defer cs.Close() //nolint:errcheck cst := cbor.NewCborStore(bs) store := adt.WrapStore(ctx, cst) tree, err := state.LoadStateTree(cst, sroot) if err != nil { return err } typeMap := make(map[abi.RegisteredPoStProof]int64) pa, err := tree.GetActor(power.Address) if err != nil { return err } ps, err := power.Load(store, pa) if err != nil { return err } dc := 0 dz := power.Claim{ RawBytePower: abi.NewStoragePower(0), QualityAdjPower: abi.NewStoragePower(0), } err = tree.ForEach(func(addr address.Address, act *types.Actor) error { if act.Code == builtin4.StorageMinerActorCodeID { ms, err := miner.Load(store, act) if err != nil { return err } mi, err := ms.Info() if err != nil { return err } if mi.WindowPoStProofType == abi.RegisteredPoStProof_StackedDrgWindow64GiBV1 { mp, f, err := ps.MinerPower(addr) if err != nil { return err } if f && mp.RawBytePower.Cmp(big.NewInt(10<<40)) >= 0 && mp.RawBytePower.Cmp(big.NewInt(20<<40)) < 0 { dc = dc + 1 dz.RawBytePower = big2.Add(dz.RawBytePower, mp.RawBytePower) dz.QualityAdjPower = big2.Add(dz.QualityAdjPower, mp.QualityAdjPower) } } c, f := typeMap[mi.WindowPoStProofType] if !f { typeMap[mi.WindowPoStProofType] = 1 } else { typeMap[mi.WindowPoStProofType] = c + 1 } } return nil }) if err != nil { return xerrors.Errorf("failed to loop over actors: %w", err) } for k, v := range typeMap { fmt.Println("Type:", k, " Count: ", v) } fmt.Println("Mismatched power (raw, QA): ", dz.RawBytePower, " ", dz.QualityAdjPower) fmt.Println("Mismatched 64 GiB miner count: ", dc) return nil }, }