/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.facebook.presto.cost;

import com.facebook.presto.Session;
import com.facebook.presto.sql.analyzer.FeaturesConfig;
import com.google.common.annotations.VisibleForTesting;

import javax.inject.Inject;

import java.util.Comparator;

import static com.google.common.base.Preconditions.checkArgument;
import static java.util.Objects.requireNonNull;

public class CostComparator
{
    private final double cpuWeight;
    private final double memoryWeight;
    private final double networkWeight;

    @Inject
    public CostComparator(FeaturesConfig featuresConfig)
    {
        this(featuresConfig.getCpuCostWeight(), featuresConfig.getMemoryCostWeight(), featuresConfig.getNetworkCostWeight());
    }

    @VisibleForTesting
    public CostComparator(double cpuWeight, double memoryWeight, double networkWeight)
    {
        checkArgument(cpuWeight >= 0, "cpuWeight cannot be negative");
        checkArgument(memoryWeight >= 0, "memoryWeight cannot be negative");
        checkArgument(networkWeight >= 0, "networkWeight cannot be negative");
        this.cpuWeight = cpuWeight;
        this.memoryWeight = memoryWeight;
        this.networkWeight = networkWeight;
    }

    public Comparator<PlanNodeCostEstimate> forSession(Session session)
    {
        requireNonNull(session, "session is null");
        return (left, right) -> this.compare(session, left, right);
    }

    public int compare(Session session, PlanNodeCostEstimate left, PlanNodeCostEstimate right)
    {
        requireNonNull(session, "session is null");
        requireNonNull(left, "left is null");
        requireNonNull(right, "right is null");
        checkArgument(!left.hasUnknownComponents() && !right.hasUnknownComponents(), "cannot compare unknown costs");

        double leftCost = left.getCpuCost() * cpuWeight
                + left.getMemoryCost() * memoryWeight
                + left.getNetworkCost() * networkWeight;

        double rightCost = right.getCpuCost() * cpuWeight
                + right.getMemoryCost() * memoryWeight
                + right.getNetworkCost() * networkWeight;

        return Double.compare(leftCost, rightCost);
    }
}

@@ -742,65 +742,78 @@
 ng;%0A
-%0A
 import 
-javax.inject.Inject;%0A%0Aimport java.util.Comparator
+com.google.common.collect.Ordering;%0A%0Aimport javax.inject.Inject
 ;%0A%0Ai
@@ -1763,26 +1763,24 @@
  public 
-Comparator
+Ordering
 %3CPlanNod
@@ -1895,16 +1895,30 @@
  return 
+Ordering.from(
 (left, r
@@ -1960,16 +1960,17 @@
 , right)
+)
 ;%0A    %7D%0A

Create Ordering in CostComparator