websocket: fix send queue buffer release

lesismal · Sep 2, 2024 · d2b0199 · d2b0199
1 parent 428f1a7
commit d2b0199
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/nbhttp/websocket/conn.go b/nbhttp/websocket/conn.go
@@ -796,13 +796,14 @@ func (c *Conn) writeFrame(messageType MessageType, sendOpcode, fin bool, data []
 				i := 0
 				for {
 					_, err := c.Conn.Write(buf)
-					c.Engine.BodyAllocator.Free(buf)
 					if err != nil {
 						c.CloseWithError(err)
 						return
 					}
-
+					c.Engine.BodyAllocator.Free(buf)
+					c.sendQueue[i] = nil
 					i++
+
 					c.mux.Lock()
 					if c.closed {
 						c.mux.Unlock()
@@ -815,7 +816,6 @@ func (c *Conn) writeFrame(messageType MessageType, sendOpcode, fin bool, data []
 					}
 
 					buf = c.sendQueue[i]
-					c.sendQueue[i] = nil
 
 					c.mux.Unlock()