package kg.apc.jmeter.vizualizers;

 import java.awt.Color;

 import kg.apc.charting.AbstractGraphRow;

 import kg.apc.jmeter.JMeterPluginsUtils;

 import kg.apc.jmeter.graphs.AbstractOverTimeVisualizer;

 import org.apache.jmeter.samplers.SampleResult;

 public class TransactionsPerSecondGui

         extends AbstractOverTimeVisualizer {

     //private static final Logger log = LoggingManager.getLoggerForClass();

     /**

      *

      */

     private static String labelAggSuccess = "Successful Transactions per Second";

     private static String labelAggFailure = "Failed Transactions per Second";

     public TransactionsPerSecondGui() {

         super();

         setGranulation(1000);

         graphPanel.getGraphObject().setYAxisLabel("Number of transactions /sec");

     }

     private void addTransaction(boolean isSuccess, String rowName, long time, double count) {

         String realRowName;

         String rowAggName;

         if (isSuccess) {

             realRowName = rowName + " (success)";

             rowAggName = labelAggSuccess;

         } else {

             realRowName = rowName + " (failure)";

             rowAggName = labelAggFailure;

         }

         AbstractGraphRow row = model.get(realRowName);

         AbstractGraphRow rowAgg = modelAggregate.get(rowAggName);

         if (row == null) {

             row = getNewRow(model, AbstractGraphRow.ROW_SUM_VALUES, realRowName, AbstractGraphRow.MARKER_SIZE_SMALL, false, false, false, true, true);

         }

         if (rowAgg == null) {

             rowAgg = getNewRow(modelAggregate, AbstractGraphRow.ROW_SUM_VALUES, rowAggName, AbstractGraphRow.MARKER_SIZE_SMALL, false, false, false, true, isSuccess ? Color.GREEN : Color.RED, true);

         }

         //fix to have trans/sec values in all cases

         if (getGranulation() > 0) {

             double tps = count * 1000.0 / getGranulation();

             row.add(time, tps);

             rowAgg.add(time, tps);

             //always add 0 to agg failure row

             if (isSuccess) {

                 rowAgg = modelAggregate.get(labelAggFailure);

                 if (rowAgg == null) {

                     rowAgg = getNewRow(modelAggregate, AbstractGraphRow.ROW_SUM_VALUES, labelAggFailure, AbstractGraphRow.MARKER_SIZE_SMALL, false, false, false, true, Color.RED, true);

                 }

                 rowAgg.add(time, 0);

             }

         }

     }

     @Override

     public String getLabelResource() {

         return this.getClass().getSimpleName();

     }

     @Override

     public String getStaticLabel() {

         return JMeterPluginsUtils.prefixLabel("Transactions per Second");

     }

     @Override

     public void add(SampleResult res) {

         if (!isSampleIncluded(res)) {

             return;

         }

         super.add(res);

         if (res.isSuccessful()) {

             addTransaction(true, res.getSampleLabel(), normalizeTime(res.getEndTime()), 1);

         } else {

             addTransaction(false, res.getSampleLabel(), normalizeTime(res.getEndTime()), 1);

         }

         updateGui(null);

     }

     @Override

     protected JSettingsPanel createSettingsPanel() {

         return new JSettingsPanel(this,

                 JSettingsPanel.TIMELINE_OPTION

                 | JSettingsPanel.GRADIENT_OPTION

                 | JSettingsPanel.FINAL_ZEROING_OPTION

                 | JSettingsPanel.LIMIT_POINT_OPTION

                 | JSettingsPanel.AGGREGATE_OPTION

                 | JSettingsPanel.MAXY_OPTION

                 | JSettingsPanel.RELATIVE_TIME_OPTION

                 | JSettingsPanel.MARKERS_OPTION);

     }

     @Override

     public String getWikiPage() {

         return "TransactionsPerSecond";

     }

 }