Add is simulate in baseapp

classic-terra · Sep 18, 2024 · 8172e9b · 8172e9b
1 parent 225886c
commit 8172e9b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/baseapp/baseapp.go b/baseapp/baseapp.go
@@ -596,6 +596,7 @@ func (app *BaseApp) getContextForTx(mode runTxMode, txBytes []byte) sdk.Context
 
 	if mode == runTxModeSimulate {
 		ctx, _ = ctx.CacheContext()
+		ctx = ctx.WithSimulate(true)
 	}
 
 	if mode == runTxModeSimulateSpecial {

diff --git a/types/context.go b/types/context.go
@@ -36,6 +36,7 @@ type Context struct {
 	taxGasMeter          TaxGasMeter
 	cacheTaxGasMeter     TaxGasMeter
 	specialSimulate      bool
+	simulate             bool
 	checkTx              bool
 	recheckTx            bool // if recheckTx == true, then checkTx must also be true
 	minGasPrice          DecCoins
@@ -63,6 +64,7 @@ func (c Context) BlockGasMeter() GasMeter                    { return c.blockGas
 func (c Context) TaxGasMeter() TaxGasMeter                   { return c.taxGasMeter }
 func (c Context) CacheTaxGasMeter() TaxGasMeter              { return c.cacheTaxGasMeter }
 func (c Context) IsSpecialSimulate() bool                    { return c.specialSimulate }
+func (c Context) IsSimulate() bool                           { return c.simulate }
 func (c Context) IsCheckTx() bool                            { return c.checkTx }
 func (c Context) IsReCheckTx() bool                          { return c.recheckTx }
 func (c Context) MinGasPrices() DecCoins                     { return c.minGasPrice }
@@ -234,6 +236,11 @@ func (c Context) WithSpecialSimulate(isSpecialSimulate bool) Context {
 	return c
 }
 
+func (c Context) WithSimulate(isSimulate bool) Context {
+	c.simulate = isSimulate
+	return c
+}
+
 // WithIsRecheckTx called with true will also set true on checkTx in order to
 // enforce the invariant that if recheckTx = true then checkTx = true as well.
 func (c Context) WithIsReCheckTx(isRecheckTx bool) Context {