close the connection when run returns

cgxxv · Jul 10, 2020 · d129136 · d129136
1 parent af72dd6
commit d129136
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 {