Prechádzať zdrojové kódy

avoid writing empty payload in bytesToBufferWriter

Darien Raymond 8 rokov pred
rodič
commit
3d2fd372e1
1 zmenil súbory, kde vykonal 3 pridanie a 4 odobranie
  1. 3 4
      common/buf/writer.go

+ 3 - 4
common/buf/writer.go

@@ -90,11 +90,10 @@ func (w *bytesToBufferWriter) WriteMultiBuffer(mb MultiBuffer) error {
 func (w *bytesToBufferWriter) ReadFrom(reader io.Reader) (int64, error) {
 	mbReader := NewReader(reader)
 	totalBytes := int64(0)
-	eof := false
-	for !eof {
+	for {
 		mb, err := mbReader.Read()
-		if err == io.EOF {
-			eof = true
+		if errors.Cause(err) == io.EOF {
+			break
 		} else if err != nil {
 			return totalBytes, err
 		}