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

bench: Support multiple piece infos in ParsePieceInfos

Browse Source
This commit is contained in:
Łukasz Magiera 2022-05-30 16:19:57 +02:00
parent e148663155
commit 790a331ca5
1 changed files with 24 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
cmd/lotus-bench/simple.go
View File

@ -959,22 +959,32 @@ var simpleProveReplicaUpdate2 = &cli.Command{
} }
func ParsePieceInfos(cctx *cli.Context, firstArg int) ([]abi.PieceInfo, error) { func ParsePieceInfos(cctx *cli.Context, firstArg int) ([]abi.PieceInfo, error) {
// supports only one for now args := cctx.Args().Len() - firstArg
if args%2 != 0 {
return nil, xerrors.Errorf("piece info argunemts need to be supplied in pairs")
}
if args < 2 {
return nil, xerrors.Errorf("need at least one piece info argument")
}
c, err := cid.Parse(cctx.Args().Get(firstArg)) out := make([]abi.PieceInfo, args/2)
for i := 0; i < args/2; i++ {
c, err := cid.Parse(cctx.Args().Get(firstArg + (i * 2)))
if err != nil { if err != nil {
return nil, xerrors.Errorf("parse piece cid: %w", err) return nil, xerrors.Errorf("parse piece cid: %w", err)
} }
psize, err := strconv.ParseUint(cctx.Args().Get(firstArg+1), 10, 64) psize, err := strconv.ParseUint(cctx.Args().Get(firstArg+(i*2)+1), 10, 64)
if err != nil { if err != nil {
return nil, xerrors.Errorf("parse piece size: %w", err) return nil, xerrors.Errorf("parse piece size: %w", err)
} }
return []abi.PieceInfo{ out[i] = abi.PieceInfo{
{
Size: abi.PaddedPieceSize(psize), Size: abi.PaddedPieceSize(psize),
PieceCID: c, PieceCID: c,
}, }
}, nil }
return out, nil
} }