chore: unlock only the once after locking

DanielleMaywood · DanielleMaywood · commit 0c8afd92dbcf · 2025-12-19T13:35:37.000Z
diff --git a/write.go b/write.go
@@ -105,13 +105,12 @@ func (c *Conn) write(ctx context.Context, typ MessageType, p []byte) (int, error
 	if err != nil {
 		return 0, err
 	}
+	defer c.msgWriter.mu.unlock()
 
 	if !c.flate() || len(p) < c.flateThreshold {
-		defer c.msgWriter.mu.unlock()
 		return c.writeFrame(ctx, true, false, c.msgWriter.opcode, p)
 	}
 
-	defer c.msgWriter.mu.unlock()
 	return c.msgWriter.writeCompressedFrame(ctx, p)
 }
 

Original file line number	Diff line number	Diff line change
`@@ -105,13 +105,12 @@ func (c *Conn) write(ctx context.Context, typ MessageType, p []byte) (int, error`
`105`	`105`	`if err != nil {`
`106`	`106`	`return 0, err`
`107`	`107`	`}`
	`108`	`+ defer c.msgWriter.mu.unlock()`
`108`	`109`
`109`	`110`	`if !c.flate() \|\| len(p) < c.flateThreshold {`
`110`		`- defer c.msgWriter.mu.unlock()`
`111`	`111`	`return c.writeFrame(ctx, true, false, c.msgWriter.opcode, p)`
`112`	`112`	`}`
`113`	`113`
`114`		`- defer c.msgWriter.mu.unlock()`
`115`	`114`	`return c.msgWriter.writeCompressedFrame(ctx, p)`
`116`	`115`	`}`
`117`	`116`