[fix go#27] remove the entity context for a cancelled stream.

cloudstateio · Jun 13, 2020 · 177f7b3 · 177f7b3
1 parent 7bd8eb4
commit 177f7b3
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/cloudstate/cloudstate.go b/cloudstate/cloudstate.go
@@ -33,7 +33,7 @@ import (
 )
 
 const (
-	SupportLibraryVersion = "0.1.0"
+	SupportLibraryVersion = "0.1.1"
 	SupportLibraryName    = "cloudstate-go-support"
 )
 

diff --git a/cloudstate/eventsourced.go b/cloudstate/eventsourced.go
@@ -147,6 +147,12 @@ func (esh *EventSourcedServer) registerEntity(ese *EventSourcedEntity) error {
 func (esh *EventSourcedServer) Handle(stream protocol.EventSourced_HandleServer) error {
 	var entityId string
 	var failed error
+
+	// "fix" for https://github.com/cloudstateio/go-support/issues/27
+	// v0.2.x will contain the proper handling for the case of a closed stream.
+	defer func() {
+		delete(esh.contexts, entityId)
+	}()
 	for {
 		if failed != nil {
 			return failed

diff --git a/tck/cmd/tck_shoppingcart/shoppingcart.go b/tck/cmd/tck_shoppingcart/shoppingcart.go
@@ -34,7 +34,7 @@ import (
 func main() {
 	server, err := cloudstate.New(cloudstate.Config{
 		ServiceName:    "shopping-cart",
-		ServiceVersion: "0.1.0",
+		ServiceVersion: "0.1.1",
 	})
 	if err != nil {
 		log.Fatalf("CloudState.New failed: %v", err)