Update tax gas meter after each tx

baseapp/baseapp.go

@@ -633,6 +633,7 @@ func (app *BaseApp) runTx(mode runTxMode, txBytes []byte) (gInfo sdk.GasInfo, re
 	var gasWanted uint64
 
 	ctx := app.getContextForTx(mode, txBytes)
+	ctx = ctx.WithNewTaxGasMeter()
 	ms := ctx.MultiStore()
 
 	// only run the tx if there is block gas remaining
@@ -756,6 +757,7 @@ func (app *BaseApp) runTx(mode runTxMode, txBytes []byte) (gInfo sdk.GasInfo, re
 			// We clear this to correctly order events without duplicates.
 			// Note that the state is still preserved.
 			postCtx := runMsgCtx.WithEventManager(sdk.NewEventManager())
+			fmt.Println("postCtx taxgas meter: ", postCtx.TaxGasMeter().GasConsumed())
 
 			newCtx, err := app.postHandler(postCtx, tx, mode == runTxModeSimulate, err == nil)
 			if err != nil {

types/context.go

@@ -240,6 +240,12 @@ func (c Context) WithMinGasPrices(gasPrices DecCoins) Context {
 	return c
 }
 
+// WithTaxGasMeter returns a Context with an updated tax gas meter value
+func (c Context) WithNewTaxGasMeter() Context {
+	c.taxGasMeter = storetypes.NewTaxGasMeter()
+	return c
+}
+
 // WithConsensusParams returns a Context with an updated consensus params
 func (c Context) WithConsensusParams(params *tmproto.ConsensusParams) Context {
 	c.consParams = params