catch raw record set write error

apacker · apacker · commit 47882d1baaf5 · 2023-10-19T17:04:49.000-04:00
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 {

Original file line number	Diff line number	Diff line change
`@@ -309,7 +309,11 @@ func (rs *RecordSet) WriteTo(w io.Writer) (int64, error) {`
`309`	`309`	`}`
`310`	`310`
`311`	`311`	`func (rrs *RawRecordSet) WriteTo(w io.Writer) (int64, error) {`
`312`		`- return io.Copy(w, rrs.Reader)`
	`312`	`+ n, err := io.Copy(w, rrs.Reader)`
	`313`	`+ if err != nil {`
	`314`	`+ return n, fmt.Errorf("kafka.(*RawRecordSet).WriteTo: %w", err)`
	`315`	`+ }`
	`316`	`+ return n, err`
`313`	`317`	`}`
`314`	`318`
`315`	`319`	`func makeTime(t int64) time.Time {`