package kg.apc.cmdtools;
import kg.apc.jmeter.PluginsCMDWorker;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;
import org.apache.log.Priority;
import java.io.PrintStream;
import java.util.ListIterator;
public class ReporterTool extends AbstractCMDTool {
private static final Logger log = LoggingManager.getLoggerForClass();
@Override
protected void showHelp(PrintStream os) {
os.println("Options for tool 'Reporter': --generate-png <filename> "
+ "--generate-csv <filename> "
+ "--input-jtl <data file> "
+ "--plugin-type <type> "
+ "[ "
+ "--width <graph width> "
+ "--height <graph height> "
+ "--granulation <ms> "
+ "--relative-times <yes/no> " // use relative X axis times, _no_ will set absolute times ||
+ "--aggregate-rows <yes/no> "
+ "--paint-gradient <yes/no> "
+ "--paint-zeroing <yes/no> "
+ "--paint-markers <yes/no> "
+ "--prevent-outliers <yes/no> " // prevent outliers on distribution graph ||
+ "--limit-rows <num of points> " // limit number of points in row ||
+ "--force-y <limit> " // force Y axis limit ||
+ "--hide-low-counts <limit> " // hide points with sample count below limit ||
+ "--include-labels <labels list> " // filter samples
+ "--exclude-labels <labels list> " // filter samples
+ "--auto-scale <yes/no> " // scaling for composites ||
+ "--line-weight <num of pixels> " // set graph row line thikness ||
+ "--extractor-regexps <regExps list> "
+ "--success-filter <true/false> "
+ "--include-label-regex <true/false> " // filter samples label with regex
+ "--exclude-label-regex <true/false> " // filter samples label with regex
+ "--start-offset <sec>" // filter samples on period time
+ "--end-offset <sec>" // filter samples on period time
+ "]");
}
@Override
protected int processParams(ListIterator args) throws UnsupportedOperationException, IllegalArgumentException {
LoggingManager.setPriority(Priority.INFO);
// first process params without worker created
while (args.hasNext()) {
String nextArg = (String) args.next();
if (nextArg.equals("--loglevel")) {
args.remove();
String loglevelStr = (String) args.next();
args.remove();
LoggingManager.setPriority(loglevelStr);
}
}
// rewind it
while (args.hasPrevious()) {
args.previous();
}
PluginsCMDWorker worker = new PluginsCMDWorker();
while (args.hasNext()) {
String nextArg = (String) args.next();
log.debug("Arg: " + nextArg);
if (nextArg.equalsIgnoreCase("--generate-png")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing PNG file name");
}
worker.addExportMode(PluginsCMDWorker.EXPORT_PNG);
worker.setOutputPNGFile((String) args.next());
} else if (nextArg.equalsIgnoreCase("--generate-csv")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing CSV file name");
}
worker.addExportMode(PluginsCMDWorker.EXPORT_CSV);
worker.setOutputCSVFile((String) args.next());
} else if (nextArg.equalsIgnoreCase("--input-jtl")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing input JTL file name");
}
worker.setInputFile((String) args.next());
} else if (nextArg.equalsIgnoreCase("--plugin-type")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing plugin type");
}
worker.setPluginType((String) args.next());
} else if (nextArg.equalsIgnoreCase("--width")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing width specification");
}
worker.setGraphWidth(Integer.parseInt((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--height")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing height specification");
}
worker.setGraphHeight(Integer.parseInt((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--aggregate-rows")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing aggregate flag");
}
worker.setAggregate(getLogicValue((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--paint-zeroing")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing zeroing flag");
}
worker.setZeroing(getLogicValue((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--relative-times")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing rel time flag");
}
worker.setRelativeTimes(getLogicValue((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--paint-gradient")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing gradient flag");
}
worker.setGradient(getLogicValue((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--paint-markers")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing markers flag");
}
worker.setMarkers(getLogicValue((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--prevent-outliers")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing outliers flag");
}
worker.setPreventOutliers(getLogicValue((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--limit-rows")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing limit rows specification");
}
worker.setRowsLimit(Integer.parseInt((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--force-y")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing limit Y specification");
}
worker.setForceY(Integer.parseInt((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--hide-low-counts")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing low counts specification");
}
worker.setHideLowCounts(Integer.parseInt((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--line-weight")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing line thickness specification");
}
worker.setLineWeight(Float.parseFloat((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--granulation")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing granulation specification");
}
worker.setGranulation(Integer.parseInt((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--include-labels")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing granulation specification");
}
worker.setIncludeLabels((String) args.next());
} else if (nextArg.equalsIgnoreCase("--exclude-labels")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing granulation specification");
}
worker.setExcludeLabels((String) args.next());
} else if (nextArg.equalsIgnoreCase("--auto-scale")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing auto-zoom flag value");
}
worker.setAutoScaleRows(getLogicValue((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--success-filter")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing success filter flag");
}
worker.setSuccessFilter(getLogicValue((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--include-label-regex")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing include label regex flag");
}
worker.setIncludeSamplesWithRegex(getLogicValue((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--exclude-label-regex")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing exclude label regex flag");
}
worker.setExcludeSamplesWithRegex(getLogicValue((String) args.next()));
} else if (nextArg.equalsIgnoreCase("--start-offset")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing start offset flag");
}
worker.setStartOffset((String) args.next());
} else if (nextArg.equalsIgnoreCase("--end-offset")) {
if (!args.hasNext()) {
throw new IllegalArgumentException("Missing end offset flag");
}
worker.setEndOffset((String) args.next());
} else {
worker.processUnknownOption(nextArg, args);
}
}
return worker.doJob();
}
}