io.prestosql.tests.statistics.MetricComparison Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of presto-tests Show documentation
There is a newer version: 350
/*
 * 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 io.prestosql.tests.statistics;

import java.util.OptionalDouble;

import static io.prestosql.tests.statistics.MetricComparison.Result.DIFFER;
import static io.prestosql.tests.statistics.MetricComparison.Result.MATCH;
import static io.prestosql.tests.statistics.MetricComparison.Result.NO_BASELINE;
import static io.prestosql.tests.statistics.MetricComparison.Result.NO_ESTIMATE;
import static java.lang.String.format;
import static java.util.Objects.requireNonNull;

public class MetricComparison
{
    private final Metric metric;
    private final OptionalDouble estimatedValue;
    private final OptionalDouble actualValue;

    public MetricComparison(Metric metric, OptionalDouble estimatedValue, OptionalDouble actualValue)
    {
        this.metric = metric;
        this.estimatedValue = estimatedValue;
        this.actualValue = actualValue;
    }

    public Metric getMetric()
    {
        return metric;
    }

    @Override
    public String toString()
    {
        return format("Metric [%s] - estimated: [%s], real: [%s]",
                metric, print(estimatedValue), print(actualValue));
    }

    public Result result(MetricComparisonStrategy metricComparisonStrategy)
    {
        requireNonNull(metricComparisonStrategy, "metricComparisonStrategy is null");

        if (!estimatedValue.isPresent() && !actualValue.isPresent()) {
            return MATCH;
        }
        if (!estimatedValue.isPresent()) {
            return NO_ESTIMATE;
        }
        if (!actualValue.isPresent()) {
            return NO_BASELINE;
        }
        return metricComparisonStrategy.matches(actualValue.getAsDouble(), estimatedValue.getAsDouble()) ? MATCH : DIFFER;
    }

    private String print(OptionalDouble value)
    {
        if (!value.isPresent()) {
            return "UNKNOWN";
        }
        return String.valueOf(value.getAsDouble());
    }

    public enum Result
    {
        NO_ESTIMATE,
        NO_BASELINE,
        DIFFER,
        MATCH
    }
}