minor code updates

Dean Karn · Dean Karn · commit c2b526c718b6 · 2019-05-09T06:43:08.000-07:00
diff --git a/io/limit_reader.go b/io/limit_reader.go
@@ -34,12 +34,12 @@ func (l *LimitedReader) Read(p []byte) (n int, err error) {
 		p = p[0 : l.N+1]
 	}
 	n, err = l.R.Read(p)
-	if err == io.EOF {
+	l.N -= int64(n)
+	if err != nil {
 		return
 	}
-	l.N -= int64(n)
 	if l.N < 0 {
-		return len(p), LimitedReaderEOF
+		return n, LimitedReaderEOF
 	}
 	return
 }

Original file line number	Diff line number	Diff line change
`@@ -34,12 +34,12 @@ func (l *LimitedReader) Read(p []byte) (n int, err error) {`
`34`	`34`	`p = p[0 : l.N+1]`
`35`	`35`	`}`
`36`	`36`	`n, err = l.R.Read(p)`
`37`		`- if err == io.EOF {`
	`37`	`+ l.N -= int64(n)`
	`38`	`+ if err != nil {`
`38`	`39`	`return`
`39`	`40`	`}`
`40`		`- l.N -= int64(n)`
`41`	`41`	`if l.N < 0 {`
`42`		`- return len(p), LimitedReaderEOF`
	`42`	`+ return n, LimitedReaderEOF`
`43`	`43`	`}`
`44`	`44`	`return`
`45`	`45`	`}`