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

Add lookback limit

Browse Source 
Signed-off-by: Jakub Sztandera <kubuxu@protocol.ai>
This commit is contained in:
Jakub Sztandera 2021-06-10 19:04:15 +02:00 committed by Steven Allen
parent 68593ce995
commit 3d3c26fa0c
2 changed files with 41 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
cmd/lotus-sim/info.go
View File

@ -101,6 +101,12 @@ var infoSimCommand = &cli.Command{
var infoCommitGasSimCommand = &cli.Command{
Name: "commit-gas",
Description: "Output information about the gas for committs",
Flags: []cli.Flag{
&cli.Int64Flag{
Name: "lookback",
Value: 0,
},
},
Action: func(cctx *cli.Context) error {
log := func(f string, i ...interface{}) {
fmt.Fprintf(os.Stderr, f, i...)
@ -119,7 +125,8 @@ var infoCommitGasSimCommand = &cli.Command{
var gasAgg, proofsAgg uint64
var gasAggMax, proofsAggMax uint64
sim.Walk(cctx.Context, func(sm *stmgr.StateManager, ts *types.TipSet, stCid cid.Cid,
sim.Walk(cctx.Context, cctx.Int64("lookback"),
func(sm *stmgr.StateManager, ts *types.TipSet, stCid cid.Cid,
messages []*simulation.AppliedMessage) error {
for _, m := range messages {
if m.ExitCode != exitcode.Ok {

8
cmd/lotus-sim/simulation/simulation.go
View File

@ -286,6 +286,7 @@ type AppliedMessage struct {
// Walk walks the simulation's chain from the current head back to the first tipset.
func (sim *Simulation) Walk(
ctx context.Context,
maxLookback int64,
cb func(sm *stmgr.StateManager,
ts *types.TipSet,
stCid cid.Cid,
@ -297,7 +298,12 @@ func (sim *Simulation) Walk(
if err != nil {
return err
}
for !ts.Equals(sim.start) && ctx.Err() == nil {
minEpoch := abi.ChainEpoch(0)
if maxLookback != 0 {
minEpoch = ts.Height() - abi.ChainEpoch(maxLookback)
}
for !ts.Equals(sim.start) && ctx.Err() == nil && ts.Height() > minEpoch {
msgs, err := sim.Chainstore.MessagesForTipset(ts)
if err != nil {
return err