From 2c572e027cd209b56327a2ce3f052a05fc40dc20 Mon Sep 17 00:00:00 2001
From: hveldstra <h@veldstra.org>
Date: Mon, 11 Feb 2019 20:41:11 +0100
Subject: [PATCH] Init

---
 README.md |  8 +++++++
 index.js  | 64 +++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 72 insertions(+)
 create mode 100644 README.md
 create mode 100644 index.js

diff --git a/README.md b/README.md
new file mode 100644
index 0000000..441c714
--- /dev/null
+++ b/README.md
@@ -0,0 +1,8 @@
+
+# Purpose
+
+Use this plugin to get a per-endpoint breakdown of latency and response codes in your Artillery HTTP tests.
+
+# License
+
+MPL 2.0
diff --git a/index.js b/index.js
new file mode 100644
index 0000000..11a43a9
--- /dev/null
+++ b/index.js
@@ -0,0 +1,64 @@
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+const url = require('url');
+
+module.exports = { Plugin: MetricsByEndpoint };
+
+// NOTE: Will not work with `parallel` - need request UIDs for that
+function MetricsByEndpoint(script, events) {
+  if (!script.config.processor) {
+    script.config.processor = {};
+  }
+
+  script.config.processor.metricsByEndpoint_beforeRequest = metricsByEndpoint_beforeRequest;
+script.config.processor.metricsByEndpoint_afterResponse = metricsByEndpoint_afterResponse;
+
+  script.scenarios.forEach(function(scenario) {
+    scenario.beforeRequest = [].concat(scenario.beforeRequest || []);
+    scenario.beforeRequest.push('metricsByEndpoint_beforeRequest');
+    scenario.afterResponse = [].concat(scenario.afterResponse || []);
+    scenario.afterResponse.push('metricsByEndpoint_afterResponse');
+  });
+}
+
+function metricsByEndpoint_beforeRequest(req, userContext, events, done) {
+  userContext.vars._metricsByEndpointStartedAt = Date.now();
+  return done();
+}
+
+function metricsByEndpoint_afterResponse(req, res, userContext, events, done) {
+  let delta = 0;
+  // TODO: If hostname is not target, keep it.
+  const baseUrl = url.parse(req.url).path;
+
+  let histoName = req.name ?
+        `${baseUrl} (${req.name})` :
+        `${baseUrl}`;
+  let counterName = histoName;
+
+  if (res.headers['server-timing']) {
+    delta = getServerTimingTotal(res.headers['server-timing']);
+    histoName = `Server-Timing ${histoName}`;
+  } else {
+    delta = Date.now() - userContext.vars._metricsByEndpointStartedAt;
+  }
+
+  events.emit('counter', `code ${res.statusCode} on ${counterName}`, 1);
+  events.emit('histogram', histoName, delta);
+  return done();
+}
+
+function getServerTimingTotal(s) {
+  const matches = s.match(/total;dur=[0-9.]+/ig);
+  if(matches !== null && matches.length > 0) {
+    // we always grab the first instance of "total" if there's more than one
+    // TODO: Use Number and round to 2 digits
+    return parseInt(matches[0]
+                    .split('=')[1]);
+  } else {
+    // no match
+    return -1;
+  }
+}