Address review comments

segmentio · Jan 27, 2023 · 006d9b5 · 006d9b5
1 parent 6656660
commit 006d9b5
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/batch.go b/batch.go
@@ -81,7 +81,11 @@ func (batch *Batch) close() (err error) {
  batch.lock = nil
  if batch.msgs != nil {
  batch.msgs.discard()
+ }
+
+ if batch.msgs.decompressed != nil {
  releaseBuffer(batch.msgs.decompressed)
+ batch.msgs.decompressed = nil
  }
 
  if err = batch.err; errors.Is(batch.err, io.EOF) {