catch raw record set write error

segmentio · Oct 19, 2023 · 47882d1 · 47882d1
1 parent df55dc2
commit 47882d1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/protocol/record.go b/protocol/record.go
@@ -309,7 +309,11 @@ func (rs *RecordSet) WriteTo(w io.Writer) (int64, error) {
 }
 
 func (rrs *RawRecordSet) WriteTo(w io.Writer) (int64, error) {
-	return io.Copy(w, rrs.Reader)
+	n, err := io.Copy(w, rrs.Reader)
+	if err != nil {
+		return n, fmt.Errorf("kafka.(*RawRecordSet).WriteTo: %w", err)
+	}
+	return n, err
 }
 
 func makeTime(t int64) time.Time {