fix: BinlogDeserializeReader leak (#36087)

#36086 Signed-off-by: fengjun2016 <[email protected]>
milvus-io · Sep 10, 2024 · 4850641 · 4850641
1 parent f4d0c58
commit 4850641
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/internal/indexnode/task_stats.go b/internal/indexnode/task_stats.go
@@ -381,6 +381,7 @@ func (st *statsTask) downloadData(ctx context.Context, numRows int64, PKFieldID
 					break
 				} else {
 					log.Warn("downloadData wrong, failed to iter through data", zap.Error(err))
+					iter.Close()
 					return nil, err
 				}
 			}
@@ -399,6 +400,7 @@ func (st *statsTask) downloadData(ctx context.Context, numRows int64, PKFieldID
 			values = append(values, iter.Value())
 			remainingRowCount++
 		}
+		iter.Close()
 	}
 
 	log.Info("download data success",