fixup! ws_js: Update to match new close code

nhooyr · nhooyr · commit 684adf9fbef2 · 2024-04-05T17:26:28.000-07:00
diff --git a/read.go b/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
diff --git a/ws_js.go b/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
