fixed separation of packet properties for each socket type when customizing
add amendPacket.. and changed registerPacket to accept single type changes The four-in-one example now using amend and register sucessfullymaster
parent
a83646c4f6
commit
3c1340baf0
|
@ -9,7 +9,6 @@ module.exports = {
|
||||||
"node": true,
|
"node": true,
|
||||||
"mocha": true
|
"mocha": true
|
||||||
},
|
},
|
||||||
"parser": "babel-eslint",
|
|
||||||
"parserOptions": {
|
"parserOptions": {
|
||||||
"ecmaVersion": 2017
|
"ecmaVersion": 2017
|
||||||
,"sourceType": "module"
|
,"sourceType": "module"
|
||||||
|
|
|
@ -1,33 +1,53 @@
|
||||||
import Base from '../src/base'
|
import Base from '../src/base'
|
||||||
|
|
||||||
const USOCKET = __dirname + '/sample.sock'
|
const USOCKET = __dirname + '/sample.sock'
|
||||||
|
|
||||||
const delay = time => new Promise(res=>setTimeout(()=>res(),time))
|
|
||||||
|
|
||||||
;
|
;
|
||||||
(async () => {
|
(async () => {
|
||||||
|
|
||||||
// let app = new Base({com:'us,uc,ts,tc', id:'example', path: USOCKET, log:false})
|
const delay = time => new Promise(res=>setTimeout(()=>res(),time))
|
||||||
let app = new Base({com:'us,uc,ts,tc', id:'example'})
|
|
||||||
|
|
||||||
|
// let app = new Base({com:'us,uc,ts,tc', id:'example', path: USOCKET, log:false})
|
||||||
|
let app = new Base({sockets:'us,uc,ts,tc', id:'example', log:false})
|
||||||
|
|
||||||
await app.init()
|
await app.init()
|
||||||
|
console.log('after init')
|
||||||
|
app.amendPacketProcessing('tc',{
|
||||||
|
log: packet => {
|
||||||
|
console.log('==============Packet returned to TCP consumer==========')
|
||||||
|
console.dir(packet)
|
||||||
|
console.log('===========================')
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
app.registerPacketProcessor('ts',
|
||||||
|
async function (packet) {
|
||||||
|
packet.test = 'this went through custom tcp socket processor'
|
||||||
|
if (!packet.cmd) return {error: 'no command in packet', packet: packet }
|
||||||
|
if (this.context) if (this.context[packet.cmd]) return await this.context[packet.cmd].bind(this.context)(packet)
|
||||||
|
if (this[packet.cmd]) return await this[packet.cmd](packet)
|
||||||
|
return {error: 'no socket processing function supplied for command', packet: packet }
|
||||||
|
})
|
||||||
|
|
||||||
|
let packet = {}
|
||||||
console.log('ready=============sending============')
|
console.log('ready=============sending============')
|
||||||
let packet = {cmd:'echo', data:'some data to echo'}
|
// packet = {cmd:'echo', data:'some data to echo'}
|
||||||
await app.send(packet)
|
// await app.send(packet)
|
||||||
app.write = function(packet){
|
|
||||||
packet.cmd='log'
|
Object.assign(app,unixfuncs)
|
||||||
packet.response='return of write command'
|
app.registerPacketContext('ts',tcpfuncs)
|
||||||
return packet
|
|
||||||
}
|
|
||||||
app.write2 = function(packet){
|
|
||||||
packet.cmd='log'
|
|
||||||
packet.response='return of write2 command'
|
|
||||||
return packet
|
|
||||||
}
|
|
||||||
app.registerPacketContext(app)
|
|
||||||
packet = {cmd:'write', data:'data to write'}
|
packet = {cmd:'write', data:'data to write'}
|
||||||
await app.send(packet)
|
await app.send(packet)
|
||||||
|
await delay(1000)
|
||||||
|
app.amendPacketContext(
|
||||||
|
{write: function(packet){
|
||||||
|
packet.cmd='log'
|
||||||
|
packet.response='return of AMMEDED write command'
|
||||||
|
return packet
|
||||||
|
}}
|
||||||
|
)
|
||||||
|
await delay(1000)
|
||||||
|
packet = {cmd:'write', data:'2ND data to write'}
|
||||||
|
await app.send(packet)
|
||||||
packet = {cmd:'write2', data:'data to write'}
|
packet = {cmd:'write2', data:'data to write'}
|
||||||
await app.send(packet)
|
await app.send(packet)
|
||||||
|
|
||||||
|
@ -37,4 +57,33 @@ const delay = time => new Promise(res=>setTimeout(()=>res(),time))
|
||||||
|
|
||||||
})().catch(err => {
|
})().catch(err => {
|
||||||
console.error('FATAL: UNABLE TO START SYSTEM!\n',err)
|
console.error('FATAL: UNABLE TO START SYSTEM!\n',err)
|
||||||
|
process.kill(process.pid, 'SIGTERM')
|
||||||
})
|
})
|
||||||
|
|
||||||
|
const unixfuncs = {
|
||||||
|
write: function(packet){
|
||||||
|
packet.cmd='log'
|
||||||
|
packet.response='return of write command'
|
||||||
|
return packet
|
||||||
|
},
|
||||||
|
write2: function(packet){
|
||||||
|
packet.cmd='log'
|
||||||
|
packet.response='return of write2 command'
|
||||||
|
return packet
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const tcpfuncs = {
|
||||||
|
write: function(packet){
|
||||||
|
packet.cmd='log'
|
||||||
|
packet.response='return of write command'
|
||||||
|
packet.via = 'tcp write'
|
||||||
|
return packet
|
||||||
|
},
|
||||||
|
write2: function(packet){
|
||||||
|
packet.cmd='log'
|
||||||
|
packet.response='return of write2 command'
|
||||||
|
packet.via = 'tcp write2'
|
||||||
|
return packet
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
|
@ -27,11 +27,9 @@
|
||||||
"@std/esm": "cjs",
|
"@std/esm": "cjs",
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@std/esm": "^0.18.0",
|
"@std/esm": "^0.18.0",
|
||||||
"babel-eslint": "^8.2.1",
|
|
||||||
"chai": "^4.1.2",
|
"chai": "^4.1.2",
|
||||||
"chai-as-promised": "^7.1.1",
|
"chai-as-promised": "^7.1.1",
|
||||||
"codecov": "^3.0.0",
|
"codecov": "^3.0.0",
|
||||||
"eslint": "^3.19.0",
|
|
||||||
"istanbul": "^0.4.5",
|
"istanbul": "^0.4.5",
|
||||||
"mocha": "^4.0.1"
|
"mocha": "^4.0.1"
|
||||||
},
|
},
|
||||||
|
|
64
src/base.mjs
64
src/base.mjs
|
@ -1,9 +1,12 @@
|
||||||
import { Socket, Consumer } from '@uci/socket'
|
// import { Socket, Consumer } from '@uci/socket'
|
||||||
|
import { Socket, Consumer } from '../../uci-socket/src'
|
||||||
import packet from './packet.mjs'
|
import packet from './packet.mjs'
|
||||||
import EventEmitter from 'events'
|
import EventEmitter from 'events'
|
||||||
|
|
||||||
const USOCKET = __dirname + '/unix.sock'
|
const USOCKET = __dirname + '/unix.sock'
|
||||||
|
|
||||||
|
const delay = time => new Promise(res=>setTimeout(()=>res(),time))
|
||||||
|
|
||||||
export default class Base extends EventEmitter {
|
export default class Base extends EventEmitter {
|
||||||
constructor(opts={}) {
|
constructor(opts={}) {
|
||||||
super()
|
super()
|
||||||
|
@ -12,50 +15,77 @@ export default class Base extends EventEmitter {
|
||||||
this.desc = opts.desc // additional details for humans
|
this.desc = opts.desc // additional details for humans
|
||||||
// attach unix socket OR consumer(default) only if path is supplied
|
// attach unix socket OR consumer(default) only if path is supplied
|
||||||
this.socket={}
|
this.socket={}
|
||||||
this.com = opts.com.split(/[,:|\s]+/)
|
opts.sockets.split(/[,:|\s]+/).forEach( sock => {
|
||||||
this.com.forEach( sock => {
|
|
||||||
opts.name = this.id +':'+sock
|
opts.name = this.id +':'+sock
|
||||||
opts.type = sock
|
if (!opts[sock]) opts[sock] = {}
|
||||||
switch (sock) {
|
switch (sock) {
|
||||||
case
|
case
|
||||||
'us':
|
'us':
|
||||||
this.socket[sock] = new Socket(opts.path,opts)
|
opts.us.path = opts.us.path || opts.path
|
||||||
|
this.socket[sock] = new Socket(opts.us.path,opts)
|
||||||
break
|
break
|
||||||
case 'uc':
|
case 'uc':
|
||||||
this.socket[sock] = new Consumer(opts.path,opts)
|
opts.uc.path = opts.uc.path || opts.path
|
||||||
|
this.socket[sock] = new Consumer(opts.uc.path,opts)
|
||||||
break
|
break
|
||||||
case 'ts':
|
case 'ts':
|
||||||
this.socket[sock]= new Socket(opts)
|
this.socket[sock]= new Socket(opts.ts, opts)
|
||||||
break
|
break
|
||||||
case 'tc':
|
case 'tc':
|
||||||
this.socket[sock]= new Consumer(opts)
|
this.socket[sock]= new Consumer(opts.tc,opts)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
} // end constructor
|
} // end constructor
|
||||||
|
|
||||||
async init (context) {
|
async init (context) {
|
||||||
context = context || this // context is Base if not supplied
|
context = context || this // context is Base for all if not supplied
|
||||||
for(let type of Object.keys(this.socket)){
|
for(let type of Object.keys(this.socket)){
|
||||||
if (type.indexOf('s')!==-1) {
|
if (type.indexOf('s')!==-1) {
|
||||||
this.socket[type].packet = packet.socket
|
this.socket[type].packet = Object.assign({},packet.socket) // default packet processing
|
||||||
await this.socket[type].create(context)
|
await this.socket[type].create(context)
|
||||||
|
// setTimeout(context =>{this.socket[type].create(context)},500)
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
this.socket[type].packet = packet.consumer
|
this.socket[type].packet = Object.assign({},packet.consumer) // default packet processing
|
||||||
await this.socket[type].connect(context)
|
await this.socket[type].connect(context)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} // init
|
} // init
|
||||||
|
|
||||||
registerPacketContext(context) {
|
registerPacketContext(type, context) {
|
||||||
for(let type of Object.keys(this.socket)){
|
if (typeof type === 'string') {
|
||||||
this.socket[type].registerPacketContext(context)
|
this.socket[type].registerPacketContext(Object.assign({},context))
|
||||||
|
} else {
|
||||||
|
context=type
|
||||||
|
for(let type of Object.keys(this.socket)){
|
||||||
|
this.socket[type].registerPacketContext(context)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
registerPacketProcessor(func) {
|
amendPacketContext(type, context) {
|
||||||
for(let type of Object.keys(this.socket)){
|
if (typeof type === 'string') {
|
||||||
this.socket[type].registerPacketProcessor(func)
|
Object.assign(this.socket[type].packet.context,context)
|
||||||
|
} else {
|
||||||
|
context=type
|
||||||
|
for(let type of Object.keys(this.socket)){
|
||||||
|
Object.assign(this.socket[type].packet.context,context)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
amendPacketProcessing(type,funcs) {
|
||||||
|
this.socket[type].packet = Object.assign({},this.socket[type].packet,funcs)
|
||||||
|
}
|
||||||
|
|
||||||
|
registerPacketProcessor(type,func) {
|
||||||
|
if (typeof type === 'string') {
|
||||||
|
this.socket[type].packet._process = func
|
||||||
|
} else {
|
||||||
|
func = type
|
||||||
|
for(let type of Object.keys(this.socket)){
|
||||||
|
this.socket[type].registerPacketProcessor(func)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -12,7 +12,7 @@ export default {
|
||||||
echo: packet => {
|
echo: packet => {
|
||||||
packet.processed = true
|
packet.processed = true
|
||||||
packet.cmd = 'log'
|
packet.cmd = 'log'
|
||||||
packet.info = 'echoed'
|
packet.info = 'default socket echo'
|
||||||
return packet
|
return packet
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
|
|
Loading…
Reference in New Issue