From 6f4042263856e842c61fe007c6c12521c763c747 Mon Sep 17 00:00:00 2001 From: you06 Date: Wed, 23 Aug 2023 15:49:45 +0800 Subject: [PATCH] record type of the prepared stmt for execute stmt Signed-off-by: you06 --- session/session.go | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/session/session.go b/session/session.go index ffb343aeb8a00..2d9849eed0dd2 100644 --- a/session/session.go +++ b/session/session.go @@ -2189,7 +2189,16 @@ func (s *session) ExecuteStmt(ctx context.Context, stmtNode ast.StmtNode) (sqlex } }) - stmtLabel := ast.GetStmtLabel(stmtNode) + var stmtLabel string + if execStmt, ok := stmtNode.(*ast.ExecuteStmt); ok { + prepareStmt, err := plannercore.GetPreparedStmt(execStmt, s.sessionVars) + if err == nil && prepareStmt.PreparedAst != nil { + stmtLabel = ast.GetStmtLabel(prepareStmt.PreparedAst.Stmt) + } + } + if stmtLabel == "" { + stmtLabel = ast.GetStmtLabel(stmtNode) + } s.setRequestSource(ctx, stmtLabel, stmtNode) // Backup the original resource group name since sql hint might change it during optimization