From b453ce010631217c13beb2a7d9c15406a3ca15af Mon Sep 17 00:00:00 2001
From: stephen <stephen5217@163.com>
Date: Mon, 2 Dec 2024 16:50:37 +0800
Subject: [PATCH] [Enhancement] add tuning guides to query profile

Signed-off-by: stephen <stephen5217@163.com>
---
 .../starrocks/qe/feedback/OperatorTuningGuides.java    |  1 +
 .../java/com/starrocks/sql/optimizer/Optimizer.java    |  8 ++++++++
 test/sql/test_feedback/R/test_feedback                 | 10 +++++++---
 test/sql/test_feedback/T/test_feedback                 |  2 ++
 4 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/fe/fe-core/src/main/java/com/starrocks/qe/feedback/OperatorTuningGuides.java b/fe/fe-core/src/main/java/com/starrocks/qe/feedback/OperatorTuningGuides.java
index a1b29b5d773e9..f81d76d661808 100644
--- a/fe/fe-core/src/main/java/com/starrocks/qe/feedback/OperatorTuningGuides.java
+++ b/fe/fe-core/src/main/java/com/starrocks/qe/feedback/OperatorTuningGuides.java
@@ -120,6 +120,7 @@ public OptimizedRecord(OperatorTuningGuides operatorTuningGuides, List<TuningGui
         public OperatorTuningGuides getOperatorTuningGuides() {
             return operatorTuningGuides;
         }
+
         public String getExplainString() {
             StringBuilder sb = new StringBuilder();
             sb.append("Plan had been tuned by Plan Advisor.").append("\n");
diff --git a/fe/fe-core/src/main/java/com/starrocks/sql/optimizer/Optimizer.java b/fe/fe-core/src/main/java/com/starrocks/sql/optimizer/Optimizer.java
index 36a640a7110c7..409397fff1885 100644
--- a/fe/fe-core/src/main/java/com/starrocks/sql/optimizer/Optimizer.java
+++ b/fe/fe-core/src/main/java/com/starrocks/sql/optimizer/Optimizer.java
@@ -25,6 +25,8 @@
 import com.starrocks.common.profile.Tracers;
 import com.starrocks.qe.ConnectContext;
 import com.starrocks.qe.SessionVariable;
+import com.starrocks.qe.feedback.OperatorTuningGuides;
+import com.starrocks.qe.feedback.PlanTuningAdvisor;
 import com.starrocks.sql.Explain;
 import com.starrocks.sql.ast.StatementBase;
 import com.starrocks.sql.optimizer.base.ColumnRefFactory;
@@ -962,6 +964,12 @@ private OptExpression dynamicRewrite(ConnectContext connectContext, TaskContext
         // if we can change the distribution to adjust the plan because of skew data, bad statistics or something else.
         result = new MarkParentRequiredDistributionRule().rewrite(result, rootTaskContext);
         result = new ApplyTuningGuideRule(connectContext).rewrite(result, rootTaskContext);
+
+        OperatorTuningGuides.OptimizedRecord optimizedRecord = PlanTuningAdvisor.getInstance()
+                .getOptimizedRecord(context.getQueryId());
+        if (optimizedRecord != null) {
+            Tracers.record(Tracers.Module.BASE, "DynamicTuningGuides", optimizedRecord.getExplainString());
+        }
         return result;
     }
 
diff --git a/test/sql/test_feedback/R/test_feedback b/test/sql/test_feedback/R/test_feedback
index 538ef6f18394c..43154d507514f 100644
--- a/test/sql/test_feedback/R/test_feedback
+++ b/test/sql/test_feedback/R/test_feedback
@@ -47,13 +47,17 @@ None
 -- !result
 add into plan advisor select count(*) from (select * from c1_skew t1 join (select * from c1_skew where c1 = 'f') t2 on t1.c2 = t2.c2) t;
 -- result:
-[REGEX]Add query into plan advisor in FE.*
+Add query into plan advisor in FE(172.17.0.1_9011_1728457174963) successfully.
 -- !result
 function: assert_explain_contains("select count(*) from (select * from c1_skew t1 join (select * from c1_skew where c1 = 'f') t2 on t1.c2 = t2.c2) t", "RightChildEstimationErrorTuningGuide")
 -- result:
 None
 -- !result
+function: assert_trace_values_contains("select count(*) from (select * from c1_skew t1 join (select * from c1_skew where c1 = 'f') t2 on t1.c2 = t2.c2) t", "RightChildEstimationErrorTuningGuide")
+-- result:
+None
+-- !result
 clear plan advisor;
 -- result:
-[REGEX]Clear all plan advisor in FE.*
--- !result
\ No newline at end of file
+Clear all plan advisor in FE(172.17.0.1_9011_1728457174963) successfully. Advisor size: 1
+-- !result
diff --git a/test/sql/test_feedback/T/test_feedback b/test/sql/test_feedback/T/test_feedback
index 3bff3efab00ab..56bf972255495 100644
--- a/test/sql/test_feedback/T/test_feedback
+++ b/test/sql/test_feedback/T/test_feedback
@@ -26,4 +26,6 @@ set enable_global_runtime_filter = false;
 function: assert_explain_not_contains("select count(*) from (select * from c1_skew t1 join (select * from c1_skew where c1 = 'f') t2 on t1.c2 = t2.c2) t", "RightChildEstimationErrorTuningGuide")
 add into plan advisor select count(*) from (select * from c1_skew t1 join (select * from c1_skew where c1 = 'f') t2 on t1.c2 = t2.c2) t;
 function: assert_explain_contains("select count(*) from (select * from c1_skew t1 join (select * from c1_skew where c1 = 'f') t2 on t1.c2 = t2.c2) t", "RightChildEstimationErrorTuningGuide")
+function: assert_trace_values_contains("select count(*) from (select * from c1_skew t1 join (select * from c1_skew where c1 = 'f') t2 on t1.c2 = t2.c2) t", "RightChildEstimationErrorTuningGuide")
+
 clear plan advisor;