fix: specialize discard for readerWithEOFContext

whyrusleeping · Mar 10, 2022 · 6eb0095 · 6eb0095
1 parent 099ab3e
commit 6eb0095
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/utils.go b/utils.go
@@ -27,6 +27,13 @@ func discard(br io.Reader, n int) error {
 	}
 
 	switch r := br.(type) {
+	case *readerWithEOFContext:
+		err := discard(r.r, n)
+		if err == io.EOF && r.hasReadOnce {
+			err = io.ErrUnexpectedEOF
+		}
+		r.hasReadOnce = true
+		return err
 	case *bytes.Buffer:
 		buf := r.Next(n)
 		if len(buf) == 0 {