com.griddynamics.jagger.engine.e1.reporting.MetricPlotsReporter.java Source code

Introduction

Here is the source code for com.griddynamics.jagger.engine.e1.reporting.MetricPlotsReporter.java

Source

/*
 * Copyright (c) 2010-2012 Grid Dynamics Consulting Services, Inc, All Rights Reserved
 * http://www.griddynamics.com
 *
 * This library is free software; you can redistribute it and/or modify it under the terms of
 * the GNU Lesser General Public License as published by the Free Software Foundation; either
 * version 2.1 of the License, or any later version.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
 * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */
package com.griddynamics.jagger.engine.e1.reporting;

import com.google.common.collect.Maps;
import com.griddynamics.jagger.engine.e1.aggregator.workload.model.MetricDetails;
import com.griddynamics.jagger.reporting.AbstractMappedReportProvider;
import com.griddynamics.jagger.reporting.chart.ChartHelper;
import com.griddynamics.jagger.util.Pair;
import net.sf.jasperreports.engine.JRDataSource;
import net.sf.jasperreports.engine.data.JRBeanCollectionDataSource;
import net.sf.jasperreports.renderers.JCommonDrawableRenderer;
import org.jfree.chart.JFreeChart;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;

import java.util.*;

/**
 * @author Nikolay Musienko
 *         Date: 19.03.13
 */

public class MetricPlotsReporter extends AbstractMappedReportProvider<String> {

    private Map<String, MetricPlotDTOs> plots;

    public static class MetricPlotDTOs {
        private Collection<MetricPlotDTO> metricPlotDTOs;

        public Collection<MetricPlotDTO> getMetricPlotDTOs() {
            return metricPlotDTOs;
        }

        public void setPlot(Collection<MetricPlotDTO> metricPlotDTOs) {
            this.metricPlotDTOs = metricPlotDTOs;
        }

        public void addPlot(MetricPlotDTO plot) {
            if (metricPlotDTOs == null) {
                metricPlotDTOs = new LinkedList<MetricPlotDTO>();
            }
            metricPlotDTOs.add(plot);
        }
    }

    public static class MetricPlotDTO {
        private JCommonDrawableRenderer metricPlot;
        private String metricName;

        public MetricPlotDTO(String metricName, JCommonDrawableRenderer metricPlot) {
            this.metricPlot = metricPlot;
            this.metricName = metricName;
        }

        public MetricPlotDTO() {
        }

        public JCommonDrawableRenderer getMetricPlot() {
            return metricPlot;
        }

        public void setMetricPlot(JCommonDrawableRenderer metricPlot) {
            this.metricPlot = metricPlot;
        }

        public String getMetricName() {
            return metricName;
        }

        public void setMetricName(String metricName) {
            this.metricName = metricName;
        }
    }

    @Override
    public JRDataSource getDataSource(String testId) {
        if (plots == null) {
            plots = createTaskPlots();
        }
        return new JRBeanCollectionDataSource(Collections.singleton(plots.get(testId)));
    }

    public Map<String, MetricPlotDTOs> createTaskPlots() {
        String sessionId = getSessionIdProvider().getSessionId();
        List<MetricDetails> metricDetails = getHibernateTemplate()
                .find("select m from MetricDetails m where m.taskData.sessionId=?", sessionId);
        Map<String, Map<String, List<MetricDetails>>> aggregatedByTasks = Maps.newLinkedHashMap();
        for (MetricDetails detail : metricDetails) {
            Map<String, List<MetricDetails>> byTaskId = aggregatedByTasks.get(detail.getTaskData().getTaskId());
            if (byTaskId == null) {
                byTaskId = Maps.newLinkedHashMap();
                aggregatedByTasks.put(detail.getTaskData().getTaskId(), byTaskId);
            }
            List<MetricDetails> taskData = byTaskId.get(detail.getMetric());
            if (taskData == null) {
                taskData = new LinkedList<MetricDetails>();
                byTaskId.put(detail.getMetric(), taskData);
            }
            taskData.add(detail);
        }
        Map<String, MetricPlotDTOs> taskPlots = Maps.newHashMap();
        for (String taskId : aggregatedByTasks.keySet()) {
            MetricPlotDTOs taskPlot = taskPlots.get(taskId);
            if (taskPlot == null) {
                taskPlot = new MetricPlotDTOs();
                taskPlots.put(taskId, taskPlot);
            }
            for (String metricName : aggregatedByTasks.get(taskId).keySet()) {
                List<MetricDetails> taskStats = aggregatedByTasks.get(taskId).get(metricName);
                XYSeries plotEntry = new XYSeries(metricName);
                for (MetricDetails stat : taskStats) {
                    plotEntry.add(stat.getTime(), stat.getValue());
                }
                XYSeriesCollection plotCollection = new XYSeriesCollection();
                plotCollection.addSeries(plotEntry);
                Pair<String, XYSeriesCollection> pair = ChartHelper.adjustTime(plotCollection, null);
                plotCollection = pair.getSecond();
                JFreeChart chartMetric = ChartHelper.createXYChart(null, plotCollection,
                        "Time (" + pair.getFirst() + ")", metricName, 2, 2, ChartHelper.ColorTheme.LIGHT);
                taskPlot.addPlot(new MetricPlotDTO(metricName, new JCommonDrawableRenderer(chartMetric)));
            }
        }
        return taskPlots;
    }
}