Explorar o código

avoid writing empty payload in bytesToBufferWriter

Darien Raymond %!s(int64=8) %!d(string=hai) anos
pai
achega
3d2fd372e1
Modificáronse 1 ficheiros con 3 adicións e 4 borrados
  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
 		}