fix: properly close records on error.

CorentinB · Jul 3, 2022 · 8a39cde · 8a39cde
1 parent c5c1111
commit 8a39cde
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/dialer.go b/dialer.go
@@ -136,9 +136,8 @@ func (d *customDialer) writeWARCFromConnection(reqPipe, respPipe *io.PipeReader,
 	close(recordChan)
 	if err != nil {
 		d.client.errChan <- err
-
 		// Make sure we close the WARC content buffers
-		for _, record := range batch.Records {
+		for record := range recordChan {
 			record.Content.Close()
 		}