Fix build

apache · Jun 27, 2024 · 053f2e9 · 053f2e9
1 parent bdecad9
commit 053f2e9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...re/multi-stage-query/src/main/java/org/apache/druid/msq/indexing/MSQCompactionRunner.java b/...re/multi-stage-query/src/main/java/org/apache/druid/msq/indexing/MSQCompactionRunner.java
@@ -256,13 +256,15 @@ private static MSQTuningConfig buildMSQTuningConfig(CompactionTask compactionTas
     // We don't consider maxRowsInMemory coming via CompactionTuningConfig since it always sets a default value if no
     // value specified by user.
     final int maxRowsInMemory = MultiStageQueryContext.getRowsInMemory(compactionTaskContext);
+    final Integer maxNumSegments = MultiStageQueryContext.getMaxNumSegments(compactionTaskContext);
 
     Integer rowsPerSegment = getRowsPerSegment(compactionTask);
 
     return new MSQTuningConfig(
         maxNumWorkers,
         maxRowsInMemory,
         rowsPerSegment,
+        maxNumSegments,
         compactionTask.getTuningConfig() != null ? compactionTask.getTuningConfig().getIndexSpec() : null
     );
   }