package kg.apc.jmeter.reporters;
import java.text.SimpleDateFormat;
import java.util.Arrays;
import java.util.Date;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.SortedMap;
import java.util.Stack;
import java.util.TreeMap;
import net.sf.json.JSONArray;
import net.sf.json.JSONObject;
import org.apache.jmeter.samplers.SampleResult;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.log.Logger;
public class LoadosophiaAggregator {
private static final Logger log = LoggingManager.getLoggerForClass();
private SortedMap<Long, List<SampleResult>> buffer = new TreeMap<Long, List<SampleResult>>();
private static final long SEND_SECONDS = 5;
private long lastTime = 0;
private SimpleDateFormat format = new SimpleDateFormat("yyyy-MM-dd'T'HH:mm:ssZ");
public void addSample(SampleResult res) {
if (log.isDebugEnabled()) {
log.debug("Got sample to process: " + res);
}
Long time = res.getEndTime() / 1000;
if (!buffer.containsKey(time)) {
// we need to create new sec list
if (time < lastTime) {
// a problem with times sequence - taking last available
Iterator<Long> it = buffer.keySet().iterator();
while (it.hasNext()) {
time = it.next();
}
}
buffer.put(time, new LinkedList<SampleResult>());
}
lastTime = time;
buffer.get(time).add(res);
}
public boolean haveDataToSend() {
return buffer.size() > SEND_SECONDS + 1;
}
public JSONArray getDataToSend() {
JSONArray data = new JSONArray();
Iterator<Long> it = buffer.keySet().iterator();
int cnt = 0;
while (cnt < SEND_SECONDS && it.hasNext()) {
Long sec = it.next();
List<SampleResult> raw = buffer.get(sec);
data.add(getAggregateSecond(raw));
it.remove();
cnt++;
}
return data;
}
private JSONObject getAggregateSecond(List<SampleResult> raw) {
/*
"rc": item.http_codes,
"net": item.net_codes
*/
JSONObject result = new JSONObject();
Date ts = new Date(raw.iterator().next().getEndTime());
result.put("ts", format.format(ts));
int threads = 0;
int avg_rt = 0;
Long[] rtimes = new Long[raw.size()];
String[] rcodes = new String[raw.size()];
int cnt = 0;
int failedCount = 0;
for (Iterator<SampleResult> it = raw.iterator(); it.hasNext();) {
SampleResult res = it.next();
threads += res.getAllThreads();
avg_rt += res.getTime();
rtimes[cnt] = Long.valueOf(res.getTime());
rcodes[cnt] = res.getResponseCode();
if (!res.isSuccessful()) {
failedCount++;
}
cnt++;
}
result.put("rps", cnt);
result.put("threads", threads / cnt);
result.put("avg_rt", avg_rt / cnt);
result.put("quantiles", getQuantilesJSON(rtimes));
result.put("net", getNetJSON(failedCount, cnt - failedCount));
result.put("rc", getRCJSON(rcodes));
result.put("planned_rps", 0); // JMeter has no such feature like Yandex.Tank
return result;
}
public static JSONObject getQuantilesJSON(Long[] rtimes) {
JSONObject result = new JSONObject();
Arrays.sort(rtimes);
double[] quantiles = {0.25, 0.50, 0.75, 0.80, 0.90, 0.95, 0.98, 0.99, 1.00};
Stack<Long> timings = new Stack();
timings.addAll(Arrays.asList(rtimes));
double level = 1.0;
long timing = 0;
for (int qn = quantiles.length - 1; qn >= 0; qn--) {
double quan = quantiles[qn];
while (level >= quan && !timings.empty()) {
timing = timings.pop();
level -= 1.0 / rtimes.length;
}
result.element(String.valueOf(quan * 100), timing);
}
return result;
}
private JSONObject getNetJSON(int failedCount, int succCount) {
JSONObject result = new JSONObject();
result.put("0", succCount);
result.put("1", failedCount);
return result;
}
private JSONObject getRCJSON(String[] rcodes) {
JSONObject result = new JSONObject();
for (int i = 0; i < rcodes.length; i++) {
int oldval = 0;
if (result.containsKey(rcodes[i])) {
oldval = (Integer) result.get(rcodes[i]);
}
result.put(rcodes[i], oldval + 1);
}
return result;
}
}