cerc-io/lotus
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 27 Wiki Activity

set proper bitwidth for verifreg command hamts

Browse Source
This commit is contained in:
whyrusleeping 2020-07-01 17:29:11 -07:00
parent 70d8378404
commit 5fa3021a79
1 changed files with 19 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
cmd/lotus-shed/verifreg.go
View File

@ -3,7 +3,9 @@ package main
import ( import (
"bytes" "bytes"
"fmt" "fmt"
"github.com/filecoin-project/lotus/build" "github.com/filecoin-project/lotus/build"
"golang.org/x/xerrors"
"github.com/filecoin-project/go-address" "github.com/filecoin-project/go-address"
"github.com/urfave/cli/v2" "github.com/urfave/cli/v2"
@ -199,7 +201,7 @@ var verifRegListVerifiersCmd = &cli.Command{
return err return err
} }
vh, err := hamt.LoadNode(ctx, cst, st.Verifiers) vh, err := hamt.LoadNode(ctx, cst, st.Verifiers, hamt.UseTreeBitWidth(5))
if err != nil { if err != nil {
return err return err
} }
@ -251,11 +253,12 @@ var verifRegListClientsCmd = &cli.Command{
return err return err
} }
vh, err := hamt.LoadNode(ctx, cst, st.VerifiedClients) vh, err := hamt.LoadNode(ctx, cst, st.VerifiedClients, hamt.UseTreeBitWidth(5))
if err != nil { if err != nil {
return err return err
} }
var keys []string
if err := vh.ForEach(ctx, func(k string, val interface{}) error { if err := vh.ForEach(ctx, func(k string, val interface{}) error {
addr, err := address.NewFromBytes([]byte(k)) addr, err := address.NewFromBytes([]byte(k))
if err != nil { if err != nil {
@ -268,13 +271,22 @@ var verifRegListClientsCmd = &cli.Command{
return err return err
} }
fmt.Printf("%s: %s\n", addr, dcap) fmt.Printf("%s: %s %v\n", addr, dcap, []byte(k))
keys = append(keys, k)
return nil return nil
}); err != nil { }); err != nil {
return err return err
} }
for _, k := range keys {
_, err := vh.FindRaw(ctx, k)
if err != nil {
fmt.Println("failed to find key: ", []byte(k), err)
}
}
return nil return nil
}, },
} }
@ -312,14 +324,15 @@ var verifRegCheckClientCmd = &cli.Command{
return err return err
} }
vh, err := hamt.LoadNode(ctx, cst, st.VerifiedClients) vh, err := hamt.LoadNode(ctx, cst, st.VerifiedClients, hamt.UseTreeBitWidth(5))
if err != nil { if err != nil {
return err return err
} }
fmt.Println("key: ", caddr.Bytes())
var dcap verifreg.DataCap var dcap verifreg.DataCap
if err := vh.Find(ctx, string(caddr.Bytes()), &dcap); err != nil { if err := vh.Find(ctx, string(caddr.Bytes()), &dcap); err != nil {
return err return xerrors.Errorf("failed to lookup address: %w", err)
} }
fmt.Println(dcap) fmt.Println(dcap)
@ -361,7 +374,7 @@ var verifRegCheckVerifierCmd = &cli.Command{
return err return err
} }
vh, err := hamt.LoadNode(ctx, cst, st.Verifiers) vh, err := hamt.LoadNode(ctx, cst, st.Verifiers, hamt.UseTreeBitWidth(5))
if err != nil { if err != nil {
return err return err
} }