re-use the usedStatsInfo

Signed-off-by: Yang Keao <[email protected]>
pingcap · Jan 17, 2025 · a1160da · a1160da
1 parent 889bf45
commit a1160da
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/sessionctx/stmtctx/stmtctx.go b/pkg/sessionctx/stmtctx/stmtctx.go
@@ -469,6 +469,9 @@ func NewStmtCtxWithTimeZone(tz *time.Location) *StatementContext {
 
 // Reset resets a statement context
 func (sc *StatementContext) Reset() {
+	usedStats := sc.usedStatsInfo.Load()
+	usedStats.store.Clear()
+
 	*sc = StatementContext{
 		ctxID:               contextutil.GenContextID(),
 		CTEStorageMap:       sc.CTEStorageMap,
@@ -494,6 +497,7 @@ func (sc *StatementContext) Reset() {
 	} else {
 		sc.ExtraWarnHandler = contextutil.NewStaticWarnHandler(0)
 	}
+	sc.usedStatsInfo.Store(usedStats)
 }
 
 // CtxID returns the context id of the statement