fix

pingcap · Oct 3, 2024 · 4aa4362 · 4aa4362
1 parent 5007897
commit 4aa4362
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/pkg/executor/distsql.go b/pkg/executor/distsql.go
@@ -540,6 +540,14 @@ func (e *IndexLookUpExecutor) Open(ctx context.Context) error {
 			return err
 		}
 	}
+
+	if e.memTracker != nil {
+		e.memTracker.Reset()
+	} else {
+		e.memTracker = memory.NewTracker(e.ID(), -1)
+	}
+	e.memTracker.AttachTo(e.stmtMemTracker)
+
 	err = e.buildTableKeyRanges()
 	if err != nil {
 		return err
@@ -921,6 +929,10 @@ func (e *IndexLookUpExecutor) Close() error {
 	e.finished = nil
 	e.workerStarted = false
 	e.resultCurr = nil
+
+	if e.memTracker != nil {
+		e.memTracker.Reset()
+	}
 	return nil
 }