WRK scripts now work, reporting and the rest of the functionality

README.md

@@ -1,2 +1,5 @@
 # WaspsWithBazookas
 Its like bees with machine guns but way more power
+
+Postman REST Docs
+[Postman REST Docs](https://documenter.getpostman.com/view/7072151/S1TR4zsf)

hive/hive.js

@@ -1,82 +1,342 @@
 #!/usr/bin/env node
+
 const http = require('http');
 const fastify = require('fastify')();
 const request = require('request');
+var convert = require('convert-units');
+
+var running = false;
 var wasps = [];
+var waspDoneCount = 0;
+var waspsRunningCount = 0;
+var runTimeStamp = 0;
 var idCount = 0;
+var report = null;
 
-fastify.get('/wasp/checkin/:port', (req, res) => {
-
-var found = null;
-    for (var i = 0; i < wasps.length; i++) {
-      if(wasps[i].ip == req.ip && wasps[i].port == req.params.port)
-      {
-        found = i;
-        console.log(found)
-        break;
-      }
+fastify.get('/wasp/checkin/:port', (req, res) =>
+{
+  var found = null;
+  for(var i = 0; i < wasps.length; i++)
+  {
+    if(wasps[i].ip == req.ip && wasps[i].port == req.params.port)
+    {
+      found = i;
+      break;
     }
+  }
 
-    var wasp = {
-      ip: req.ip,
-      port: req.params.port,
-      id: 'wasp'+idCount++
-    }
-    if(found == null)
+  var wasp = {
+    ip: req.ip,
+    port: req.params.port,
+    id: 'wasp' + idCount++
+  }
+  if(found == null)
+  {
+    wasps.push(wasp);
+  }
+  else
+  {
+    wasps[found] = wasp;
+  }
+
+
+  res.code('200').send(
+  {
+    id: wasp.id
+  });
+
+  console.log(`Wasp ${idCount-1} checking in at ${wasp.ip}!`);
+  console.log(`Total Wasps: ${wasps.length}`)
+
+})
+
+fastify.get('/wasp/list', (req, res) =>
+{
+  res.code('200').send(wasps);
+})
+
+fastify.put('/wasp/reportin/:id', (req, res) =>
+{
+  var wasp = wasps.find(w =>
+  {
+    return w.id == req.params.id;
+  });
+
+  if(wasp)
+  {
+    waspDoneCount++;
+    report.wasp.reports.push(
+    {
+      wasp: wasp,
+      status: 'complete',
+      stats: req.body
+    })
+    report.status.completed += 1;
+    report.totalRPS += req.body.totalRPS;
+    report.read += req.body.read;
+    report.totalRequests += req.body.totalRequests;
+    report.tps += req.body.tps;
+    report.errors.connect += req.body.errors.connect || 0;
+    report.errors.read += req.body.errors.read || 0;
+    report.errors.write += req.body.errors.write || 0;
+    report.errors.timeout += req.body.errors.timeout || 0;
+
+    report.nonSuccessRequest += req.body.nonSuccessRequests;
+
+    res.send();
+  }
+  else
+  {
+    gError('/wasp/reportin/:id', res);
+  }
+  if(waspDoneCount == waspsRunningCount)
+  {
+    genReport();
+  }
+})
+
+fastify.put('/wasp/reportin/:id/failed', (req, res) =>
+{
+
+  var wasp = wasps.find(w =>
+  {
+    return w.id == req.params.id;
+  });
+
+  if(wasp)
+  {
+    waspDoneCount++;
+    report.wasp.reports.push(
     {
-      wasps.push(wasp);
+      wasp: wasp,
+      status: 'failed',
+      error: req.body
+    });
+    report.status.failed += 1;
+
+    res.send();
+  }
+  else
+  {
+    gError('/wasp/reportin/:id/failed', res);
+  }
+  if(waspDoneCount == waspsRunningCount)
+  {
+    genReport();
+  }
+
+})
+
+fastify.put('/hive/poke', (req, res) =>
+{
+  if(!isRunningRes(res))
+  {
+    if(!req.body.target)
+    {
+      res.code(400).send('need a target, cant shoot into the darkness...')
     }
     else
     {
-      wasps[found] = wasp;
-    }
-
+      req.body.t = req.body.t || 10;
+      req.body.c = req.body.c || 50;
+      req.body.d = req.body.d || 30;
 
-    res.code('200').send({id:wasp.id});
+      for(var i = 0; i < wasps.length; i++)
+      {
+        request(
+        {
+          method: 'PUT',
+          uri: `http://${wasps[i].ip}:${wasps[i].port}/fire`,
+          json: true,
+          body: req.body
+        })
+      }
 
-    console.log(`Wasp ${idCount-1} checking in at ${wasp.ip}!`);
-    console.log(`Total Wasps: ${wasps.length}`)
+      res.code(200).send('Angry wasp noises');
+      console.log('Sending command to fire!');
+      setRunning(true);
 
+      //shit went down if they don't all respond in duration + 5 seconds
+      setTimeout(() =>
+      {
+        if(running)
+        {
+          genReport();
+        }
+      }, (req.body.d + 5) * 1000);
+    }
+  }
 })
 
-fastify.get('/wasp/list', (req, res) => {
-    res.code('200').send(wasps);
+fastify.delete('/hive/torch', (req, res) =>
+{
+  res.code(200).send(`R.I.P All ${wasps.length} wasps. :'(`);
+  wasps = [];
+  console.log('f');
 })
 
-fastify.put('/wasp/reportin/:id', (req, res) => {
-  console.log(req.params.id,req.body)
-  res.send();
+fastify.get('/hive/status/done', (req, res) =>
+{
+  if(!isRunningRes(res))
+  {
+    res.code(200).send('done');
+  }
 })
 
-fastify.put('/hive/poke', (req, res) => {
-  for (var i = 0; i < wasps.length; i++) {
-      request({
-            method: 'PUT',
-            uri: `http://${wasps[i].ip}:${wasps[i].port}/fire`,
-            json: true,
-            body: req.body
-        })
+fastify.get('/hive/status/report', (req, res) =>
+{
+  if(!isRunningRes(res))
+  {
+    if(report)
+    {
+      res.code(200).send(report);
+    }
+    else
+    {
+      res.code(400).send('No report yet.');
+    }
+
   }
-  res.code(200).send('Angry wasp noises');
-  console.log('Sending command to fire!');
 })
 
-fastify.delete('/hive/torch', (req, res) => {
-    res.code(200).send(`R.I.P All ${wasps.length} wasps. :'('`);
-    wasps = [];
-    console.log('f');
+fastify.get('/hive/status/report/:val', (req, res) =>
+{
+  if(!isRunningRes(res))
+  {
+    if(report && report[req.params.val])
+    {
+      res.code(200).send(report[req.params.val]);
+    }
+    else
+    {
+      res.code(400).send('No hive information on that.');
+    }
+  }
 })
 
-fastify.get('/hive/status/done', (req, res) => {
 
+
+fastify.get('/hive/status', (req, res) =>
+{
+  if(!isRunningRes(res, 200))
+  {
+    res.code(200).send(`Hive is oprational with ${wasps.length} wasps ready and waiting orders.`);
+  }
 })
 
-fastify.get('/hive/status/report', (req, res) => {
 
-})
+var isRunningRes = function(res, code)
+{
+  if(running)
+  {
+    res.code(code || 425).send(((waspDoneCount / waspsRunningCount) * 100) + "% complete, eta " + ((Number(process.hrtime.bigint()) / 1000000) - runTimeStamp) + "ms to go.");
+    return true;
+  }
+  return false;
+}
 
-fastify.get('/hive/status', (req, res) => {
+var setRunning = function(run)
+{
+  if(run)
+  {
+    running = true;
+    runTimeStamp = Number(process.hrtime.bigint()) / 1000000;
+    report = {
+      wasp:
+      {
+        reports: []
+      },
+      status:
+      {
+        completed: 0,
+        failed: 0
+      },
+      latency:
+      {
+        avg: 0,
+        max: 0,
+      },
+      rps:
+      {
+        avg: 0,
+        max: 0,
+      },
 
-})
+      totalRPS: 0,
+      totalRequests: 0,
+      read: 0,
+      tps: 0,
+      nonSuccessRequests: 0,
+      errors:
+      {
+        connect: 0,
+        read: 0,
+        write: 0,
+        timeout: 0
+      },
+    };
+
+    waspsRunningCount = wasps.length;
+  }
+  else
+  {
+    runTimeStamp = 0;
+    waspDoneCount = 0;
+    waspsRunningCount = 0;
+    running = false;
+  }
+}
+
+var genReport = function()
+{
+  console.log(`Reports are in lets see how they are.`);
+  for(var i = 0; i < report.wasp.reports.length; i++)
+  {
+    var wasp = report.wasp.reports[i];
+    if(wasp.stats)
+    {
+      report.latency.avg += wasp.stats.latency.avg;
+      report.rps.avg += wasp.stats.rps.avg;
+
+      if(wasp.stats.latency.max > report.latency.max)
+      {
+        report.latency.max = wasp.stats.latency.max;
+      }
+      if(wasp.stats.rps.max > report.rps.max)
+      {
+        report.rps.max = wasp.stats.rps.max;
+      }
+    }
+  }
+  report.latency.avg = wasp.stats.latency.avg / report.status.completed;
+  report.rps.avg = wasp.stats.rps.avg / report.status.completed;
+  report.read = (
+  {
+    val,
+    unit
+  } = convert(report.read).from('B').toBest(),
+  {
+    val,
+    unit
+  });
+  report.tps = (
+  {
+    val,
+    unit
+  } = convert(report.tps).from('B').toBest(),
+  {
+    val,
+    unit
+  });
+
+  setRunning(false);
+}
+
+var gError = function(route, res)
+{
+  res.code(412).send(`I'm a little wasp`);
+  console.log(`Bad thingz happened in the ${route} sectorz.`);
+}
 
+console.log('Hive ready to release the wasps!')
 fastify.listen(process.argv[3] || process.env.WWB_HIVE_PORT || 4269)