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();