ambari-metrics-alertservice/src/main/java/org/apache/ambari/metrics/alertservice/common/TimelineMetrics.java

package org.apache.ambari.metrics.alertservice.common;

import org.apache.hadoop.classification.InterfaceAudience;
import org.apache.hadoop.classification.InterfaceStability;

import javax.xml.bind.annotation.XmlAccessType;
import javax.xml.bind.annotation.XmlAccessorType;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlRootElement;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;

/**
 * The class that hosts a list of timeline entities.
 */
@XmlRootElement(name = "metrics")
@XmlAccessorType(XmlAccessType.NONE)
@InterfaceAudience.Public
@InterfaceStability.Unstable
public class TimelineMetrics implements Serializable {

    private List<TimelineMetric> allMetrics = new ArrayList<TimelineMetric>();

    public TimelineMetrics() {}

    @XmlElement(name = "metrics")
    public List<TimelineMetric> getMetrics() {
        return allMetrics;
    }

    public void setMetrics(List<TimelineMetric> allMetrics) {
        this.allMetrics = allMetrics;
    }

    private boolean isEqualTimelineMetrics(TimelineMetric metric1,
                                           TimelineMetric metric2) {

        boolean isEqual = true;

        if (!metric1.getMetricName().equals(metric2.getMetricName())) {
            return false;
        }

        if (metric1.getHostName() != null) {
            isEqual = metric1.getHostName().equals(metric2.getHostName());
        }

        if (metric1.getAppId() != null) {
            isEqual = metric1.getAppId().equals(metric2.getAppId());
        }

        return isEqual;
    }

    /**
     * Merge with existing TimelineMetric if everything except startTime is
     * the same.
     * @param metric {@link TimelineMetric}
     */
    public void addOrMergeTimelineMetric(TimelineMetric metric) {
        TimelineMetric metricToMerge = null;

        if (!allMetrics.isEmpty()) {
            for (TimelineMetric timelineMetric : allMetrics) {
                if (timelineMetric.equalsExceptTime(metric)) {
                    metricToMerge = timelineMetric;
                    break;
                }
            }
        }

        if (metricToMerge != null) {
            metricToMerge.addMetricValues(metric.getMetricValues());
            if (metricToMerge.getTimestamp() > metric.getTimestamp()) {
                metricToMerge.setTimestamp(metric.getTimestamp());
            }
            if (metricToMerge.getStartTime() > metric.getStartTime()) {
                metricToMerge.setStartTime(metric.getStartTime());
            }
        } else {
            allMetrics.add(metric);
        }
    }

    // Optimization that addresses too many TreeMaps from getting created.
    public void addOrMergeTimelineMetric(SingleValuedTimelineMetric metric) {
        TimelineMetric metricToMerge = null;

        if (!allMetrics.isEmpty()) {
            for (TimelineMetric timelineMetric : allMetrics) {
                if (metric.equalsExceptTime(timelineMetric)) {
                    metricToMerge = timelineMetric;
                    break;
                }
            }
        }

        if (metricToMerge != null) {
            metricToMerge.getMetricValues().put(metric.getTimestamp(), metric.getValue());
            if (metricToMerge.getTimestamp() > metric.getTimestamp()) {
                metricToMerge.setTimestamp(metric.getTimestamp());
            }
            if (metricToMerge.getStartTime() > metric.getStartTime()) {
                metricToMerge.setStartTime(metric.getStartTime());
            }
        } else {
            allMetrics.add(metric.getTimelineMetric());
        }
    }
}