From cc6ff973b5b58e80106fd1f2ee22e57d047664dd Mon Sep 17 00:00:00 2001
From: Danielv123 <tranen.bananen@gmail.com>
Date: Tue, 7 Mar 2017 11:19:40 +0100
Subject: [PATCH] Production graph DB for master

#35
---
 client.js |  5 +++--
 master.js | 50 ++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 53 insertions(+), 2 deletions(-)

diff --git a/client.js b/client.js
index 55b35074..6f159a65 100644
--- a/client.js
+++ b/client.js
@@ -437,6 +437,7 @@ function instanceManagement() {
 	setInterval(function(){
 		fs.readFile(instancedirectory + "/script-output/flows.txt", {encoding: "utf8"}, function(err, data) {
 			if(!err && data) {
+				let timestamp = Date.now();
 				data = data.split("\n");
 				let flowStats = [];
 				for(let i = 0; i < data.length; i++) {
@@ -466,9 +467,9 @@ function instanceManagement() {
 						//console.log(payload[key])
 					}
 					console.log("Recorded flows, copper plate since last time: " + payload["copper-plate"]);
-					/*needle.post(config.masterIP + ":" + config.masterPort + '/uploadMod', payload, function (err, response, body) {
+					needle.post(config.masterIP + ":" + config.masterPort + '/logStats', {timestamp: timestamp, slaveID: instanceconfig.unique,data: payload}, function (err, response, body) {
 						// we did it, keep going
-					}*/
+					})
 				}
 				oldFlowStats = totalFlows
 				fs.writeFileSync(instancedirectory + "/script-output/flows.txt", "");
diff --git a/master.js b/master.js
index 53051391..112abdd9 100644
--- a/master.js
+++ b/master.js
@@ -29,9 +29,17 @@ app.use(express.static(masterModFolder));
 // set up database
 var Datastore = require('nedb');
 db = {};
+
+// database for items in system
 db.items = new Datastore({ filename: 'database/items.db', autoload: true });
+
+// in memory database for combinator signals
 db.signals = new Datastore({ filename: 'database/signals.db', autoload: true, inMemoryOnly: true});
 db.signals.ensureIndex({ fieldName: 'time', expireAfterSeconds: 3600 }, function (err) {});
+
+// production chart database
+db.flows = new Datastore({ filename: "database/flows.db", autoload: true})
+db.flows.ensureIndex({ fieldName: "slaveID", expireAfterSeconds: 2592000}); // expire after 30 days
 // db.slaves = new Datastore({ filename: 'database/slaves.db', autoload: true, inMemoryOnly: false});
 
 db.items.additem = function(object) {
@@ -180,6 +188,48 @@ app.get("/inventory", function(req, res) {
 		res.send(docs);
 	});
 });
+
+// post flowstats here for production graphs
+// {timestamp: Date, slaveID: string, data: {"item":number}}
+app.post("/logStats", function(req,res) {
+	if(typeof req.body == "object" && req.body.slaveID && req.body.timestamp && req.body.data) {
+		db.flows.insert({
+			slaveID: req.body.slaveID,
+			timestamp: req.body.timestamp,
+			data: req.body.data,
+		});
+		console.log("inserted: " + req.body.slaveID + " | " + req.body.timestamp)
+	} else {
+		res.send("failure")
+	}
+})
+// {slaveID: string, fromTime: Date, toTime, Date}
+app.post("/getStats", function(req,res) {
+	if(typeof req.body == "object" && req.body.slaveID) {
+		// if not specified, get stats for last 24 hours
+		if(!req.body.fromTime) {
+			req.body.fromTime = Date.now() - 86400000 // 24 hours in MS
+		}
+		if(!req.body.toTime) {
+			req.body.toTime = Date.now();
+		}
+		db.flows.find({
+			$and: [
+				{
+					slaveID: req.body.slaveID,
+				}, {
+					timestamp: {
+						$gte: req.body.toTime,
+						$lt: req.body.fromTime,
+					}
+				}
+			]
+		}, function(err, docs) {
+			
+		})
+	}
+})
+
 // endpoint for getting the chartjs library
 app.get("/chart.js", function(req, res) {
 	res.header("Access-Control-Allow-Origin", "*");