Add features to errorhandler

This commit is contained in:
2020-01-20 01:03:35 +01:00
parent c4e280077b
commit 98b271758b
5 changed files with 98 additions and 24 deletions
+1 -1
View File
@@ -1,6 +1,6 @@
{ {
"name": "rpclibrary", "name": "rpclibrary",
"version": "1.6.0", "version": "1.6.1",
"description": "rpclibrary is a websocket on steroids!", "description": "rpclibrary is a websocket on steroids!",
"main": "./js/Index.js", "main": "./js/Index.js",
"repository": { "repository": {
+4 -4
View File
@@ -40,8 +40,8 @@ export class RPCServer<
this.sesame = U.makeSesameFunction(conf.sesame) this.sesame = U.makeSesameFunction(conf.sesame)
} }
this.errorHandler = (socket:I.Socket) => (error:any) => { this.errorHandler = (socket:I.Socket) => (error:any, rpcName:string, args: any[]) => {
if(conf.errorHandler) conf.errorHandler(socket, error) if(conf.errorHandler) conf.errorHandler(socket, error, rpcName, args)
else throw error else throw error
} }
@@ -69,14 +69,14 @@ export class RPCServer<
try{ try{
this.io.attach(this.ws) this.io.attach(this.ws)
this.io.on('socket', (socket:I.Socket) => { this.io.on('socket', (socket:I.Socket) => {
socket.on('error', (err) => this.errorHandler(socket, err)) socket.on('error', (err) => this.errorHandler(socket, err, "bind", []))
socket.on('close', () => this.closeHandler(socket)) socket.on('close', () => this.closeHandler(socket))
this.connectionHandler(socket) this.connectionHandler(socket)
this.initRPCs(socket) this.initRPCs(socket)
}) })
this.ws.listen(this.port, this.visibility) this.ws.listen(this.port, this.visibility)
}catch(e){ }catch(e){
this.errorHandler(this.io, e) this.errorHandler(this.io, e, 'start', [])
} }
} }
+1 -1
View File
@@ -5,7 +5,7 @@ export type HookFunction<F extends AnyFunction = AnyFunction, SubresT = {}> = (.
export type Visibility = "127.0.0.1" | "0.0.0.0" export type Visibility = "127.0.0.1" | "0.0.0.0"
export type ConnectionHandler = (socket:I.Socket) => void export type ConnectionHandler = (socket:I.Socket) => void
export type ErrorHandler = (socket:I.Socket, error:any) => void export type ErrorHandler = (socket:I.Socket, error:any, rpcName: string, args: any[]) => void
export type CloseHandler = (socket:I.Socket) => void export type CloseHandler = (socket:I.Socket) => void
export type SesameFunction = (sesame : string) => boolean export type SesameFunction = (sesame : string) => boolean
export type ExceptionHandling = 'local' | 'remote' export type ExceptionHandling = 'local' | 'remote'
+5 -5
View File
@@ -74,7 +74,7 @@ export function rpcHooker<SubResT = {}>(socket: I.Socket, exporter:I.RPCExporter
else else
rpcFunction = info.call rpcFunction = info.call
socket.hook(ret.uniqueName, callGenerator(socket, rpcFunction, errorHandler)) socket.hook(ret.uniqueName, callGenerator(info.name, socket, rpcFunction, errorHandler))
return ret return ret
}) })
} }
@@ -83,7 +83,7 @@ export function rpcHooker<SubResT = {}>(socket: I.Socket, exporter:I.RPCExporter
* Decorate an RPC with the error handler * Decorate an RPC with the error handler
* @param rpcFunction the function to decorate * @param rpcFunction the function to decorate
*/ */
const callGenerator = (socket: I.Socket, rpcFunction : T.AnyFunction, errorHandler: T.ErrorHandler) : T.AnyFunction => { const callGenerator = (rpcName : string, socket: I.Socket, rpcFunction : T.AnyFunction, errorHandler: T.ErrorHandler) : T.AnyFunction => {
const argsArr = extractArgs(rpcFunction) const argsArr = extractArgs(rpcFunction)
const args = argsArr.join(',') const args = argsArr.join(',')
const argsStr = argsArr.map(stripAfterEquals).join(',') const argsStr = argsArr.map(stripAfterEquals).join(',')
@@ -92,7 +92,7 @@ const callGenerator = (socket: I.Socket, rpcFunction : T.AnyFunction, errorHandl
try{ try{
return await rpcFunction(`+argsStr+`) return await rpcFunction(`+argsStr+`)
}catch(e){ }catch(e){
errorHandler(socket)(e) errorHandler(socket)(e, rpcName, [`+args+`])
} }
}`) }`)
} }
@@ -126,7 +126,7 @@ const hookGenerator = (rpc:T.HookRPC<any, any, any>, errorHandler: T.ErrorHandle
}) })
return res return res
}catch(e){ }catch(e){
errorHandler(socket, e) errorHandler(socket)(e, rpc.name, [`+args+`])
} }
}`) }`)
} }
@@ -139,7 +139,7 @@ const hookGenerator = (rpc:T.HookRPC<any, any, any>, errorHandler: T.ErrorHandle
}) })
return res return res
}catch(e){ }catch(e){
errorHandler(socket, e) errorHandler(socket)(e, rpc.name, [`+args+`])
} }
}`) }`)
} }
+87 -13
View File
@@ -38,9 +38,9 @@ function makeServer(){
function triggerCallback(...messages:any[]):number {return subcallback.apply({}, messages)}, function triggerCallback(...messages:any[]):number {return subcallback.apply({}, messages)},
] ]
}],{ }],{
connectionHandler: (socket) => { console.log("connectionHandler OK") }, connectionHandler: (socket) => { },
closeHandler: (socket) => { console.log("closeHandler OK") }, closeHandler: (socket) => { },
errorHandler: (socket, err) => { console.error("errorHandler OK SO YOU SHOULDN'T SEE THIS"); throw err } errorHandler: (socket, err) => { throw err }
}) })
} }
@@ -193,12 +193,12 @@ describe('It should do unhook', () => {
} }
}, },
function checkCandy():string { cb(candy); return candy }, function checkCandy():string { cb(candy); return candy },
function stealCandy():string { candy = "_OK"; cb(candy); cb = (...any) => console.log.apply(console,["Server:", ...any]); return candy } function stealCandy():string { candy = "_OK"; cb(candy); cb = () => {}; return candy }
] ]
}],{ }],{
connectionHandler: (socket) => { console.log("connectionHandler OK") }, connectionHandler: (socket) => { },
closeHandler: (socket) => { console.log("closeHandler OK") }, closeHandler: (socket) => { },
errorHandler: (socket, err) => { console.error("errorHandler OK SO YOU SHOULDN'T SEE THIS"); throw err } errorHandler: (socket, err) => { throw err }
}) })
client = new RPCSocket(21000, "localhost") client = new RPCSocket(21000, "localhost")
return await client.connect() return await client.connect()
@@ -210,14 +210,12 @@ describe('It should do unhook', () => {
}) })
it('Unhook+unsubscribe should stop callbacks', (done) => { it('Unhook+unsubscribe should stop callbacks', (done) => {
client['test'].subscribe(c => console.log("Client: "+c)).then( async (res: SubscriptionResponse) => { client['test'].subscribe(c => {}).then( async (res: SubscriptionResponse) => {
const r1 = await client['test'].checkCandy() const r1 = await client['test'].checkCandy()
const r3 = await client['test'].stealCandy() const r3 = await client['test'].stealCandy()
client.unhook(res.uuid) client.unhook(res.uuid)
console.log("---- No client output below this line")
const r2 = await client['test'].checkCandy() const r2 = await client['test'].checkCandy()
const r4 = await client['test'].checkCandy() const r4 = await client['test'].checkCandy()
console.log("---- More output below")
if(r1 === "OK" && r3 === "_OK" && r2 === "_OK" && r4 === "_OK") if(r1 === "OK" && r3 === "_OK" && r2 === "_OK" && r4 === "_OK")
done() done()
@@ -382,15 +380,14 @@ describe('Error handling', ()=>{
}) })
}) })
it("RPC throws server with handler", (done)=>{ it("RPC throws on server with handler", (done)=>{
let server = new RPCServer(21004, [ { let server = new RPCServer(21004, [ {
name: 'createUser' as 'createUser', name: 'createUser' as 'createUser',
exportRPCs: () => [{ exportRPCs: () => [{
name: 'createUser' as 'createUser', name: 'createUser' as 'createUser',
call: createUser call: createUser
}]}], { }]}], {
errorHandler: (socket, e) => { errorHandler: (socket, e, rpcName, args) => {
//console.log("EXPECTED SERVER EXCEPTION", String(e));
done() done()
} }
}) })
@@ -416,3 +413,80 @@ describe('Error handling', ()=>{
}) })
}) })
}) })
describe("Errorhandler functionality", ()=>{
let createUser = async( user: {a:any,b:any}) => {
throw new Error("BAD BAD BAD")
}
it("correct values are passed to the handler", (done)=>{
let server = new RPCServer(21004, [ {
name: 'createUser' as 'createUser',
exportRPCs: () => [{
name: 'createUser' as 'createUser',
call: createUser
}]}], {
errorHandler: (socket, e, rpcName, args) => {
if(e.message === "BAD BAD BAD" && rpcName === "createUser" && args[0]['a'] === 'a' && args[0]['b'] === 'b')
done()
}
})
let sock = new RPCSocket(21004, 'localhost')
sock.connect().then((cli) => {
cli["createUser"]["createUser"]({
a:'a',
b:'b'
})
.then(r => {
if(r != null)
done("UNEXPECTED RESULT " + r)
})
.catch((e) => {
done("UNEXPECTED CLIENT ERROR " + e)
done(e)
})
.finally(() => {
sock.destroy()
server.destroy()
})
})
})
it("handler sees sesame", (done)=>{
let sesame = "AAAAAAAAAAAAAAA"
let server = new RPCServer(21004, [ {
name: 'createUser' as 'createUser',
exportRPCs: () => [{
name: 'createUser' as 'createUser',
call: createUser
}]}], {
sesame: sesame,
errorHandler: (socket, e, rpcName, args) => {
if(e.message === "BAD BAD BAD" && rpcName === "createUser" && args[0] === sesame && args[1]['a'] === 'a' && args[1]['b'] === 'b')
done()
}
})
let sock = new RPCSocket(21004, 'localhost')
sock.connect(sesame).then((cli) => {
cli["createUser"]["createUser"]({
a:'a',
b:'b'
})
.then(r => {
if(r != null)
done("UNEXPECTED RESULT " + r)
})
.catch((e) => {
done("UNEXPECTED CLIENT ERROR " + e)
done(e)
})
.finally(() => {
sock.destroy()
server.destroy()
})
})
})
})