fixup! ws_js: Update to match new close code

Author: nhooyr

read.go

@@ -64,9 +64,10 @@ func (c *Conn) Read(ctx context.Context) (MessageType, []byte, error) {
 // This function is idempotent.
 func (c *Conn) CloseRead(ctx context.Context) context.Context {
 	c.closeReadMu.Lock()
-	if c.closeReadCtx != nil {
+	ctx := c.closeReadCtx
+	if ctx != nil {
 		c.closeReadMu.Unlock()
-		return c.closeReadCtx
+		return ctx
 	}
 	ctx, cancel := context.WithCancel(ctx)
 	c.closeReadCtx = ctx

ws_js.go

@@ -392,9 +392,10 @@ func (w *writer) Close() error {
 // CloseRead implements *Conn.CloseRead for wasm.
 func (c *Conn) CloseRead(ctx context.Context) context.Context {
 	c.closeReadMu.Lock()
-	if c.closeReadCtx != nil {
+	ctx := c.closeReadCtx
+	if ctx != nil {
 		c.closeReadMu.Unlock()
-		return c.closeReadCtx
+		return ctx
 	}
 	ctx, cancel := context.WithCancel(ctx)
 	c.closeReadCtx = ctx