go/oasis-node/cmd/debug/consim: Wire in the pruner

oasisprotocol · Apr 9, 2020 · 41b800c · 41b800c
1 parent 7249fc4
commit 41b800c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/go/oasis-node/cmd/debug/consim/consim.go b/go/oasis-node/cmd/debug/consim/consim.go
@@ -28,6 +28,7 @@ import (
 )
 
 const (
+	cfgNumKept      = "consim.num_kept"
 	cfgWorkload     = "consim.workload"
 	cfgWorkloadSeed = "consim.workload.seed"
 )
@@ -113,6 +114,7 @@ func doRun(cmd *cobra.Command, args []string) error {
 		genesisDoc:    genesisDoc,
 		tmChainID:     tmChainID,
 		txAuthHandler: txAuthApp.(abci.TransactionAuthHandler),
+		numVersions:   viper.GetInt64(cfgNumKept),
 	}
 	mockChain, err := initMockChain(ctx, cfg)
 	if err != nil {
@@ -245,6 +247,7 @@ func Register(parentCmd *cobra.Command) {
 }
 
 func init() {
+	flagsConsim.Int64(cfgNumKept, 0, "number of versions kept (0 = all)")
 	flagsConsim.String(cfgWorkload, fileWorkloadName, "workload to execute")
 	flagsConsim.String(cfgWorkloadSeed, "seeeeeeeeeeeeeeeeeeeeeeeeeeeeeed", "DRBG seed for workloads")
 	_ = viper.BindPFlags(flagsConsim)

diff --git a/go/oasis-node/cmd/debug/consim/mockchain.go b/go/oasis-node/cmd/debug/consim/mockchain.go
@@ -25,6 +25,7 @@ type mockChainCfg struct {
 	genesisDoc    *genesis.Document
 	tmChainID     string
 	txAuthHandler abci.TransactionAuthHandler
+	numVersions   int64
 }
 
 type mockChain struct {
@@ -158,6 +159,10 @@ func initMockChain(ctx context.Context, cfg *mockChainCfg) (*mockChain, error) {
 		MinGasPrice:     0, // XXX: Should this be configurable?
 		OwnTxSigner:     localSigner.Public(),
 	}
+	if cfg.numVersions > 0 {
+		muxCfg.Pruning.Strategy = abci.PruneKeepN
+		muxCfg.Pruning.NumKept = cfg.numVersions
+	}
 	mux, err := abci.NewMockMux(ctx, upgrade.NewDummyUpgradeManager(), muxCfg)
 	if err != nil {
 		logger.Error("failed to initialize mock mux",