chore: fix a few conditions

panjf2000 · Apr 17, 2024 · d386d1f · d386d1f
1 parent 0bf879b
commit d386d1f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/connection_unix.go b/connection_unix.go
@@ -396,13 +396,13 @@ func (c *conn) Next(n int) (buf []byte, err error) {
 	}
 	head, tail := c.inboundBuffer.Peek(n)
 	defer c.inboundBuffer.Discard(n) //nolint:errcheck
-	if len(head) >= n {
+	if len(head) == n {
 		return head[:n], err
 	}
 	c.loop.cache.Reset()
 	c.loop.cache.Write(head)
 	c.loop.cache.Write(tail)
-	if inBufferLen >= n {
+	if inBufferLen == n {
 		return c.loop.cache.Bytes(), err
 	}
 
@@ -435,13 +435,13 @@ func (c *conn) Peek(n int) (buf []byte, err error) {
 		return c.buffer[:n], err
 	}
 	head, tail := c.inboundBuffer.Peek(n)
-	if len(head) >= n {
+	if len(head) == n {
 		return head[:n], err
 	}
 	c.loop.cache.Reset()
 	c.loop.cache.Write(head)
 	c.loop.cache.Write(tail)
-	if inBufferLen >= n {
+	if inBufferLen == n {
 		return c.loop.cache.Bytes(), err
 	}
 

diff --git a/pkg/buffer/elastic/elastic_ring_list_buffer.go b/pkg/buffer/elastic/elastic_ring_list_buffer.go
@@ -60,7 +60,7 @@ func (mb *Buffer) Peek(n int) ([][]byte, error) {
 		return nil, io.ErrShortBuffer
 	}
 	head, tail := mb.ringBuffer.Peek(n)
-	if mb.ringBuffer.Buffered() >= n {
+	if mb.ringBuffer.Buffered() == n {
 		return [][]byte{head, tail}, nil
 	}
 	return mb.listBuffer.PeekWithBytes(n, head, tail)