ElasticsearchBackendListener: added label option: "verbose" with options

always or ifError or never - `always` send ResponseData always - `ifError` send ResponseData if some error in sampler - `never` never send ResponseData It's useful for optimalization of data sent to Elasticsearch
rdpanek · Sep 22, 2016 · 969901e · 969901e
1 parent 8ad5b63
commit 969901e
Show file tree

Hide file tree

Showing 4 changed files with 106 additions and 22 deletions.
diff --git a/.idea/workspace.xml b/.idea/workspace.xml
diff --git a/out/artifacts/ElasticSearchBackendListenerClient_jar/ElasticSearchBackendListenerClient.jar b/out/artifacts/ElasticSearchBackendListenerClient_jar/ElasticSearchBackendListenerClient.jar
diff --git a/...ackendListenerClient/cz/rdp/jmeter/elasticsearch/ElasticSearchBackendListenerClient.class b/...ackendListenerClient/cz/rdp/jmeter/elasticsearch/ElasticSearchBackendListenerClient.class
diff --git a/src/cz/rdp/jmeter/elasticsearch/ElasticSearchBackendListenerClient.java b/src/cz/rdp/jmeter/elasticsearch/ElasticSearchBackendListenerClient.java
@@ -35,19 +35,20 @@ public void handleSampleResults(List<SampleResult> results,
                                     BackendListenerContext context) {
         String indexNameToUse = indexName;
         for(SampleResult result : results) {
-            Map<String,Object> jsonObject = getMap(result);
+            Map<String,Object> jsonObject = getMap(result, context);
             if(dateTimeAppendFormat != null) {
                 SimpleDateFormat sdf = new SimpleDateFormat(dateTimeAppendFormat);
                 indexNameToUse = indexName + sdf.format(jsonObject.get(TIMESTAMP));
             }
             jsonObject.put("testPlanName", context.getParameter("testPlanName"));
             jsonObject.put("release", context.getParameter("release"));
+            jsonObject.put("verbose", context.getParameter("verbose"));
             client.prepareIndex(indexNameToUse, sampleType).setSource(jsonObject).execute().actionGet();
         }
 
     }
 
-    private Map<String, Object> getMap(SampleResult result) {
+    private Map<String, Object> getMap(SampleResult result, BackendListenerContext context) {
         Map<String, Object> map = new HashMap<String, Object>();
         String[] sampleLabels = result.getSampleLabel().split(VAR_DELIMITER);
         map.put("SampleLabel", sampleLabels[0]);
@@ -61,7 +62,15 @@ private Map<String, Object> getMap(SampleResult result) {
         map.put("RequestHeaders", result.getRequestHeaders());
         map.put("ResponseCode", result.getResponseCode());
         map.put("ResponseMessage", result.getResponseMessage());
-        map.put("ResponseData", result.getResponseDataAsString());
+        if (context.getParameter("verbose").equals("always")) {
+            map.put("ResponseData", result.getResponseDataAsString());
+        } else if (  context.getParameter("verbose").equals("ifError") && (result.getErrorCount() > 0) ) {
+            map.put("ResponseData", result.getResponseDataAsString());
+        } else if (  context.getParameter("verbose").equals("never") ) {
+            map.put("ResponseData", "");
+        } else {
+            map.put("ResponseData", "");
+        }
         map.put("SubResults", result.getSubResults());
         map.put("DataEncoding", result.getDataEncodingWithDefault());
         map.put("ThreadName", result.getThreadName());
@@ -73,7 +82,7 @@ private Map<String, Object> getMap(SampleResult result) {
         map.put("Latency", result.getLatency());
         map.put("ConnectTime", result.getConnectTime());
         map.put("SampleCount", result.getSampleCount());
-        //map.put("SamplerData", result.getSamplerData());
+        map.put("SamplerData", result.getSamplerData());
         map.put("ErrorCount", result.getErrorCount());
         map.put("Bytes", result.getBytes());
         map.put("BodySize", result.getBodySize());
@@ -148,6 +157,7 @@ public Arguments getDefaultParameters() {
         arguments.addArgument("runId", "${__UUID()}");
         arguments.addArgument("release", "");
         arguments.addArgument("testPlanName", "");
+        arguments.addArgument("verbose", "always|ifError|never");
         return arguments;