node-ipc/dao/client.js

239 lines
6.8 KiB
JavaScript
Raw Normal View History

2016-01-10 23:18:14 +11:00
'use strict';
const net = require('net'),
tls = require('tls'),
2016-01-11 15:59:40 +11:00
eventParser = require('./eventParser.js'),
2016-01-10 23:18:14 +11:00
Pubsub = require('event-pubsub'),
Message = require('js-message'),
fs = require('fs'),
Queue = require('js-queue');
2014-02-22 20:13:31 +11:00
function init(config,log){
2016-01-10 23:18:14 +11:00
let client={
2014-02-22 20:13:31 +11:00
config : config,
queue : new Queue,
2014-02-22 20:13:31 +11:00
socket : false,
connect : connect,
emit : emit,
log : log,
retriesRemaining:config.maxRetries||0
2016-01-10 23:18:14 +11:00
};
2016-01-10 22:53:33 +11:00
new Pubsub(client);
2014-02-22 20:13:31 +11:00
return client;
}
function emit(type,data){
2016-01-12 09:25:48 +11:00
this.log('dispatching event to '.debug, (this.id).variable, this.path.variable,' : ', type.data,',', data);
2016-01-10 23:18:14 +11:00
let message=new Message;
message.type=type;
message.data=data;
2015-08-23 15:46:55 +10:00
if(this.config.rawBuffer){
message=new Buffer(type,this.encoding);
2015-08-23 15:46:55 +10:00
}else{
message=eventParser.format(message);
2015-08-23 15:46:55 +10:00
}
if(!this.config.sync){
this.socket.write(message);
return;
}
this.queue.add(
syncEmit.bind(this,message)
);
2016-01-10 23:18:14 +11:00
}
2014-02-22 20:13:31 +11:00
function syncEmit(message){
2016-01-12 09:25:48 +11:00
this.log('dispatching event to '.debug, (this.id).variable, this.path.variable,' : ', message.data);
this.socket.write(message);
}
2014-02-22 20:13:31 +11:00
function connect(){
//init client object for scope persistance especially inside of socket events.
2016-01-10 23:18:14 +11:00
let client=this;
2014-02-22 20:13:31 +11:00
client.log('requested connection to '.debug, client.id.variable, client.path.variable);
if(!this.path){
client.log('\n\n######\nerror: '.error, client.id .info,' client has not specified socket path it wishes to connect to.'.error);
return;
}
2014-02-26 12:15:43 +11:00
if(!client.port){
client.log('Connecting client on Unix Socket :'.debug, client.path.variable);
2016-01-10 23:18:14 +11:00
let path = client.path;
if (process.platform ==='win32' && !client.path.startsWith('\\\\.\\pipe\\')){
path = path.replace(/^\//, '');
path = path.replace(/\//g, '-');
2016-01-11 16:10:58 +11:00
path= `\\\\.\\pipe\\${path}`;
}
2014-02-26 12:15:43 +11:00
client.socket = net.connect(
{
path: path
2014-02-26 12:15:43 +11:00
}
);
}else{
if(!client.config.tls){
client.log('Connecting client via TCP to'.debug, client.path.variable ,client.port);
client.socket = net.connect(
{
port:client.port,
host:client.path
}
);
}else{
client.log('Connecting client via TLS to'.debug, client.path.variable ,client.port,client.config.tls);
if(client.config.tls.private){
client.config.tls.key=fs.readFileSync(client.config.tls.private);
2014-02-26 12:15:43 +11:00
}
if(client.config.tls.public){
client.config.tls.cert=fs.readFileSync(client.config.tls.public);
}
if(client.config.tls.trustedConnections){
if(typeof client.config.tls.trustedConnections === 'string'){
client.config.tls.trustedConnections=[client.config.tls.trustedConnections];
}
client.config.tls.ca=[];
2016-01-10 23:18:14 +11:00
for(let i=0; i<client.config.tls.trustedConnections.length; i++){
client.config.tls.ca.push(
fs.readFileSync(client.config.tls.trustedConnections[i])
);
}
}
client.config.tls.host=client.path;
client.config.tls.port=client.port;
client.socket = tls.connect(
client.config.tls
);
}
2014-02-26 12:15:43 +11:00
}
2014-02-22 20:13:31 +11:00
client.socket.setEncoding(this.config.encoding);
2014-02-22 20:13:31 +11:00
client.socket.on(
'error',
function(err){
client.log('\n\n######\nerror: '.error, err);
2015-12-01 12:43:43 +11:00
client.trigger('error', err);
2014-02-22 20:13:31 +11:00
}
);
2014-02-22 20:13:31 +11:00
client.socket.on(
'connect',
2016-01-10 23:18:14 +11:00
function connectionMade(){
2014-02-22 20:13:31 +11:00
client.trigger('connect');
2015-08-23 16:16:20 +10:00
client.retriesRemaining=client.config.maxRetries;
2015-12-01 12:43:43 +11:00
client.log('retrying reset');
2014-02-22 20:13:31 +11:00
}
);
2014-02-22 20:13:31 +11:00
client.socket.on(
'close',
2016-01-10 23:18:14 +11:00
function connectionClosed(){
2016-01-11 16:10:58 +11:00
client.log('connection closed'.notice ,client.id.variable , client.path.variable,
`${client.retriesRemaining} tries remaining of ${client.config.maxRetries}`
);
if(
2015-08-23 16:16:20 +10:00
client.config.stopRetrying || client.retriesRemaining<1
2015-08-23 15:46:55 +10:00
){
2015-12-03 05:05:47 +11:00
client.trigger('disconnect');
client.log(
2016-01-12 09:25:48 +11:00
(client.config.id).variable,
'exceeded connection rety amount of'.warn,
2016-01-10 23:18:14 +11:00
' or stopRetrying flag set.'
);
client.socket.destroy();
2015-12-03 05:05:47 +11:00
client.trigger('destroy');
client=undefined;
return;
}
client.isRetrying=true;
2014-02-22 20:13:31 +11:00
setTimeout(
2016-01-10 23:18:14 +11:00
function retryTimeout(){
client.retriesRemaining--;
client.isRetrying=false;
client.connect();
setTimeout(
function resetRetriesCheck(){
if(!client.isRetrying){
client.retriesRemaining=client.config.maxRetries;
}
},
100
);
}.bind(null,client),
2014-02-22 20:13:31 +11:00
client.config.retry
);
2014-02-22 20:13:31 +11:00
client.trigger('disconnect');
}
);
2014-02-22 20:13:31 +11:00
client.socket.on(
'data',
function(data) {
client.log('## recieved events ##'.rainbow);
2015-08-23 15:46:55 +10:00
if(client.config.rawBuffer){
client.trigger(
'data',
new Buffer(data,this.encoding)
);
if(!client.config.sync){
return;
}
client.queue.next();
2015-08-23 15:46:55 +10:00
return;
}
2016-01-10 23:18:14 +11:00
if(!this.ipcBuffer){
2014-03-04 07:02:56 +11:00
this.ipcBuffer='';
2016-01-10 23:18:14 +11:00
}
2014-03-04 07:02:56 +11:00
data=(this.ipcBuffer+=data);
if(data.slice(-1)!=eventParser.delimiter || data.indexOf(eventParser.delimiter) == -1){
2014-09-24 20:26:05 +10:00
client.log('Implementing larger buffer for this socket message. You may want to consider smaller messages'.notice);
2014-03-04 07:02:56 +11:00
return;
}
2014-03-04 07:02:56 +11:00
this.ipcBuffer='';
2016-01-10 23:18:14 +11:00
const events = eventParser.parse(data);
const eCount = events.length;
for(let i=0; i<eCount; i++){
let message=new Message;
message.load(events[i]);
2016-01-10 23:18:14 +11:00
client.log('detected event of type '.debug, message.type.data, message.data);
2014-02-22 20:13:31 +11:00
client.trigger(
message.type,
message.data
2014-02-22 20:13:31 +11:00
);
}
if(!client.config.sync){
return;
}
client.queue.next();
2014-02-22 20:13:31 +11:00
}
);
}
module.exports=init;