fix

englefly · Dec 5, 2023 · 90273d1 · 90273d1
1 parent 34c85c9
commit 90273d1
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/fe/fe-core/src/main/java/org/apache/doris/common/profile/PlanTreeBuilder.java b/fe/fe-core/src/main/java/org/apache/doris/common/profile/PlanTreeBuilder.java
@@ -67,10 +67,16 @@ private void buildFragmentPlans() {
                 }
                 sb.append("\n[Fragment: ").append(fragment.getFragmentSequenceNum()).append("]");
                 sb.append("\n").append(sink.getExplainString("", TExplainLevel.BRIEF));
-                sinkNode = new PlanTreeNode(
-                        sink instanceof MultiCastDataSink ? ((MultiCastDataSink) sink).getDataStreamSinks().stream()
-                                .map(s -> s.getExchNodeId()).collect(Collectors.toList())
-                                : ImmutableList.of(sink.getExchNodeId()), sb.toString());
+                List<PlanNodeId> exchangeIds;
+                if (sink instanceof MultiCastDataSink) {
+                    exchangeIds = ((MultiCastDataSink) sink).getDataStreamSinks().stream()
+                            .map(s -> s.getExchNodeId()).collect(Collectors.toList());
+                } else if (sink.getExchNodeId() != null) {
+                    exchangeIds = ImmutableList.of(sink.getExchNodeId());
+                } else {
+                    exchangeIds = ImmutableList.of();
+                }
+                sinkNode = new PlanTreeNode(exchangeIds, sb.toString());
                 if (i == 0) {
                     // sink of first fragment, set it as tree root
                     treeRoot = sinkNode;