adarshsanjeev commented on code in PR #13205:
URL: https://github.com/apache/druid/pull/13205#discussion_r1022435808


##########
extensions-core/multi-stage-query/src/main/java/org/apache/druid/msq/kernel/controller/ControllerStageTracker.java:
##########
@@ -227,19 +221,28 @@ WorkerInputs getWorkerInputs()
     return workerInputs;
   }
 
+  /**
+   * Returns the merged key statistics.
+   */
+  @Nullable
+  public WorkerAggregatedKeyStatistics getAggregatedKeyStatistics()
+  {
+    return aggregatedKeyStatistics;
+  }
+
   /**
    * Adds result key statistics for a particular worker number. If statistics 
have already been added for this worker,
    * then this call ignores the new ones and does nothing.
    *
    * @param workerNumber the worker
-   * @param snapshot     worker statistics
+   * @param aggregatedKeyStatistics aggregated key statistics
    */
-  ControllerStagePhase addResultKeyStatisticsForWorker(
+  ControllerStagePhase addAggregatedStatisticsForWorker(
       final int workerNumber,
-      final ClusterByStatisticsSnapshot snapshot
+      final WorkerAggregatedKeyStatistics aggregatedKeyStatistics

Review Comment:
   Changed to PartialKeyStatisticsInformation



##########
extensions-core/multi-stage-query/src/main/java/org/apache/druid/msq/kernel/controller/ControllerStageTracker.java:
##########
@@ -227,19 +221,28 @@ WorkerInputs getWorkerInputs()
     return workerInputs;
   }
 
+  /**
+   * Returns the merged key statistics.
+   */
+  @Nullable
+  public WorkerAggregatedKeyStatistics getAggregatedKeyStatistics()
+  {
+    return aggregatedKeyStatistics;
+  }
+
   /**
    * Adds result key statistics for a particular worker number. If statistics 
have already been added for this worker,
    * then this call ignores the new ones and does nothing.
    *
    * @param workerNumber the worker
-   * @param snapshot     worker statistics
+   * @param aggregatedKeyStatistics aggregated key statistics
    */
-  ControllerStagePhase addResultKeyStatisticsForWorker(
+  ControllerStagePhase addAggregatedStatisticsForWorker(
       final int workerNumber,
-      final ClusterByStatisticsSnapshot snapshot
+      final WorkerAggregatedKeyStatistics aggregatedKeyStatistics

Review Comment:
   Changed to PartialKeyStatisticsInformation. Is this fine?



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to