cherry pick #28480 to release-5.2

Signed-off-by: ti-srebot <ti-srebot@pingcap.com>
pingcap · Oct 9, 2021 · b79d4e7 · b79d4e7
1 parent 466f6e7
commit b79d4e7
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