From e56a2c5a858c713a866415bb005ed9e19fb565b0 Mon Sep 17 00:00:00 2001
From: vedhavyas <vedhavyas.singareddi@gmail.com>
Date: Mon, 23 Sep 2019 23:23:27 +0200
Subject: [PATCH] add relase call

---
 bridge.go               |  50 +++++++++++++++++++++++++++++++++++++---
 examples/wasm/main.wasm | Bin 2934596 -> 2934596 bytes
 go-converts/wasm.go     |   8 ++++++-
 imports.go              |  16 +++++++++----
 4 files changed, 65 insertions(+), 9 deletions(-)

diff --git a/bridge.go b/bridge.go
index 81a8400..7da6b40 100644
--- a/bridge.go
+++ b/bridge.go
@@ -18,9 +18,14 @@ import (
 	"github.com/wasmerio/go-ext-wasm/wasmer"
 )
 
-var undefined = &struct{}{}
-var bridges = map[string]*Bridge{}
-var mu sync.RWMutex // to protect bridges
+const release_call = "_release_"
+
+var (
+	undefined = &struct{}{}
+	bridges   = map[string]*Bridge{}
+	mu        sync.RWMutex // to protect bridges
+)
+
 type bctx struct{ n string }
 
 func getCtxData(b *Bridge) (unsafe.Pointer, error) {
@@ -55,6 +60,30 @@ type Bridge struct {
 	cancF    context.CancelFunc
 }
 
+// releaseRef removes the ref from the refs.
+// Returns the  idx and true if remove was successful.
+func (b *Bridge) releaseRef(v interface{}) (int, bool) {
+	idx, ok := b.refs[v]
+	if !ok {
+		return 0, false
+	}
+
+	delete(b.refs, v)
+	return idx, true
+}
+
+// releaseVal removes val from the valueMap
+// Returns the value and true if remove was successful
+func (b *Bridge) releaseVal(idx int) (interface{}, bool) {
+	v, ok := b.valueMap[idx]
+	if !ok {
+		return nil, false
+	}
+
+	delete(b.valueMap, idx)
+	return v, true
+}
+
 func BridgeFromBytes(name string, bytes []byte, imports *wasmer.Imports) (*Bridge, error) {
 	b := new(Bridge)
 	if imports == nil {
@@ -513,6 +542,21 @@ func (b *Bridge) SetFunc(fname string, fn Func) error {
 	return nil
 }
 
+func (b *Bridge) releaseFunc(v interface{}) Func {
+	return Func(func(args []interface{}) (interface{}, error) {
+		b.valuesMu.Lock()
+		defer b.valuesMu.Unlock()
+
+		idx, ok := b.releaseRef(v)
+		if !ok {
+			return nil, nil
+		}
+
+		b.releaseVal(idx)
+		return nil, nil
+	})
+}
+
 func Bytes(v interface{}) ([]byte, error) {
 	arr, ok := v.(*array)
 	if !ok {
diff --git a/examples/wasm/main.wasm b/examples/wasm/main.wasm
index 51d65fcb75eadc0988914af93521fd5ac2bad744..46875cce909fb5939e2aa5d0a8ab3d252b56eb35 100755
GIT binary patch
delta 269
zcmWN=IZML;7zSXHc(!W1k9yzMgFs4&j;TkBNJSE>oy1lJi%K+x+VU<AFWtodA`;xQ
zy7mWjbl2JMfhT(VP$K!BbgphIm;3emN-gtr;h4KyfxG7#X2A*0a$Zo~N#{$?W#2pT
z%En#R-adRu`?(|Ay*aS*=e6RoQ3?tvKkGf3r;bxiS!Ska)n7p<gi2^c8_`a55S>I9
z(M`mNIMGA&5`BbD^b-TbATdM?6C=bZF-D9N6Q4+2+e*CAvJ-MrPRVIGBWLBDoR<r7
mQ7*}4xguBPTG+CaKi{!mP5IN*NKwBUO?_Q%giU?(A^Zo#fM$FE

delta 269
zcmWN=IZML;7zSX{c-AA{N4;;Wh7d`$jv121QcXQjJBeVcLX3E!hIesz=_dXckqoX~
zUHbz%`UiCMd*G?PeJC~go<7JpH%FDSQ*=+SwsQ~rkr{cead)ntIaXQUy51@JW)$37
z*$pGe+1dPa=H4sVhZT2o_vEGI8)3n6{ZerJ=<S71dHr(TyzpN^C`5u#i6)|%XdzmO
zHlm$K5*<V*(M5C<8qq`a5`9EJF+dCwL&PvKLX3VQRkoF6wPB~^m>icAa#BvoX*nZj
n<(!<C3vy8|$>q3Vr+>bazpC=5CTdFJt6JAq<Z4{k*24H7aCBu3

diff --git a/go-converts/wasm.go b/go-converts/wasm.go
index 94ac596..73697fc 100644
--- a/go-converts/wasm.go
+++ b/go-converts/wasm.go
@@ -7,8 +7,14 @@ import "syscall/js"
 func ToBytes(v js.Value) []byte {
 	buf := make([]byte, v.Length(), v.Length())
 	for i := 0; i < v.Length(); i++ {
-		buf[i] = byte(v.Index(i).Int())
+		sv := v.Index(i)
+		buf[i] = byte(sv.Int())
 	}
 
 	return buf
 }
+
+// Free frees the value for GC
+func free(v js.Value) {
+	v.Call("_release_")
+}
diff --git a/imports.go b/imports.go
index bbac0ad..9b77c70 100644
--- a/imports.go
+++ b/imports.go
@@ -156,13 +156,19 @@ func valueSetIndex(_ unsafe.Pointer, _ int32) {
 func valueCall(ctx unsafe.Pointer, sp int32) {
 	b := getBridge(ctx)
 	v := b.loadValue(sp + 8)
+	var f Func
+	var args []interface{}
 	str := b.loadString(sp + 16)
-	args := b.loadSliceOfValues(sp + 32)
-	f, ok := v.(*object).props[str].(Func)
-	if !ok {
-		panic(fmt.Sprintln("valueCall: prop not found in ", v, str))
+	if str == release_call {
+		f = b.releaseFunc(v)
+	} else {
+		args = b.loadSliceOfValues(sp + 32)
+		var ok bool
+		f, ok = v.(*object).props[str].(Func)
+		if !ok {
+			panic(fmt.Sprintln("valueCall: prop not found in ", v, str))
+		}
 	}
-
 	sp = b.getSP()
 	res, err := f(args)
 	if err != nil {