com.centurylink.mdw.model.report.MetricDataList Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mdw-common Show documentation
MDW is a microservices based workflow framework
There is a newer version: 6.1.39
package com.centurylink.mdw.model.report;

import com.centurylink.mdw.model.JsonObject;
import org.json.JSONArray;
import org.json.JSONObject;

import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

/**
 * Rolling, fixed size list for holding metric data.
 */
public class MetricDataList {

    private static final DateTimeFormatter timeFormatter = DateTimeFormatter.ofPattern("hh:mm:ss");

    private int max;
    private int period;
    private List dataList;

    public MetricDataList(int period, int max) {
        this.period = period;
        this.max = max;
        dataList = new ArrayList<>();
    }

    public void add(MetricData metricData) {
        while (dataList.size() >= max) {
            dataList.remove(0);
        }
        dataList.add(metricData);
    }

    public MetricData get(int i) {
        return dataList.get(i);
    }

    public JSONObject getJson(int span) {
        JSONObject json = new JsonObject();
        for (MetricData metricData : getData(span)) {
            JSONArray jsonArray = new JSONArray();
            for (Metric metric : metricData.getMetrics()) {
                jsonArray.put(metric.getJson());
            }
            json.put(timeFormatter.format(metricData.getTime()), jsonArray);
        }
        return json;
    }

    /**
     * Accumulated averages.
     */
    public List getAverages(int span) {
        Map accum = new LinkedHashMap<>();
        int count = span / period;
        if (count > dataList.size())
            count = dataList.size();
        for (int i = dataList.size() - count; i < dataList.size(); i++) {
            MetricData metricData = dataList.get(i);
            for (Metric metric : metricData.getMetrics()) {
                Metric total = accum.get(metric.getName());
                if (total == null) {
                    total = new Metric(metric.getId(), metric.getName(), metric.getValue());
                    accum.put(metric.getName(), total);
                }
                else {
                    total.setValue(total.getValue() + metric.getValue());
                }
            }
        }
        for (Metric metric : accum.values()) {
            metric.setValue(Math.round((double)metric.getValue() / count));
        }
        return new ArrayList<>(accum.values());
    }

    /**
     * Returns a left-padded list.
     */
    public List getData(int span) {
        int count = span / period;
        if (dataList.size() < count) {
            if (dataList.isEmpty()) {
                return dataList;
            }
            else {
                // left-pad
                List padded = new ArrayList<>(dataList);
                MetricData first = dataList.get(0);
                List pads = new ArrayList<>();
                for (Metric metric : first.getMetrics()) {
                    pads.add(new Metric(metric.getId(), metric.getName(), 0));
                }
                LocalDateTime time = first.getTime();
                while (padded.size() < count) {
                    time = time.minusSeconds(period);
                    padded.add(0, new MetricData(time, pads));
                }
                return padded;
            }
        }
        else {
            return dataList.subList(dataList.size() - count, dataList.size() - 1);
        }
    }

    public int size() {
        return dataList.size();
    }
}