Merge pull request 99designs#1248 from sotoslammer/master

close the connection when run returns
cgxxv · Jul 14, 2020 · 2f79571 · 2f79571
2 parents f0054af + d129136
commit 2f79571
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 {