Merge pull request #998 from wladh/master

Check for insufficient data when we try to get the records.
IBM · Dec 5, 2017 · 8f05b2f · 8f05b2f
2 parents 703eaab + eee4c71
commit 8f05b2f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/record_batch.go b/record_batch.go
@@ -161,6 +161,11 @@ func (b *RecordBatch) decode(pd packetDecoder) (err error) {
 	bufSize := int(batchLen) - recordBatchOverhead
 	recBuffer, err := pd.getRawBytes(bufSize)
 	if err != nil {
+		if err == ErrInsufficientData {
+			b.PartialTrailingRecord = true
+			b.Records = nil
+			return nil
+		}
 		return err
 	}