Review

apache · Oct 11, 2023 · 0def1aa · 0def1aa
1 parent 6766b34
commit 0def1aa
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/...nsions-core/multi-stage-query/src/main/java/org/apache/druid/msq/exec/ControllerImpl.java b/...nsions-core/multi-stage-query/src/main/java/org/apache/druid/msq/exec/ControllerImpl.java
@@ -1771,10 +1771,7 @@ private static QueryDefinition makeQueryDefinition(
       StageDefinition finalShuffleStageDef = queryDef.getFinalStageDefinition();
       if (finalShuffleStageDef.doesSortDuringShuffle()) {
         final QueryDefinitionBuilder builder = QueryDefinition.builder();
-        for (final StageDefinition stageDef : queryDef.getStageDefinitions()) {
-          builder.add(StageDefinition.builder(stageDef));
-        }
-
+        builder.addAll(queryDef);
         builder.add(StageDefinition.builder(queryDef.getNextStageNumber())
                                    .inputs(new StageInputSpec(queryDef.getFinalStageDefinition().getStageNumber()))
                                    .maxWorkerCount(tuningConfig.getMaxNumWorkers())