com.netflix.search.query.report.summary.SummaryReportItem Maven / Gradle / Ivy

Go to download
/**
 * Copyright 2016 Netflix, Inc.
 *
 * 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.netflix.search.query.report.summary;

import java.util.Map;

import com.netflix.search.query.report.ReportItem;
import com.netflix.search.query.report.ResultType;

public class SummaryReportItem extends ReportItem {
    public SummaryReportItem(String name, Integer titles, Integer queries, Double precision, Double recall, Double fmeasure, Map counters) {
        super();

        setValue(SummaryReportHeader.name.toString(), name);
        setValue(SummaryReportHeader.titles.toString(), String.valueOf(titles));
        setValue(SummaryReportHeader.queries.toString(), String.valueOf(queries));
        for (ResultType type : ResultType.values()) {
            Integer counter = counters.get(type);
            if (counter == null)
                counter = 0;
            setValue(type.toString(), String.valueOf(counter));
        }
        setValue(SummaryReportHeader.precision.toString(), (String.format("%.2f", (precision * 100)) + SummaryReport.PERCENT_SIGN));
        setValue(SummaryReportHeader.recall.toString(), (String.format("%.2f", (recall * 100)) + SummaryReport.PERCENT_SIGN));
        setValue(SummaryReportHeader.fmeasure.toString(), (String.format("%.2f", (fmeasure * 100)) + SummaryReport.PERCENT_SIGN));
        setKey(getKeyFromNamedValues());
    }

    public SummaryReportItem(Map namedValues) {
        super(namedValues);
    }

    public SummaryReportItem() {
        super();
    }

    @Override
    protected void appendKeyFromNamedValues(String headerValue, String value)
    {
        if (headerValue.equals(SummaryReportHeader.name.toString()))
            setKey(value);
    }

    @Override
    protected String getKeyFromNamedValues()
    {
        return getNamedValues().get(SummaryReportHeader.name.toString());
    }

    @Override
    public boolean equals(Object obj)
    {
        if (this == obj)
            return true;
        if (obj == null)
            return false;
        if (getClass() != obj.getClass())
            return false;
        SummaryReportItem other = (SummaryReportItem) obj;
        if (getKey() == null) {
            if (other.getKey() != null)
                return false;
        } else if (!getKey().equals(other.getKey()))
            return false;
        return true;
    }

}