chore: improve the length check for conn.Discard

panjf2000 · Nov 16, 2024 · b610595 · b610595
1 parent adcbdd2
commit b610595
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/connection_unix.go b/connection_unix.go
@@ -368,10 +368,9 @@ func (c *conn) Discard(n int) (int, error) {
 	}
 
 	inBufferLen := c.inboundBuffer.Buffered()
-	tempBufferLen := len(c.buffer)
-	if inBufferLen+tempBufferLen < n || n <= 0 {
+	if totalLen := inBufferLen + len(c.buffer); n >= totalLen || n <= 0 {
 		c.resetBuffer()
-		return inBufferLen + tempBufferLen, nil
+		return totalLen, nil
 	}
 
 	if c.inboundBuffer.IsEmpty() {