txn: fix retried transactions' statements is not inclueded in TIDB_TRX (

#28480) (#28670)
pingcap · Oct 13, 2021 · 9751790 · 9751790
1 parent c62fc9f
commit 9751790
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/session/session.go b/session/session.go
@@ -927,7 +927,10 @@ func (s *session) retry(ctx context.Context, maxCnt uint) (err error) {
 					zap.Uint("retryCnt", retryCnt),
 					zap.Int("queryNum", i))
 			}
+			_, digest := s.sessionVars.StmtCtx.SQLDigest()
+			s.txn.onStmtStart(digest.String())
 			_, err = st.Exec(ctx)
+			s.txn.onStmtEnd()
 			if err != nil {
 				s.StmtRollback()
 				break