variable: fix data race in GetTemporaryTable (pingcap#55338)

close pingcap#55337
qingfeng777 · Aug 9, 2024 · 2675c66 · 2675c66
1 parent 240702e
commit 2675c66
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/sessionctx/variable/session.go b/pkg/sessionctx/variable/session.go
@@ -2765,6 +2765,8 @@ func (s *SessionVars) GetDivPrecisionIncrement() int {
 // GetTemporaryTable returns a TempTable by tableInfo.
 func (s *SessionVars) GetTemporaryTable(tblInfo *model.TableInfo) tableutil.TempTable {
 	if tblInfo.TempTableType != model.TempTableNone {
+		s.TxnCtxMu.Lock()
+		defer s.TxnCtxMu.Unlock()
 		if s.TxnCtx.TemporaryTables == nil {
 			s.TxnCtx.TemporaryTables = make(map[int64]tableutil.TempTable)
 		}