Add prune wasm codes proposal

x/wasm/keeper/bench_test.go

@@ -8,7 +8,9 @@ import (
 	"github.com/cosmos/cosmos-sdk/crypto/keys/secp256k1"
 	"github.com/stretchr/testify/require"
 
+	"github.com/CosmWasm/wasmd/x/wasm/keeper/wasmtesting"
 	"github.com/CosmWasm/wasmd/x/wasm/types"
+	wasmvm "github.com/CosmWasm/wasmvm"
 )
 
 // BenchmarkVerification benchmarks secp256k1 verification which is 1000 gas based on cpu time.
@@ -100,3 +102,26 @@ func BenchmarkCompilation(b *testing.B) {
 		})
 	}
 }
+
+// Calculate the time it takes to prune about 100k wasm codes
+func BenchmarkPruningWasmCodes(b *testing.B) {
+	mockWasmVM := wasmtesting.MockWasmer{
+		RemoveCodeFn: func(checksum wasmvm.Checksum) error {
+			return nil
+		},
+	}
+	wasmtesting.MakeInstantiable(&mockWasmVM)
+	ctx, keepers := CreateTestInput(b, false, AvailableCapabilities, WithWasmEngine(&mockWasmVM))
+
+	for i := 0; i < 100000; i++ {
+		contract := StoreRandomContract(b, ctx, keepers, &mockWasmVM)
+		if i%10 == 0 {
+			keepers.ContractKeeper.PinCode(ctx, contract.CodeID)
+		}
+	}
+
+	b.ResetTimer()
+	for i := 0; i < b.N; i++ {
+		keepers.WasmKeeper.PruneWasmCodes(ctx, 100000)
+	}
+}

x/wasm/keeper/wasmtesting/mock_engine.go

@@ -167,7 +167,7 @@ func (m *MockWasmer) Unpin(checksum wasmvm.Checksum) error {
 }
 
 func (m *MockWasmer) RemoveCode(checksum wasmvm.Checksum) error {
-	if m.UnpinFn == nil {
+	if m.RemoveCodeFn == nil {
 		panic("not supposed to be called!")
 	}
 	return m.RemoveCodeFn(checksum)