[enhancement](util) print if using nereids planner when be coredump #31981

be/src/common/signal_handler.h

@@ -54,6 +54,7 @@ namespace doris::signal {
 inline thread_local uint64 query_id_hi;
 inline thread_local uint64 query_id_lo;
 inline thread_local int64_t tablet_id = 0;
+inline thread_local bool is_nereids = false;
 
 namespace {
 
@@ -244,6 +245,9 @@ void DumpTimeInfo() {
     formatter.AppendString("-");
     formatter.AppendUint64(query_id_lo, 16);
     formatter.AppendString(" ***\n");
+    formatter.AppendString("*** is nereids: ");
+    formatter.AppendUint64(is_nereids, 10);
+    formatter.AppendString(" ***\n");
     formatter.AppendString("*** tablet id: ");
     formatter.AppendUint64(tablet_id, 10);
     formatter.AppendString(" ***\n");
@@ -436,6 +440,10 @@ inline void set_signal_task_id(TUniqueId tid) {
     query_id_lo = tid.lo;
 }
 
+inline void set_signal_is_nereids(bool is_nereids_arg) {
+    is_nereids = is_nereids_arg;
+}
+
 inline void InstallFailureSignalHandler() {
     // Build the sigaction struct.
     struct sigaction sig_action;

be/src/runtime/fragment_mgr.cpp

@@ -681,7 +681,7 @@ Status FragmentMgr::_get_query_ctx(const Params& params, TUniqueId query_id, boo
         // This may be a first fragment request of the query.
         // Create the query fragments context.
         query_ctx = QueryContext::create_shared(params.fragment_num_on_host, _exec_env,
-                                                params.query_options);
+                                                params.query_options,params.is_nereids);
         query_ctx->query_id = query_id;
         RETURN_IF_ERROR(DescriptorTbl::create(&(query_ctx->obj_pool), params.desc_tbl,
                                               &(query_ctx->desc_tbl)));
@@ -812,6 +812,7 @@ Status FragmentMgr::exec_plan_fragment(const TExecPlanFragmentParams& params,
     std::shared_ptr<QueryContext> query_ctx;
     bool pipeline_engine_enabled = params.query_options.__isset.enable_pipeline_engine &&
                                    params.query_options.enable_pipeline_engine;
+
     RETURN_IF_ERROR(
             _get_query_ctx(params, params.params.query_id, pipeline_engine_enabled, query_ctx));
     {

be/src/runtime/query_context.h

@@ -51,8 +51,10 @@ class QueryContext {
     ENABLE_FACTORY_CREATOR(QueryContext);
 
 public:
-    QueryContext(int total_fragment_num, ExecEnv* exec_env, const TQueryOptions& query_options)
-            : fragment_num(total_fragment_num),
+    QueryContext(int total_fragment_num, ExecEnv* exec_env, const TQueryOptions& query_options,
+                 bool is_nereids)
+            : _is_nereids(is_nereids),
+              fragment_num(total_fragment_num),
               timeout_second(-1),
               _exec_env(exec_env),
               _runtime_filter_mgr(new RuntimeFilterMgr(TUniqueId(), this)),
@@ -222,12 +224,16 @@ class QueryContext {
 public:
     TUniqueId query_id;
     DescriptorTbl* desc_tbl;
+
+    bool is_nereids() const { return _is_nereids; }
+
     bool set_rsc_info = false;
     std::string user;
     std::string group;
     TNetworkAddress coord_addr;
     TNetworkAddress current_connect_fe;
     TQueryGlobals query_globals;
+    const bool _is_nereids;
 
     /// In the current implementation, for multiple fragments executed by a query on the same BE node,
     /// we store some common components in QueryContext, and save QueryContext in FragmentMgr.

be/src/runtime/runtime_state.cpp

@@ -34,6 +34,7 @@
 #include "runtime/load_path_mgr.h"
 #include "runtime/memory/mem_tracker_limiter.h"
 #include "runtime/memory/thread_mem_tracker_mgr.h"
+#include "runtime/query_context.h"
 #include "runtime/runtime_filter_mgr.h"
 #include "runtime/thread_context.h"
 #include "util/timezone_utils.h"
@@ -390,4 +391,7 @@ bool RuntimeState::enable_page_cache() const {
            (_query_options.__isset.enable_page_cache && _query_options.enable_page_cache);
 }
 
+bool RuntimeState::is_nereids() const {
+    return _query_ctx->is_nereids();
+}
 } // end namespace doris

be/src/runtime/runtime_state.h

@@ -450,6 +450,9 @@ class RuntimeState {
                        : 0;
     }
 
+
+    bool is_nereids() const;
+
 private:
     Status create_error_log_file();
 

be/src/runtime/thread_context.cpp

@@ -40,6 +40,7 @@ AttachTask::AttachTask(const std::shared_ptr<MemTrackerLimiter>& mem_tracker,
 AttachTask::AttachTask(RuntimeState* runtime_state) {
     SwitchBthreadLocal::switch_to_bthread_local();
     signal::set_signal_task_id(runtime_state->query_id());
+    signal::set_signal_is_nereids(runtime_state->is_nereids());
     thread_context()->attach_task(runtime_state->query_id(), runtime_state->fragment_instance_id(),
                                   runtime_state->query_mem_tracker());
 }

fe/fe-core/src/main/java/org/apache/doris/qe/Coordinator.java

@@ -3203,6 +3203,7 @@ List<TExecPlanFragmentParams> toThrift(int backendNum) {
             for (int i = 0; i < instanceExecParams.size(); ++i) {
                 final FInstanceExecParam instanceExecParam = instanceExecParams.get(i);
                 TExecPlanFragmentParams params = new TExecPlanFragmentParams();
+                params.setIsNereids(ConnectContext.get() != null ? ConnectContext.get().getState().isNereids() : false);
                 params.setProtocolVersion(PaloInternalServiceVersion.V1);
                 params.setFragment(fragment.toThrift());
                 params.setDescTbl(descTable);
@@ -3302,6 +3303,8 @@ Map<TNetworkAddress, TPipelineFragmentParams> toTPipelineParams(int backendNum)
                     TPipelineFragmentParams params = new TPipelineFragmentParams();
 
                     // Set global param
+                    params.setIsNereids(
+                            ConnectContext.get() != null ? ConnectContext.get().getState().isNereids() : false);
                     params.setProtocolVersion(PaloInternalServiceVersion.V1);
                     params.setDescTbl(descTable);
                     params.setQueryId(queryId);

gensrc/thrift/PaloInternalService.thrift

@@ -464,6 +464,8 @@ struct TExecPlanFragmentParams {
   // scan node id -> scan range params, only for external file scan
   24: optional map<Types.TPlanNodeId, PlanNodes.TFileScanRangeParams> file_scan_params
 
+  31: optional bool is_nereids = true;
+
   32: optional Types.TNetworkAddress current_connect_fe
 }
 
@@ -680,6 +682,8 @@ struct TPipelineFragmentParams {
   // scan node id -> scan range params, only for external file scan
   29: optional map<Types.TPlanNodeId, PlanNodes.TFileScanRangeParams> file_scan_params
   43: optional Types.TNetworkAddress current_connect_fe
+
+  40: optional bool is_nereids = true;
 }
 
 struct TPipelineFragmentParamsList {