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

lotus-bench: make start tipset walkback start from end tipset.

Browse Source
This commit is contained in:
Raúl Kripalani 2020-11-03 18:15:55 +00:00
parent 581ac5b019
commit 18418128c3
1 changed files with 31 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
cmd/lotus-bench/import.go
View File

@ -379,33 +379,7 @@ var importBenchCmd = &cli.Command{
return err return err
} }
var ( // Resolve the end tipset, falling back to head if not provided.
startEpoch = abi.ChainEpoch(1)
start *types.TipSet
)
if tsk := cctx.String("start-tipset"); tsk != "" {
cids, err := lcli.ParseTipSetString(tsk)
if err != nil {
return xerrors.Errorf("failed to start genesis tipset key: %w", err)
}
start, err = cs.LoadTipSet(types.NewTipSetKey(cids...))
} else if h := cctx.Int64("start-height"); h != 0 {
log.Infof("getting start tipset at height %d...", h)
start, err = cs.GetTipsetByHeight(context.TODO(), abi.ChainEpoch(h), head, true)
}
if err != nil {
return err
}
if start != nil {
startEpoch = start.Height()
if err := cs.SetHead(start); err != nil {
return err
}
}
end := head end := head
if tsk := cctx.String("end-tipset"); tsk != "" { if tsk := cctx.String("end-tipset"); tsk != "" {
cids, err := lcli.ParseTipSetString(tsk) cids, err := lcli.ParseTipSetString(tsk)
@ -422,6 +396,36 @@ var importBenchCmd = &cli.Command{
return err return err
} }
// Resolve the start tipset, if provided; otherwise, fallback to
// height 1 for a start point.
var (
startEpoch = abi.ChainEpoch(1)
start *types.TipSet
)
if tsk := cctx.String("start-tipset"); tsk != "" {
cids, err := lcli.ParseTipSetString(tsk)
if err != nil {
return xerrors.Errorf("failed to start genesis tipset key: %w", err)
}
start, err = cs.LoadTipSet(types.NewTipSetKey(cids...))
} else if h := cctx.Int64("start-height"); h != 0 {
log.Infof("getting start tipset at height %d...", h)
// lookback from the end tipset (which falls back to head if not supplied).
start, err = cs.GetTipsetByHeight(context.TODO(), abi.ChainEpoch(h), end, true)
}
if err != nil {
return err
}
if start != nil {
startEpoch = start.Height()
if err := cs.SetHead(start); err != nil {
return err
}
}
inverseChain := append(make([]*types.TipSet, 0, end.Height()), end) inverseChain := append(make([]*types.TipSet, 0, end.Height()), end)
for ts := end; ts.Height() > startEpoch; { for ts := end; ts.Height() > startEpoch; {
if h := ts.Height(); h%100 == 0 { if h := ts.Height(); h%100 == 0 {