Merge pull request #1 from jamesbarnett91/websocketsv0.2

Switch from http polling to websockets

1 files changed, 43 insertions, 0 deletions

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