Fix performance degradation while host functions call

Signed-off-by: Dmitry Volodin <dmvolod@gmail.com>
knqyf263 · Apr 19, 2023 · fb23714 · fb23714
1 parent 898cc7b
commit fb23714
Show file tree

Hide file tree

Showing 6 changed files with 11 additions and 0 deletions.
diff --git a/examples/host-function-library/library/json-parser/export/library_plugin.pb.go b/examples/host-function-library/library/json-parser/export/library_plugin.pb.go
diff --git a/examples/host-function-library/proto/greet_plugin.pb.go b/examples/host-function-library/proto/greet_plugin.pb.go
diff --git a/examples/host-functions/greeting/greet_plugin.pb.go b/examples/host-functions/greeting/greet_plugin.pb.go
diff --git a/gen/plugin.go b/gen/plugin.go
@@ -115,6 +115,7 @@ func genHostFunctions(g *protogen.GeneratedFile, f *fileInfo) {
 			}
 			ptr, size := %s(buf)
 			ptrSize := _%s(ptr, size)
+			%s(ptr)
 
 			ptr = uint32(ptrSize >> 32)
 			size = uint32(ptrSize)
@@ -132,6 +133,7 @@ func genHostFunctions(g *protogen.GeneratedFile, f *fileInfo) {
 			g.QualifiedGoIdent(method.Output.GoIdent),
 			g.QualifiedGoIdent(pluginWasmPackage.Ident("ByteToPtr")),
 			importedName,
+			g.QualifiedGoIdent(pluginWasmPackage.Ident("FreePtr")),
 			g.QualifiedGoIdent(pluginWasmPackage.Ident("PtrToByte")),
 			g.QualifiedGoIdent(method.Output.GoIdent),
 		))

diff --git a/tests/host-functions/proto/host_plugin.pb.go b/tests/host-functions/proto/host_plugin.pb.go
diff --git a/wasm/plugin.go b/wasm/plugin.go
@@ -33,3 +33,7 @@ func ByteToPtr(buf []byte) (uint32, uint32) {
 
 	return uint32(uintptr(ptr)), uint32(len(buf))
 }
+
+func FreePtr(ptr uint32) {
+	C.free(unsafe.Pointer(uintptr(ptr)))
+}