chore: correctly clear context when proxy is done (#86)

* chore: correctly clear context when proxy is done

* Update proxywasm/abi_lifecycle.go

Co-authored-by: Takeshi Yoneda <[email protected]>

* Update proxywasm/abi_lifecycle.go

Co-authored-by: Takeshi Yoneda <[email protected]>

* Update proxywasm/abi_lifecycle.go

Co-authored-by: Takeshi Yoneda <[email protected]>

Co-authored-by: Wim Spaargaren <[email protected]>
Co-authored-by: Takeshi Yoneda <[email protected]>

Author: 3 people

proxywasm/abi_lifecycle.go

@@ -31,15 +31,19 @@ func proxyOnContextCreate(contextID uint32, rootContextID uint32) {
 func proxyOnDone(contextID uint32) bool {
 	if ctx, ok := currentState.streams[contextID]; ok {
 		currentState.setActiveContextID(contextID)
+		delete(currentState.streams, contextID)
 		ctx.OnStreamDone()
 		return true
 	} else if ctx, ok := currentState.httpStreams[contextID]; ok {
 		currentState.setActiveContextID(contextID)
 		ctx.OnHttpStreamDone()
+		delete(currentState.httpStreams, contextID)
 		return true
 	} else if ctx, ok := currentState.rootContexts[contextID]; ok {
 		currentState.setActiveContextID(contextID)
-		return ctx.context.OnVMDone()
+		response := ctx.context.OnVMDone()
+		delete(currentState.rootContexts, contextID)
+		return response
 	} else {
 		panic("invalid context on proxy_on_done")
 	}