From d5089d83323229623dfc0a21bec7b773a20f8a0f Mon Sep 17 00:00:00 2001
From: you06 <you1474600@gmail.com>
Date: Wed, 23 Aug 2023 21:08:12 +0800
Subject: [PATCH] session: set request source to type of the prepared stmt for
 execute stmt (#46348) (#46362)

close pingcap/tidb#46349
---
 session/session.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/session/session.go b/session/session.go
index ac26339953ffd..c705acd0ce1b2 100644
--- a/session/session.go
+++ b/session/session.go
@@ -2197,7 +2197,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)
 
 	// Transform abstract syntax tree to a physical plan(stored in executor.ExecStmt).