Merge pull request #1248 from sotoslammer/master

close the connection when run returns
99designs · Jul 14, 2020 · 0fbf293 · 0fbf293
2 parents d7eabaf + cc9c520
commit 0fbf293
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/graphql/handler/transport/websocket.go b/graphql/handler/transport/websocket.go
@@ -136,7 +136,10 @@ func (c *wsConnection) run() {
 	// We create a cancellation that will shutdown the keep-alive when we leave
 	// this function.
 	ctx, cancel := context.WithCancel(c.ctx)
-	defer cancel()
+	defer func() {
+		cancel()
+		c.close(websocket.CloseAbnormalClosure, "unexpected closure")
+	}()
 
 	// Create a timer that will fire every interval to keep the connection alive.
 	if c.KeepAlivePingInterval != 0 {