add groups feature

lib/units/groups-engine/index.js

@@ -0,0 +1,115 @@
+/**
+* Copyright © 2019 code initially contributed by Orange SA, authors: Denis Barbaron - Licensed under the Apache license 2.0
+**/
+
+const events = require('events')
+const Promise = require('bluebird')
+const logger = require('../../util/logger')
+const zmqutil = require('../../util/zmqutil')
+const srv = require('../../util/srv')
+const lifecycle = require('../../util/lifecycle')
+const wireutil = require('../../wire/util')
+
+const groupsScheduler = require('./scheduler')
+const groupsWatcher = require('./watchers/groups')
+const devicesWatcher = require('./watchers/devices')
+const usersWatcher = require('./watchers/users')
+
+module.exports = function(options) {
+  const log = logger.createLogger('groups-engine')
+  const channelRouter = new events.EventEmitter()
+
+  const push = zmqutil.socket('push')
+  Promise.map(options.endpoints.push, function(endpoint) {
+    return srv.resolve(endpoint).then(function(records) {
+      return srv.attempt(records, function(record) {
+        log.info('Sending output to "%s"', record.url)
+        push.connect(record.url)
+        return Promise.resolve(true)
+      })
+    })
+  })
+  .catch(function(err) {
+    log.fatal('Unable to connect to push endpoint', err)
+    lifecycle.fatal()
+  })
+
+  // Input
+  const sub = zmqutil.socket('sub')
+  Promise.map(options.endpoints.sub, function(endpoint) {
+    return srv.resolve(endpoint).then(function(records) {
+      return srv.attempt(records, function(record) {
+        log.info('Receiving input from "%s"', record.url)
+        sub.connect(record.url)
+        return Promise.resolve(true)
+      })
+    })
+  })
+  .catch(function(err) {
+    log.fatal('Unable to connect to sub endpoint', err)
+    lifecycle.fatal()
+  })
+
+  const pushdev = zmqutil.socket('push')
+  Promise.map(options.endpoints.pushdev, function(endpoint) {
+    return srv.resolve(endpoint).then(function(records) {
+      return srv.attempt(records, function(record) {
+        log.info('Sending output to "%s"', record.url)
+        pushdev.connect(record.url)
+        return Promise.resolve(true)
+      })
+    })
+  })
+  .catch(function(err) {
+    log.fatal('Unable to connect to pushdev endpoint', err)
+    lifecycle.fatal()
+  })
+
+  const subdev = zmqutil.socket('sub')
+  Promise.map(options.endpoints.subdev, function(endpoint) {
+    return srv.resolve(endpoint).then(function(records) {
+      return srv.attempt(records, function(record) {
+        log.info('Receiving input from "%s"', record.url)
+        subdev.connect(record.url)
+        return Promise.resolve(true)
+      })
+    })
+  })
+  .catch(function(err) {
+    log.fatal('Unable to connect to subdev endpoint', err)
+    lifecycle.fatal()
+  })
+
+  // Establish always-on channels
+  ;[wireutil.global].forEach(function(channel) {
+    log.info('Subscribing to permanent channel "%s"', channel)
+    sub.subscribe(channel)
+    subdev.subscribe(channel)
+  })
+
+  sub.on('message', function(channel, data) {
+    channelRouter.emit(channel.toString(), channel, data)
+  })
+
+  subdev.on('message', function(channel, data) {
+    channelRouter.emit(channel.toString(), channel, data)
+  })
+
+  groupsScheduler()
+  groupsWatcher(push, pushdev, channelRouter)
+  devicesWatcher(push, pushdev, channelRouter)
+  usersWatcher(pushdev)
+
+  lifecycle.observe(function() {
+    [push, sub, pushdev, subdev].forEach(function(sock) {
+      try {
+        sock.close()
+      }
+      catch (err) {
+        // No-op
+      }
+    })
+  })
+
+  log.info('Groups engine started')
+}