diff --git a/presto-main/src/main/java/com/facebook/presto/cost/TaskCountEstimator.java b/presto-main/src/main/java/com/facebook/presto/cost/TaskCountEstimator.java index 3f192bea26eb4..f553ab080c542 100644 --- a/presto-main/src/main/java/com/facebook/presto/cost/TaskCountEstimator.java +++ b/presto-main/src/main/java/com/facebook/presto/cost/TaskCountEstimator.java @@ -34,9 +34,10 @@ public TaskCountEstimator(NodeSchedulerConfig nodeSchedulerConfig, InternalNodeM { requireNonNull(nodeSchedulerConfig, "nodeSchedulerConfig is null"); requireNonNull(nodeManager, "nodeManager is null"); + boolean schedulerIncludeCoordinator = nodeSchedulerConfig.isIncludeCoordinator(); this.numberOfNodes = () -> { Set activeNodes = nodeManager.getAllNodes().getActiveNodes(); - if (nodeSchedulerConfig.isIncludeCoordinator()) { + if (schedulerIncludeCoordinator) { return activeNodes.size(); } return toIntExact(activeNodes.stream()