mist/modules/socketManager.js

const _ = require('./utils/underscore.js');
const Q = require('bluebird');
const log = require('./utils/logger').create('Sockets');

const Web3IpcSocket = require('./sockets/web3Ipc');
const Web3HttpSocket = require('./sockets/web3Http');

/**
 * `Socket` manager.
 */
class SocketManager {
  constructor() {
    this._sockets = {};
  }

  /**
   * Get socket with given id, creating it if it does not exist.
   *
   * @return {Socket}
   */
  create(id, type) {
    log.debug(`Create socket, id=${id}, type=${type}`);

    switch (type) {
      case 'ipc':
        this._sockets[id] = new Web3IpcSocket(this, id);
        break;
      case 'http':
        this._sockets[id] = new Web3HttpSocket(this, id);
        break;
      default:
        throw new Error(`Unrecognized socket type: ${type}`);
    }

    return this._sockets[id];
  }

  /**
   * Get socket with given id, creating it if it does not exist.
   *
   * @return {Socket}
   */
  get(id, type) {
    if (!this._sockets[id]) {
      this.create(id, type);
    }

    return this._sockets[id];
  }

  /**
   * @return {Promise}
   */
  destroyAll() {
    log.info('Destroy all sockets');

    return Q.all(
      _.map(this._sockets, (s, id) => {
        this.remove(id);
        return s.destroy();
      })
    );
  }

  /**
   * Remove socket with given id from this manager.
   *
   * Usually called by `Socket` instances when they're destroyed.
   */
  remove(id) {
    log.debug(`Remove socket, id=${id}`);

    delete this._sockets[id];
  }
}

module.exports = new SocketManager();