From 24b4a39fce61dd9bd8ab7757f3bbda0636adc8c9 Mon Sep 17 00:00:00 2001
From: James Barnett <noreply@jamesbarnett.xyz>
Date: Sat, 7 Apr 2018 18:49:48 +0100
Subject: Switch from http polling to data push via websockets

---
 services/data-broadcaster.js | 42 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)
 create mode 100644 services/data-broadcaster.js

(limited to 'services/data-broadcaster.js')

diff --git a/services/data-broadcaster.js b/services/data-broadcaster.js
new file mode 100644
index 0000000..a49f4a7
--- /dev/null
+++ b/services/data-broadcaster.js
@@ -0,0 +1,42 @@
+const app = require('../app');
+
+function broadcastRealtimeUsageUpdate(data) {
+  broadcast(generatePayload('realtimeUsage', data));
+}
+
+function broadcastDailyUsageUpdate(data) {
+  broadcast(generatePayload('dailyUsage', data));
+}
+
+function broadcastMonthlyUsageUpdate(data) {
+  broadcast(generatePayload('monthlyUsage', data));
+}
+
+function broadcastPowerStateUpdate(data) {
+  broadcast(generatePayload('powersState', data));
+}
+
+function broadcast(payload) {
+  app.getWsClients().forEach(client => {
+    client.send(payload);
+  })
+}
+
+function generatePayload(dataType, data) {
+
+  let payload = {
+    dataType: dataType,
+    data: data
+  }
+
+  return JSON.stringify(payload);
+}
+
+
+module.exports = {
+  broadcastRealtimeUsageUpdate: broadcastRealtimeUsageUpdate,
+  broadcastDailyUsageUpdate: broadcastDailyUsageUpdate,
+  broadcastMonthlyUsageUpdate: broadcastMonthlyUsageUpdate,
+  broadcastPowerStateUpdate: broadcastPowerStateUpdate,
+  generatePayload: generatePayload
+}
\ No newline at end of file
-- 
cgit v1.2.3