refactored connected and added auto reconnect ability like the unix/tcp socket including authentication and per consumer ping
added in consumer pinghttp-server
parent
d3694887eb
commit
9b62c8e410
|
@ -1,27 +1,83 @@
|
||||||
import Socket from '../src'
|
import Socket from '../src'
|
||||||
|
|
||||||
async function packetProcess(packet) {
|
const ANONYMOUS = true
|
||||||
// Here one registers a packet processor which gets an plain object hash from the client
|
const PORT = 8090
|
||||||
// should return a promise
|
|
||||||
// for example if client sends a property cmd on can take action on that
|
function packetProcess (packet) {
|
||||||
return new Promise(resolve => {
|
|
||||||
let res = {}
|
const cmd = {
|
||||||
switch (packet.cmd) {
|
ack: function(packet){
|
||||||
case 'echo':
|
return new Promise( async (resolve) => {
|
||||||
res.msg = 'Echoing Back any payload propery'
|
packet.ack = true
|
||||||
res.payload = packet.payload
|
packet.sender= packet.sender || (packet._header ? packet._header.sender.name : 'unknown')
|
||||||
this.push(res)
|
packet.msg=`this is acknlowdgement that ${packet.sender} ack was received`
|
||||||
break
|
this.push(packet) // push to all active socket servers
|
||||||
default:
|
return resolve(packet) // this goes back to sender
|
||||||
res.msg = `.cmd, command ${packet.cmd} was unknown at server`
|
|
||||||
res.sent_packet = packet
|
|
||||||
}
|
|
||||||
resolve(res)
|
|
||||||
})
|
})
|
||||||
|
},
|
||||||
|
switch:{
|
||||||
|
status: function(packet){
|
||||||
|
return new Promise( async (resolve) => {
|
||||||
|
packet.cmd='switch/status'
|
||||||
|
packet.state=this.switches[packet.id-1]
|
||||||
|
packet.sender= packet.sender || (packet._header ? packet._header.sender.name : 'unknown')
|
||||||
|
this.push(packet) // push to all active socket servers
|
||||||
|
let res = { response:'status pushed on to all clients'}
|
||||||
|
return resolve(res) // this goes back to sender
|
||||||
|
})
|
||||||
|
},
|
||||||
|
on: function(packet){
|
||||||
|
return new Promise( async (resolve) => {
|
||||||
|
packet.cmd='switch/status'
|
||||||
|
packet.state='on'
|
||||||
|
this.switches[packet.id-1] = 'on'
|
||||||
|
packet.sender= packet.sender || (packet._header ? packet._header.sender.name : 'unknown')
|
||||||
|
this.push(packet) // push to all active socket servers
|
||||||
|
let res = { response:'status change - pushed on to all clients', id:packet.id}
|
||||||
|
return resolve(res) // this goes back to sender
|
||||||
|
})
|
||||||
|
},
|
||||||
|
off: function(packet){
|
||||||
|
return new Promise( async (resolve) => {
|
||||||
|
packet.cmd='switch/status'
|
||||||
|
packet.state='off'
|
||||||
|
this.switches[packet.id-1] = 'off'
|
||||||
|
packet.sender= packet.sender || (packet._header ? packet._header.sender.name : 'unknown')
|
||||||
|
this.push(packet) // push to all active socket servers
|
||||||
|
let res = { response:'status change - pushed off to all clients'}
|
||||||
|
return resolve(res) // this goes back to sender
|
||||||
|
})
|
||||||
|
},
|
||||||
|
toggle: function(packet){
|
||||||
|
return new Promise( async (resolve) => {
|
||||||
|
this.switches[packet.id-1] = this.switches[packet.id-1]==='on' ? 'off' : 'on'
|
||||||
|
packet.state = this.switches[packet.id-1]
|
||||||
|
packet.cmd ='switch/status'
|
||||||
|
packet.sender= packet.sender || (packet._header ? packet._header.sender.name : 'unknown')
|
||||||
|
this.push(packet) // push to all active socket servers
|
||||||
|
let res = { response:`status change - pushed toggled state of ${packet.state} to all clients`}
|
||||||
|
return resolve(res) // this goes back to sender
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
console.log('this in packet processor',packet.cmd)
|
||||||
|
let cmdProps = packet.cmd.split('/')
|
||||||
|
let func = cmd[cmdProps[0]]
|
||||||
|
console.log(func)
|
||||||
|
if (cmdProps[1]) func = func[cmdProps[1]]
|
||||||
|
console.log(func)
|
||||||
|
if (typeof func === 'function') return func.call(this,packet)
|
||||||
|
else {
|
||||||
|
packet.error='no function for command'
|
||||||
|
return packet
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// let test = new Test()
|
// let test = new Test()
|
||||||
let test = new Socket({ port: 8090, clientTracking: true })
|
let test = new Socket({ port: PORT, allowAnonymous:ANONYMOUS })
|
||||||
|
test.switches = ['off','off','off','off'] // database to hold switch states
|
||||||
test.registerPacketProcessor(packetProcess)
|
test.registerPacketProcessor(packetProcess)
|
||||||
;(async () => {
|
;(async () => {
|
||||||
console.log(await test.create())
|
console.log(await test.create())
|
||||||
|
|
|
@ -4,8 +4,9 @@
|
||||||
"description": "JSON packet host websocket server",
|
"description": "JSON packet host websocket server",
|
||||||
"main": "src",
|
"main": "src",
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"example": "node -r esm examples/server",
|
"server": "node -r esm examples/server",
|
||||||
"dev": "UCI_ENV=dev UCI_LOG_LEVEL=debug ./node_modules/.bin/nodemon -r esm examples/server"
|
"server:dev": "UCI_ENV=dev ./node_modules/.bin/nodemon -r esm examples/server",
|
||||||
|
"server:debug": "UCI_LOG_LEVEL=debug npm run server:dev"
|
||||||
},
|
},
|
||||||
"author": "David Kebler",
|
"author": "David Kebler",
|
||||||
"license": "MIT",
|
"license": "MIT",
|
||||||
|
|
311
src/socket.js
311
src/socket.js
|
@ -1,6 +1,6 @@
|
||||||
import WebSocket from 'ws'
|
import { Server as WSS } from 'ws'
|
||||||
|
import { Server } from 'http'
|
||||||
import btc from 'better-try-catch'
|
import btc from 'better-try-catch'
|
||||||
import to from 'await-to-js'
|
|
||||||
import { promisify } from 'util'
|
import { promisify } from 'util'
|
||||||
import _ON_DEATH from 'death' //this is intentionally ugly
|
import _ON_DEATH from 'death' //this is intentionally ugly
|
||||||
import clone from 'clone'
|
import clone from 'clone'
|
||||||
|
@ -8,23 +8,41 @@ import clone from 'clone'
|
||||||
import logger from '@uci-utils/logger'
|
import logger from '@uci-utils/logger'
|
||||||
let log = {}
|
let log = {}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Socket - a websocket that supports uci packets
|
* Socket - Description
|
||||||
* @class WebSocket
|
* @extends Server
|
||||||
* @extends ws.server
|
*/
|
||||||
*
|
class Socket extends Server {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* constructor - Description
|
||||||
|
*
|
||||||
|
* @param {object} [opts={}] Description
|
||||||
|
*
|
||||||
|
* @returns {type} Description
|
||||||
*/
|
*/
|
||||||
class Socket extends WebSocket.Server {
|
|
||||||
constructor(opts = {}) {
|
constructor(opts = {}) {
|
||||||
|
super(opts)
|
||||||
opts.host = opts.host || '0.0.0.0'
|
opts.host = opts.host || '0.0.0.0'
|
||||||
opts.port = opts.port || 8090
|
opts.port = opts.port || 8090
|
||||||
opts.clientTracking = opts.clientTracking || true
|
|
||||||
super(opts)
|
|
||||||
this.id = opts.id || opts.name || 'Websocket:' + new Date().getTime()
|
this.id = opts.id || opts.name || 'Websocket:' + new Date().getTime()
|
||||||
this.opts = opts // for use to recover from selected errors
|
this.opts = opts // for use to recover from selected errors
|
||||||
//self bindings
|
this.wss = {} // web socket instance goes here
|
||||||
this._listen = this._listen.bind(this)
|
this.allowAnonymous = (!opts.tokens || !!process.env.UCI_ANON || opts.allowAnonymous) ? true : false
|
||||||
|
this.tokens = opts.tokens || []
|
||||||
|
this.pingInterval = opts.pingInterval === false ? opts.pingInterval : (opts.pingInterval * 1000 || 5000)
|
||||||
|
this.nextconsumerID = 0 // incrementer for default initial consumer ID
|
||||||
|
this.consumers = new Map()
|
||||||
|
this.errors = []
|
||||||
|
this.errorCount = 0
|
||||||
this.create = this.create.bind(this)
|
this.create = this.create.bind(this)
|
||||||
|
this._destroy = this._destroy.bind(this)
|
||||||
|
this.authenticateconsumer = this.authenticateconsumer.bind(this)
|
||||||
|
this._authenticate = this._authenticate.bind(this)
|
||||||
this.close = promisify(this.close).bind(this)
|
this.close = promisify(this.close).bind(this)
|
||||||
log = logger({
|
log = logger({
|
||||||
file: 'src/socket.js',
|
file: 'src/socket.js',
|
||||||
|
@ -34,76 +52,227 @@ class Socket extends WebSocket.Server {
|
||||||
})
|
})
|
||||||
} // end constructor
|
} // end constructor
|
||||||
|
|
||||||
|
/**
|
||||||
|
* create - Description
|
||||||
|
*
|
||||||
|
* @returns {type} Description
|
||||||
|
*/
|
||||||
async create() {
|
async create() {
|
||||||
|
// return Promise.resolve('what')
|
||||||
return new Promise((resolve, reject) => {
|
return new Promise((resolve, reject) => {
|
||||||
_ON_DEATH(async () => {
|
_ON_DEATH(async () => {
|
||||||
log.error({method:'create', line:39, msg:'\nhe\'s dead jim'})
|
log.error({method:'create', line:51, msg:'\nhe\'s dead jim'})
|
||||||
await this._destroy()
|
await this._destroy()
|
||||||
})
|
})
|
||||||
|
|
||||||
|
// kills nodemon properly
|
||||||
process.once('SIGUSR2', async () => {
|
process.once('SIGUSR2', async () => {
|
||||||
await this._destroy
|
await this._destroy()
|
||||||
process.kill(process.pid, 'SIGUSR2')
|
process.kill(process.pid, 'SIGUSR2')
|
||||||
})
|
})
|
||||||
|
|
||||||
this.on('error', async err => {
|
this.once('error', async err => {
|
||||||
log.error({method:'creaete', line:48, err:err, msg:'socket server error'})
|
log.error({method:'creaete', line:48, err:err, msg:'http server error'})
|
||||||
reject(err)
|
reject(err)
|
||||||
})
|
})
|
||||||
|
|
||||||
this.on('listening', async () => {
|
this.once('listening', () => {
|
||||||
this._listen()
|
|
||||||
log.info({method:'create', line:54, msg:'websocket server created and listening at', address:this.address()})
|
log.info({method:'create', line:54, msg:'websocket server created and listening at', address:this.address()})
|
||||||
resolve(
|
// emit ready
|
||||||
`websocket ready and listening at ${this.address().address}:${
|
this.on('error', err => {
|
||||||
this.address().port
|
this.errorCount +=1 // log errors here
|
||||||
}`
|
this.errors.push(err)
|
||||||
)
|
if(this.errorCount>2) this.emit('warn', {msg:'something bad maybe going on, 3 errors', errors:this.errors})
|
||||||
|
if(this.errorCount>5) this.emit('fatal', {msg:'something fatal is going on, 6 errors', errors:this.errors})
|
||||||
})
|
})
|
||||||
|
this.wss = new WSS({server:this})
|
||||||
|
this.wss.on('error', err => {this.emit('error', err)}) // bubble up errors
|
||||||
|
this.wss.on('connection', this._connectionHandler.bind(this))
|
||||||
|
resolve(`websocket ready and listening at ${this.address().address}:${this.address().port}`)
|
||||||
|
})
|
||||||
|
super.listen(this.opts)
|
||||||
})
|
})
|
||||||
} // end create
|
} // end create
|
||||||
|
|
||||||
registerPacketProcessor(func) {
|
registerPacketProcessor(func) {
|
||||||
this._packetProcess = func
|
this._packetProcess = func.bind(this)
|
||||||
}
|
}
|
||||||
|
|
||||||
_listen() {
|
addTokens(tokens) {
|
||||||
this.on('connection', async (socket, req) => {
|
if (typeof tokens ==='string'){
|
||||||
let send = this._send.bind(socket)
|
tokens = tokens.split(',')
|
||||||
log.debug({method:'_listen', line:71, req: req, msg: 'new consumer connecting'})
|
}
|
||||||
socket.address = req.remoteAddress
|
this.tokens = this.tokens.concat(tokens)
|
||||||
|
if (this.tokens.length>0) this.allowAnonymous = false
|
||||||
|
}
|
||||||
|
|
||||||
socket.on('error', (err) => {
|
removeTokens(tokens) {
|
||||||
log.error({msg:'client connection error during listen',error:err})
|
if (typeof tokens ==='string'){
|
||||||
|
if (tokens === 'all') {
|
||||||
|
this.tokens = []
|
||||||
|
this.allowAnonymous = true
|
||||||
|
return
|
||||||
|
}
|
||||||
|
tokens = tokens.split(',')
|
||||||
|
}
|
||||||
|
this.tokens = this.tokens.filter(token => !tokens.includes(token))
|
||||||
|
if (this.tokens.length===0) {
|
||||||
|
log.warn({msg:'all tokens have been removed, switching to allow anonymous connections'})
|
||||||
|
this.allowAnonymous = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
registerTokenValidator (func) {
|
||||||
|
this.allowAnonymous = false
|
||||||
|
this._validateToken = func
|
||||||
|
}
|
||||||
|
|
||||||
|
registerAuthenticator (func) {
|
||||||
|
this.allowAnonymous = false
|
||||||
|
this._authenticate = func
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* push
|
||||||
|
*
|
||||||
|
* @param {Object} packet - this is the parameter packet
|
||||||
|
* @param {<type>} id - this is the parameter id
|
||||||
|
*
|
||||||
|
*/
|
||||||
|
async push(packet={},id) {
|
||||||
|
packet._header = {id: id || 'pushed'}
|
||||||
|
if (this.consumers.size > 0) {
|
||||||
|
log.debug({method:'push', line:142, id:packet._header.id, opts: this.opts, packet: packet, msg:'pushing a packet to all connected consumers'})
|
||||||
|
this.consumers.forEach(async consumer => {
|
||||||
|
this._seend(consumer,packet)
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
log.debug({method:'push', line:165, id:packet._header.id, opts: this.opts, packet: packet, msg:'no connected consumers, push ignored'})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
removeconsumer (id) {
|
||||||
|
let consumer = this.consumers.get(id)
|
||||||
|
clearInterval(consumer._ping)
|
||||||
|
consumer.removeAllListeners()
|
||||||
|
log.warn({msg:`consumer ${id}:${consumer.name} removed from server tracking`, id:id, name:consumer.name, curconsumerCount:this.consumers.size})
|
||||||
|
this.consumers.delete(id)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
async authenticateconsumer(consumer) {
|
||||||
|
// let server = this
|
||||||
|
return new Promise(async function(resolve, reject) {
|
||||||
|
// when consumer gets the handshake they must follow with authentication
|
||||||
|
consumer.on('message', authenticate.bind(this,consumer))
|
||||||
|
let [err] = await btc(this._send)(consumer,{_handshake: true, id:consumer.id})
|
||||||
|
if (err) {
|
||||||
|
log.error({msg:'error in handshake send', error:err})
|
||||||
|
reject(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
async function authenticate (consumer,message) {
|
||||||
|
let [err,packet] = btc(JSON.parse)(message)
|
||||||
|
if (err) reject('unable to parse authentication packet from consumer')
|
||||||
|
log.debug({msg:`authentication packet from consumer ${consumer.id}`, packet:packet})
|
||||||
|
if (!packet._authenticate) reject('first consumer packet was not authentication')
|
||||||
|
else {
|
||||||
|
let [err, res] = await btc(this._authenticate)(packet)
|
||||||
|
consumer.authenticated = this.allowAnonymous ? 'anonymous' : (err ? false : res)
|
||||||
|
consumer.name = packet.consumerName
|
||||||
|
packet.authenticated = consumer.authenticated
|
||||||
|
if (err && !this.allowAnonymous) packet.reason = err
|
||||||
|
log.debug({msg:'sending authorization result to consumer', packet:packet})
|
||||||
|
await this._send(consumer,packet) // send either way
|
||||||
|
if (packet.reason) {
|
||||||
|
log.info({msg:'consumer authentication failed', consumer:consumer.name, consumer_id:consumer.id, reason:err})
|
||||||
|
reject(packet.reason)
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
log.info({msg:'consumer authenticated successfuly', consumer:consumer.name, consumer_id:consumer.id})
|
||||||
|
if (this.allowAnonymous) log.warn({msg:'consumer connected anonymously', consumer:consumer.name, consumer_id:consumer.id})
|
||||||
|
resolve(consumer.authenticated)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}.bind(this))
|
||||||
|
}
|
||||||
|
|
||||||
|
// private methods
|
||||||
|
|
||||||
|
|
||||||
|
// default validator
|
||||||
|
_validateToken (token) {
|
||||||
|
if (token) return this.tokens.includes(token)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// default authenticator - reject value should be reason which is returned to consumer
|
||||||
|
async _authenticate (packet) {
|
||||||
|
if (!this._validateToken(packet.token)) return Promise.reject('invalid token')
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
async _connectionHandler(consumer) {
|
||||||
|
log.debug({method:'_connectionHandler', line:76, msg: 'new web consumer connecting'})
|
||||||
|
|
||||||
|
consumer.id = ++this.nextconsumerID // server assigned ID
|
||||||
|
this.consumers.set(consumer.id,consumer)
|
||||||
|
consumer.authenticated = false
|
||||||
|
consumer.connected = true
|
||||||
|
|
||||||
|
// add listeners
|
||||||
|
const consumerCloseHandler = (id) => {
|
||||||
|
log.warn({msg:'consumer connection closed during listen,',id:id})
|
||||||
|
this.removeconsumer(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
consumer.on('close', consumerCloseHandler.bind(this,consumer.id))
|
||||||
|
|
||||||
|
consumer.on('error', (err) => {
|
||||||
|
log.error({msg:'consumer connection error during listen',error:err})
|
||||||
|
// TODO do more handling than just logging
|
||||||
})
|
})
|
||||||
|
|
||||||
socket.on('close', (msg) => {
|
let [err] = await btc(this.authenticateconsumer)(consumer)
|
||||||
log.warn({msg:'client connection closed during listen',error:msg})
|
if (!this.allowAnonymous) {
|
||||||
})
|
if (err) {
|
||||||
|
consumer.close()// abort new connection consumer, cleanup, remove listeners
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (this.opts.conPacket) {
|
||||||
|
this.opts.conPacket._header = { id: 'pushed' }
|
||||||
|
log.debug({method:'_listen', line:171, conPacket: this.opts.conPacket, msg:'pushing a preset command to just connected consumer'})
|
||||||
|
this._send(consumer,this.opts.conPacket) // send a packet command on to consumer on connection
|
||||||
|
}
|
||||||
|
|
||||||
socket.on('message', messageProcess.bind(this, socket))
|
consumer._ping = setInterval( () => {
|
||||||
|
consumer.ping(JSON.stringify({pingInterval:this.pingInterval}))
|
||||||
|
this._send(consumer,{_header:{id:'ping'},pingInterval:this.pingInterval})
|
||||||
|
},this.pingInterval)
|
||||||
|
|
||||||
async function messageProcess(client, strPacket) {
|
consumer.on('message', messageProcess.bind(this, consumer))
|
||||||
log.debug({method:'_listen', line:76, packet: strPacket, msg:' incoming packet on web socket side'})
|
|
||||||
|
async function messageProcess(consumer, strPacket) {
|
||||||
|
log.debug({method:'_listen', line:76, packet: strPacket, msg:' incoming packet from web consumer'})
|
||||||
let res = {}
|
let res = {}
|
||||||
let [err, packet] = btc(JSON.parse)(strPacket)
|
let [err, packet] = btc(JSON.parse)(strPacket)
|
||||||
log.debug({method:'_listen', line:79, error:err, packet:packet, msg:'parsed websocket packet'})
|
|
||||||
if (err) {
|
if (err) {
|
||||||
res = { error: `Could not parse JSON: ${packet}` }
|
res = { error: 'Could not JSON parse packet', packet:strPacket }
|
||||||
} else {
|
}
|
||||||
if (packet.clientID) {
|
else {
|
||||||
client.ID = packet.clientID
|
log.warn({method:'_listen', line:266, packet:packet, msg:'parsed packet'})
|
||||||
res.cmd = 'ackID'
|
|
||||||
} else {
|
|
||||||
res = (await this._packetProcess(clone(packet))) || {}
|
res = (await this._packetProcess(clone(packet))) || {}
|
||||||
if (Object.keys(res).length === 0)
|
if (Object.keys(res).length === 0)
|
||||||
res = {
|
res = {
|
||||||
error:
|
error:
|
||||||
'socket packet command function likely did not return a promise',
|
'consumer packet command function likely did not return a promise',
|
||||||
packet: packet
|
packet: packet
|
||||||
}
|
}
|
||||||
}
|
|
||||||
}
|
|
||||||
if (packet) {
|
if (packet) {
|
||||||
res._header = clone(packet._header, false) || {} //make sure return packet has header with id in case it was removed in processing
|
res._header = clone(packet._header, false) || {} //make sure return packet has header with id in case it was removed in processing
|
||||||
delete packet._header // remove before adding to response header as request
|
delete packet._header // remove before adding to response header as request
|
||||||
|
@ -112,17 +281,21 @@ class Socket extends WebSocket.Server {
|
||||||
res._header.responder = { name: this.name, instanceID: this.id }
|
res._header.responder = { name: this.name, instanceID: this.id }
|
||||||
res._header.socket = this.address()
|
res._header.socket = this.address()
|
||||||
if (!res.cmd) res.cmd = 'reply' // by default return command is 'reply'
|
if (!res.cmd) res.cmd = 'reply' // by default return command is 'reply'
|
||||||
let sres = await send(res)
|
let [err] = await btc(this._send)(consumer,res)
|
||||||
log.debug({method:'_listen', line:105, packet: res, sendResponse:sres, msg:'packet reply send back to client'})
|
if (err) log.error({msg:err, error:err})
|
||||||
}
|
}
|
||||||
}) // end connected consumer
|
} // end message process
|
||||||
log.debug('socket created')
|
|
||||||
} // end listen
|
} // end connection Handler
|
||||||
|
|
||||||
async _destroy() {
|
async _destroy() {
|
||||||
log.debug({method:'_listen', line:105, msg:'closing down socket'})
|
log.debug({method:'_destroy', line:302, msg:'closing down http server and attached sockets', port:this.port})
|
||||||
|
this.consumers.forEach(consumer => {
|
||||||
|
consumer.terminate()
|
||||||
|
consumer.emit('close')
|
||||||
|
})
|
||||||
await this.close()
|
await this.close()
|
||||||
log.debug({method:'_listen', line:105, msg:'all connections closed....exiting'})
|
log.debug({method:'_listen', line:105, msg:'all connections closed, all consumer sockets deleted....exiting'})
|
||||||
process.exit()
|
process.exit()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -133,27 +306,23 @@ class Socket extends WebSocket.Server {
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
async push(packet, id) {
|
|
||||||
packet._header = { id: id || 'pushed' }
|
|
||||||
this.clients.forEach(async client => {
|
async _send(consumer, packet) {
|
||||||
if (client.readyState === WebSocket.OPEN) {
|
log.debug({msg:`sending to client:${consumer.id}:${consumer.name}`, packet:packet})
|
||||||
if (!id || id === client.ID) await this._send.bind(client)(packet)
|
return new Promise(async (resolve, reject) => {
|
||||||
}
|
// if (!consumer._connected) reject('can not send no connection')
|
||||||
|
let [err, message] = btc(JSON.stringify)(packet)
|
||||||
|
if (err) reject('unable to serialze the packet')
|
||||||
|
consumer.send(message,(err) => {
|
||||||
|
if (!err) resolve('packet written to socket stream successfully')
|
||||||
|
else reject(`error sending: ${err}`)
|
||||||
|
})
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// must have a consumer socket instance bound to call this function
|
|
||||||
async _send(packet) {
|
|
||||||
if (this.readyState !== 1) log.error({method:'_send', line:147, msg:`Connection not Ready, CODE:${this.readyState}`})
|
|
||||||
else {
|
|
||||||
let [err, message] = btc(JSON.stringify)(packet)
|
|
||||||
if (err) log.error({method:'_send', line:147, msg:`WS Could not JSON stringify: ${packet}`})
|
|
||||||
else {
|
|
||||||
try { this.send(message) }
|
|
||||||
catch(err) { log.error({method:'_send', line:153, msg:'error sending from ws server', error:err}) }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} // end class
|
} // end class
|
||||||
|
|
||||||
export default Socket
|
export default Socket
|
||||||
|
|
|
@ -1,2 +0,0 @@
|
||||||
*.sock
|
|
||||||
/node_modules/
|
|
|
@ -1 +0,0 @@
|
||||||
/node_modules/
|
|
|
@ -1,4 +0,0 @@
|
||||||
export default async function (packet) {
|
|
||||||
packet.payload = this.opts.port +':'+packet.payload
|
|
||||||
return packet
|
|
||||||
}
|
|
|
@ -1,14 +0,0 @@
|
||||||
import { Socket } from '../../src'
|
|
||||||
import process from './tcp-process'
|
|
||||||
|
|
||||||
let socket = new Socket({port:9080, name:'tcp socket 9080'})
|
|
||||||
|
|
||||||
socket.registerPacketProcessor(process)
|
|
||||||
;
|
|
||||||
(async () => {
|
|
||||||
|
|
||||||
await socket.create()
|
|
||||||
|
|
||||||
})().catch(err => {
|
|
||||||
console.error('FATAL: UNABLE TO START SYSTEM!\n',err)
|
|
||||||
})
|
|
|
@ -1,15 +0,0 @@
|
||||||
import { Socket } from '../../src'
|
|
||||||
import process from './tcp-process'
|
|
||||||
|
|
||||||
let socket = new Socket({name:'tcp socket'})
|
|
||||||
|
|
||||||
socket.registerPacketProcessor(process)
|
|
||||||
|
|
||||||
;
|
|
||||||
(async () => {
|
|
||||||
|
|
||||||
await socket.create()
|
|
||||||
|
|
||||||
})().catch(err => {
|
|
||||||
console.error('FATAL: UNABLE TO START SYSTEM!\n',err)
|
|
||||||
})
|
|
|
@ -1,11 +0,0 @@
|
||||||
import { Socket } from '../../src'
|
|
||||||
|
|
||||||
let socket = new Socket({path:true,name:'default-unix-socket'})
|
|
||||||
|
|
||||||
;
|
|
||||||
(async () => {
|
|
||||||
await socket.create()
|
|
||||||
|
|
||||||
})().catch(err => {
|
|
||||||
console.error('FATAL: UNABLE TO START SYSTEM!\n',err)
|
|
||||||
})
|
|
|
@ -1,23 +0,0 @@
|
||||||
import { Socket } from '../../src'
|
|
||||||
|
|
||||||
const USOCKET = 'usocket'
|
|
||||||
|
|
||||||
let socket = new Socket({path:USOCKET,name:'default-unix-socket'})
|
|
||||||
|
|
||||||
socket.test = 'at socket => '
|
|
||||||
|
|
||||||
socket.registerPacketProcessor(async function (packet) {
|
|
||||||
return new Promise((resolve) => {
|
|
||||||
packet.payload = 'overwrite default processor from instance '+ this.test + packet.payload
|
|
||||||
resolve(packet)
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
;
|
|
||||||
(async () => {
|
|
||||||
|
|
||||||
await socket.create()
|
|
||||||
|
|
||||||
})().catch(err => {
|
|
||||||
console.error('FATAL: UNABLE TO START SYSTEM!\n',err)
|
|
||||||
})
|
|
|
@ -1,78 +0,0 @@
|
||||||
import { spawn } from 'child_process'
|
|
||||||
import chai from 'chai'
|
|
||||||
import chaiAsPromised from 'chai-as-promised'
|
|
||||||
import btc from 'better-try-catch'
|
|
||||||
chai.use(chaiAsPromised)
|
|
||||||
const expect = chai.expect
|
|
||||||
|
|
||||||
import { Consumer } from '../src'
|
|
||||||
|
|
||||||
let tcpsocket_default = {}
|
|
||||||
let tcpsocket_9080 = {}
|
|
||||||
|
|
||||||
describe('Connects and Processes a payload in a JSON packet via TCP Socket', function(){
|
|
||||||
|
|
||||||
before(async function(){
|
|
||||||
tcpsocket_default = spawn('node',['-r', '@std/esm', './test/sockets/tcpsocket-default'])
|
|
||||||
tcpsocket_default.stdout.on('data', function(buf) {
|
|
||||||
console.log('[Socket]', String(buf))
|
|
||||||
})
|
|
||||||
|
|
||||||
tcpsocket_9080 = spawn('node',['-r', '@std/esm', './test/sockets/tcpsocket-9080'])
|
|
||||||
tcpsocket_9080.stdout.on('data', function(buf) {
|
|
||||||
console.log('[Socket]', String(buf))
|
|
||||||
})
|
|
||||||
|
|
||||||
})
|
|
||||||
|
|
||||||
after(async function(){
|
|
||||||
tcpsocket_default.kill()
|
|
||||||
tcpsocket_9080.kill()
|
|
||||||
})
|
|
||||||
|
|
||||||
it('with default host and port', async function () {
|
|
||||||
let tcpconsumer_default = new Consumer({name:'tcpconsumer'})
|
|
||||||
|
|
||||||
let [err] = await btc(tcpconsumer_default.connect)()
|
|
||||||
if (err) {
|
|
||||||
console.log('unable to connect to socket to start test', tcpconsumer_default.port)
|
|
||||||
process.kill(process.pid, 'SIGTERM')
|
|
||||||
}
|
|
||||||
|
|
||||||
tcpconsumer_default.registerPacketProcessor(async function (packet) {
|
|
||||||
return new Promise((resolve) => {
|
|
||||||
packet.payload = packet.payload +':local'
|
|
||||||
resolve(packet)})
|
|
||||||
})
|
|
||||||
|
|
||||||
let packet = {payload:'tcp payload'}
|
|
||||||
let res = await tcpconsumer_default.send(packet)
|
|
||||||
expect(res.payload).to.equal('8080:tcp payload:local')
|
|
||||||
|
|
||||||
}) // end tcp socket test
|
|
||||||
|
|
||||||
it('with alternate port, and registered consumer processor', async function () {
|
|
||||||
|
|
||||||
let tcpconsumer_9080 = new Consumer({port:9080, name:'tcp-consumer-9080'})
|
|
||||||
|
|
||||||
|
|
||||||
let [err] = await btc(tcpconsumer_9080.connect)()
|
|
||||||
if (err) {
|
|
||||||
console.log('unable to connect to socket to start test', tcpconsumer_9080.port)
|
|
||||||
process.kill(process.pid, 'SIGTERM')
|
|
||||||
}
|
|
||||||
|
|
||||||
tcpconsumer_9080.registerPacketProcessor(async function (packet) {
|
|
||||||
return new Promise((resolve) => {
|
|
||||||
packet.payload = packet.payload +':local'
|
|
||||||
resolve(packet)})
|
|
||||||
})
|
|
||||||
|
|
||||||
let packet = {payload:'tcp payload'}
|
|
||||||
let res = await tcpconsumer_9080.send(packet)
|
|
||||||
expect(res.payload).to.equal('9080:tcp payload:local')
|
|
||||||
|
|
||||||
}) // end tcp socket 2 test
|
|
||||||
|
|
||||||
|
|
||||||
}) // end describe
|
|
|
@ -1,83 +0,0 @@
|
||||||
import { spawn } from 'child_process'
|
|
||||||
import chai from 'chai'
|
|
||||||
import chaiAsPromised from 'chai-as-promised'
|
|
||||||
import btc from 'better-try-catch'
|
|
||||||
chai.use(chaiAsPromised)
|
|
||||||
const expect = chai.expect
|
|
||||||
|
|
||||||
import { Consumer } from '../src'
|
|
||||||
|
|
||||||
const SOCKET_FILE = 'usocket-default'
|
|
||||||
|
|
||||||
let consumer = new Consumer({path:true,name:'unix-consumer'})
|
|
||||||
let consumer2 = new Consumer({path:true, name:'unix-consumer2'})
|
|
||||||
|
|
||||||
let socket = {}
|
|
||||||
|
|
||||||
describe('Connects and Processes a payload via Unix Socket using JSON packet with defaults', function(){
|
|
||||||
|
|
||||||
before(async function(){
|
|
||||||
socket = spawn('node',['-r', '@std/esm', './test/sockets/'+SOCKET_FILE])
|
|
||||||
socket.stdout.on('data', function(buf) {
|
|
||||||
console.log('[Socket]', String(buf))
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
after(async function(){
|
|
||||||
socket.kill()
|
|
||||||
})
|
|
||||||
|
|
||||||
const TIMES = 3000
|
|
||||||
|
|
||||||
it(`Tests unix socket with default echo JSON packet procssing with ${TIMES} packets sent`, async function () {
|
|
||||||
|
|
||||||
let [err] = await btc(consumer.connect)()
|
|
||||||
if (err) {
|
|
||||||
console.log('unable to connect to socket to start test', consumer.path)
|
|
||||||
process.kill(process.pid, 'SIGTERM')
|
|
||||||
}
|
|
||||||
|
|
||||||
consumer.registerPacketProcessor(async function (packet) {
|
|
||||||
return new Promise((resolve) => {
|
|
||||||
packet.times += 1
|
|
||||||
if (packet.times === TIMES) packet.payload = 'local1:'+packet.payload
|
|
||||||
resolve(packet)})
|
|
||||||
})
|
|
||||||
|
|
||||||
|
|
||||||
let packet = {payload:'payload', times:0}
|
|
||||||
for (let i = 1; i <= TIMES; i++) {
|
|
||||||
packet = await consumer.send(packet)
|
|
||||||
}
|
|
||||||
expect(packet.payload+':'+packet.times).to.equal('local1:payload:'+TIMES)
|
|
||||||
|
|
||||||
}) // end unix socket test
|
|
||||||
|
|
||||||
|
|
||||||
it(`unix socket with two consumers alternating packets, ${TIMES} packets each and local processing`, async function () {
|
|
||||||
|
|
||||||
|
|
||||||
let [err] = await btc(consumer2.connect)()
|
|
||||||
if (err) {
|
|
||||||
console.log('unable to connect to socket to start test', consumer.path)
|
|
||||||
process.kill(process.pid, 'SIGTERM')
|
|
||||||
}
|
|
||||||
|
|
||||||
consumer2.registerPacketProcessor(async function (packet) {
|
|
||||||
return new Promise((resolve) => {
|
|
||||||
packet.times += 1
|
|
||||||
if (packet.times === TIMES) packet.payload = 'local2:'+packet.payload
|
|
||||||
resolve(packet)})
|
|
||||||
})
|
|
||||||
|
|
||||||
let packet = {consumer:1, payload:'payload', times:-1}
|
|
||||||
for (let i = 0; i < TIMES; i++) {
|
|
||||||
packet = await consumer.send(packet)
|
|
||||||
if (packet.times === TIMES) packet.times = 1
|
|
||||||
packet = await consumer2.send(packet)
|
|
||||||
}
|
|
||||||
expect(packet.payload+':'+packet.times).to.equal('local2:local1:payload:'+TIMES)
|
|
||||||
|
|
||||||
}) // end unix socket test
|
|
||||||
|
|
||||||
}) // end describe
|
|
|
@ -1,51 +0,0 @@
|
||||||
import { spawn } from 'child_process'
|
|
||||||
import chai from 'chai'
|
|
||||||
import chaiAsPromised from 'chai-as-promised'
|
|
||||||
import btc from 'better-try-catch'
|
|
||||||
chai.use(chaiAsPromised)
|
|
||||||
const expect = chai.expect
|
|
||||||
|
|
||||||
import { Consumer } from '../src'
|
|
||||||
|
|
||||||
const SOCKET_FILE = 'usocket'
|
|
||||||
|
|
||||||
let consumer = new Consumer({path:SOCKET_FILE,name:'unix-consumer'})
|
|
||||||
|
|
||||||
// const delay = time => new Promise(res=>setTimeout(()=>res(),time))
|
|
||||||
|
|
||||||
let socket = {}
|
|
||||||
|
|
||||||
describe('Connects and Processes a payload via Unix Socket using JSON packet with alt processor', function(){
|
|
||||||
|
|
||||||
before(async function(){
|
|
||||||
socket = spawn('node',['-r', '@std/esm', './test/sockets/'+SOCKET_FILE])
|
|
||||||
socket.stdout.on('data', function(buf) {
|
|
||||||
console.log('[Socket]', String(buf))
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
after(async function(){
|
|
||||||
socket.kill()
|
|
||||||
})
|
|
||||||
|
|
||||||
it('Tests alternate JSON packet procssing at socket and consumer', async function () {
|
|
||||||
|
|
||||||
let [err] = await btc(consumer.connect)()
|
|
||||||
if (err) {
|
|
||||||
console.log('unable to connect to socket to start test', consumer.path)
|
|
||||||
process.kill(process.pid, 'SIGTERM')
|
|
||||||
}
|
|
||||||
|
|
||||||
consumer.registerPacketProcessor(async function (packet) {
|
|
||||||
return new Promise((resolve) => {
|
|
||||||
packet.payload = 'local:'+packet.payload
|
|
||||||
resolve(packet)})
|
|
||||||
})
|
|
||||||
let packet = { payload:'payload'}
|
|
||||||
let res = await consumer.send(packet)
|
|
||||||
expect(res.payload).to.equal('local:overwrite default processor from instance at socket => payload')
|
|
||||||
|
|
||||||
|
|
||||||
}) // end unix socket test
|
|
||||||
|
|
||||||
}) // end describe
|
|
Loading…
Reference in New Issue