From 3b56c4bcfa6d35348b6a32aa307d767b387aed2b Mon Sep 17 00:00:00 2001
From: starocean999 <40539150+starocean999@users.noreply.github.com>
Date: Thu, 7 Mar 2024 16:23:35 +0800
Subject: [PATCH] [enhancement](nereids)send is_nereids flag to be (#31752)

---
 fe/fe-core/src/main/java/org/apache/doris/qe/Coordinator.java | 2 ++
 gensrc/thrift/PaloInternalService.thrift                      | 3 +++
 2 files changed, 5 insertions(+)

diff --git a/fe/fe-core/src/main/java/org/apache/doris/qe/Coordinator.java b/fe/fe-core/src/main/java/org/apache/doris/qe/Coordinator.java
index 872a149b17..85a6516153 100644
--- a/fe/fe-core/src/main/java/org/apache/doris/qe/Coordinator.java
+++ b/fe/fe-core/src/main/java/org/apache/doris/qe/Coordinator.java
@@ -3605,6 +3605,7 @@ public class Coordinator implements CoordInterface {
             for (int i = 0; i < instanceExecParams.size(); ++i) {
                 final FInstanceExecParam instanceExecParam = instanceExecParams.get(i);
                 TExecPlanFragmentParams params = new TExecPlanFragmentParams();
+                params.setIsNereids(context != null ? context.getState().isNereids() : false);
                 params.setProtocolVersion(PaloInternalServiceVersion.V1);
                 params.setFragment(fragment.toThrift());
                 params.setDescTbl(descTable);
@@ -3717,6 +3718,7 @@ public class Coordinator implements CoordInterface {
                     TPipelineFragmentParams params = new TPipelineFragmentParams();
 
                     // Set global param
+                    params.setIsNereids(context != null ? context.getState().isNereids() : false);
                     params.setProtocolVersion(PaloInternalServiceVersion.V1);
                     params.setDescTbl(descTable);
                     params.setQueryId(queryId);
diff --git a/gensrc/thrift/PaloInternalService.thrift b/gensrc/thrift/PaloInternalService.thrift
index 8cdf758f6d..64024bed2a 100644
--- a/gensrc/thrift/PaloInternalService.thrift
+++ b/gensrc/thrift/PaloInternalService.thrift
@@ -496,6 +496,8 @@ struct TExecPlanFragmentParams {
 
   30: optional list<TPipelineWorkloadGroup> workload_groups
 
+  31: optional bool is_nereids = true;
+
   // For cloud
   1000: optional bool is_mow_table;
 }
@@ -716,6 +718,7 @@ struct TPipelineFragmentParams {
   37: optional i32 parallel_instances
   38: optional i32 total_instances
   39: optional map<i32, i32> shuffle_idx_to_instance_idx
+  40: optional bool is_nereids = true;
 
   // For cloud
   1000: optional bool is_mow_table;