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

…e query (#41151) close #41057
pingcap · Feb 9, 2023 · c34238b · c34238b
1 parent a22d690
commit c34238b
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()