don't reimplement io.ReadFull

libp2p · Feb 17, 2021 · 083dccd · 083dccd
1 parent b6118a4
commit 083dccd
Showing 1 changed file with 2 additions and 13 deletions.
diff --git a/util.go b/util.go
@@ -151,21 +151,10 @@ func (s *segmentedBuffer) Read(b []byte) (int, error) {
 
 func (s *segmentedBuffer) Append(input io.Reader, length int) error {
 	dst := pool.Get(length)
-	n := 0
-	read := 0
-	var err error
-	for n < length && err == nil {
-		read, err = input.Read(dst[n:])
-		n += read
-	}
+	n, err := io.ReadFull(input, dst)
 	if err == io.EOF {
-		if length == n {
-			err = nil
-		} else {
-			err = io.ErrUnexpectedEOF
-		}
+		err = io.ErrUnexpectedEOF
 	}
-
 	s.bm.Lock()
 	defer s.bm.Unlock()
 	if n > 0 {