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

…e query (#41151) (#41249) close #41057
pingcap · Feb 17, 2023 · 0991a0b · 0991a0b
1 parent e3c13ec
commit 0991a0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/session/session.go b/session/session.go
@@ -1614,7 +1614,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()