cleanup vm after finish the simulation (#3090)

okx · Apr 11, 2023 · 9fc23a8 · 9fc23a8
1 parent a4c6b66
commit 9fc23a8
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/x/wasm/keeper/keeper.go b/x/wasm/keeper/keeper.go
@@ -1262,6 +1262,10 @@ func (k Keeper) Logger(ctx sdk.Context) log.Logger {
 	return moduleLogger(ctx)
 }
 
+func (k *Keeper) Cleanup() {
+	k.wasmVM.Cleanup()
+}
+
 func (k *Keeper) SetInnerTxKeeper(innertxKeeper innertx.InnerTxKeeper) {
 	k.innertxKeeper = innertxKeeper
 }

diff --git a/x/wasm/simulate.go b/x/wasm/simulate.go
@@ -59,6 +59,7 @@ func (w *Simulator) Release() {
 		return
 	}
 	proxy.PutBackStorePool(w.ctx.MultiStore().(sdk.CacheMultiStore))
+	w.k.Cleanup()
 }
 
 func NewProxyKeeper() keeper.Keeper {