session: fix processInfo time when using driver to prepare and execut…

…e query (pingcap#41151) close pingcap#41057
blacktear23 · Feb 15, 2023 · 0bc5691 · 0bc5691
1 parent 989aaee
commit 0bc5691
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/session/session.go b/session/session.go
@@ -1601,7 +1601,7 @@ func (s *session) SetProcessInfo(sql string, t time.Time, command byte, maxExecu
 		}
 	}
 	// We set process info before building plan, so we extended execution time.
-	if oldPi != nil && oldPi.Info == pi.Info {
+	if oldPi != nil && oldPi.Info == pi.Info && oldPi.Command == pi.Command {
 		pi.Time = oldPi.Time
 	}
 	_, digest := s.sessionVars.StmtCtx.SQLDigest()