noflo-runtime-base/src/DirectRuntime.js

Base library for building NoFlo runtimes

/* eslint class-methods-use-this: ["error", { "exceptMethods": ["send", "sendAll"] }] */
const Base = require('./Base');

class DirectRuntime extends Base {
  constructor(options) {
    super(options);
    this.clients = [];
  }

  _connect(client) {
    this.clients.push(client);
    client.on('send', (msg) => {
      // Capture context
      this._receive(msg, { client });
    });
  }

  _disconnect(client) {
    if (this.clients.indexOf(client) === -1) { return; }
    this.clients.splice(this.clients.indexOf(client), 1);
    client.removeAllListeners('send'); // XXX: a bit heavy
  }

  _receive(msg, context) {
    // Forward to Base
    this.receive(msg.protocol, msg.command, msg.payload, context);
  }

  send(protocol, topic, payload, context) {
    if (!context || !context.client) { return; }
    const m = {
      protocol,
      command: topic,
      payload,
    };
    context.client._receive(m);
  }

  sendAll(protocol, topic, payload) {
    const m = {
      protocol,
      command: topic,
      payload,
    };
    this.clients.forEach((client) => {
      client._receive(m);
    });
  }
}

module.exports = DirectRuntime;