Address feedback around defer unlock

graphql/handler/transport/websocket.go

@@ -238,13 +238,12 @@
 
 func (c *wsConnection) write(msg *message) {
 	c.mu.Lock()
+	defer c.mu.Unlock()
 	// don't write anything to a closed / closing connection
 	if c.serverClosed {
-		c.mu.Unlock()
 		return
 	}
 	c.handlePossibleError(c.me.Send(msg), false)
-	c.mu.Unlock()
 }
 
 func (c *wsConnection) run() {
@@ -548,7 +547,7 @@
 
 	closeTimeout := c.closeTimeout
 	if closeTimeout == 0 {
 		closeTimeout = time.Second * 3
 	}
 	closeTimer := time.NewTimer(c.closeTimeout)