瀏覽代碼

Add features to errorhandler

master
peter 4 年之前
父節點
當前提交
98b271758b
共有 5 個檔案被更改,包括 98 行新增24 行删除
  1. 1
    1
      package.json
  2. 4
    4
      src/Backend.ts
  3. 1
    1
      src/Types.ts
  4. 5
    5
      src/Utils.ts
  5. 87
    13
      test/Test.ts

+ 1
- 1
package.json 查看文件

1
 {
1
 {
2
   "name": "rpclibrary",
2
   "name": "rpclibrary",
3
-  "version": "1.6.0",
3
+  "version": "1.6.1",
4
   "description": "rpclibrary is a websocket on steroids!",
4
   "description": "rpclibrary is a websocket on steroids!",
5
   "main": "./js/Index.js",
5
   "main": "./js/Index.js",
6
   "repository": {
6
   "repository": {

+ 4
- 4
src/Backend.ts 查看文件

40
             this.sesame = U.makeSesameFunction(conf.sesame)
40
             this.sesame = U.makeSesameFunction(conf.sesame)
41
         }
41
         }
42
 
42
 
43
-        this.errorHandler = (socket:I.Socket) => (error:any) => { 
44
-            if(conf.errorHandler) conf.errorHandler(socket, error)
43
+        this.errorHandler = (socket:I.Socket) => (error:any, rpcName:string, args: any[]) => { 
44
+            if(conf.errorHandler) conf.errorHandler(socket, error, rpcName, args)
45
             else throw error
45
             else throw error
46
         }
46
         }
47
         
47
         
69
         try{
69
         try{
70
             this.io.attach(this.ws)
70
             this.io.attach(this.ws)
71
             this.io.on('socket', (socket:I.Socket) => {
71
             this.io.on('socket', (socket:I.Socket) => {
72
-                socket.on('error', (err) => this.errorHandler(socket, err))
72
+                socket.on('error', (err) => this.errorHandler(socket, err, "bind", []))
73
                 socket.on('close', () => this.closeHandler(socket))
73
                 socket.on('close', () => this.closeHandler(socket))
74
                 this.connectionHandler(socket)
74
                 this.connectionHandler(socket)
75
                 this.initRPCs(socket)
75
                 this.initRPCs(socket)
76
             })
76
             })
77
             this.ws.listen(this.port, this.visibility)
77
             this.ws.listen(this.port, this.visibility)
78
         }catch(e){
78
         }catch(e){
79
-            this.errorHandler(this.io, e)
79
+            this.errorHandler(this.io, e, 'start', [])
80
         }
80
         }
81
     }
81
     }
82
 
82
 

+ 1
- 1
src/Types.ts 查看文件

5
 
5
 
6
 export type Visibility = "127.0.0.1" | "0.0.0.0"
6
 export type Visibility = "127.0.0.1" | "0.0.0.0"
7
 export type ConnectionHandler = (socket:I.Socket) => void
7
 export type ConnectionHandler = (socket:I.Socket) => void
8
-export type ErrorHandler = (socket:I.Socket, error:any) => void
8
+export type ErrorHandler = (socket:I.Socket, error:any, rpcName: string, args: any[]) => void
9
 export type CloseHandler = (socket:I.Socket) =>  void
9
 export type CloseHandler = (socket:I.Socket) =>  void
10
 export type SesameFunction = (sesame : string) => boolean
10
 export type SesameFunction = (sesame : string) => boolean
11
 export type ExceptionHandling = 'local' | 'remote'
11
 export type ExceptionHandling = 'local' | 'remote'

+ 5
- 5
src/Utils.ts 查看文件

74
         else
74
         else
75
             rpcFunction = info.call
75
             rpcFunction = info.call
76
 
76
 
77
-        socket.hook(ret.uniqueName, callGenerator(socket, rpcFunction, errorHandler))
77
+        socket.hook(ret.uniqueName, callGenerator(info.name, socket, rpcFunction, errorHandler))
78
         return ret
78
         return ret
79
     })
79
     })
80
 }
80
 }
83
  * Decorate an RPC with the error handler
83
  * Decorate an RPC with the error handler
84
  * @param rpcFunction the function to decorate
84
  * @param rpcFunction the function to decorate
85
  */
85
  */
86
-const callGenerator = (socket: I.Socket, rpcFunction : T.AnyFunction, errorHandler: T.ErrorHandler) : T.AnyFunction => {
86
+const callGenerator = (rpcName : string, socket: I.Socket, rpcFunction : T.AnyFunction, errorHandler: T.ErrorHandler) : T.AnyFunction => {
87
     const argsArr = extractArgs(rpcFunction)
87
     const argsArr = extractArgs(rpcFunction)
88
     const args = argsArr.join(',')
88
     const args = argsArr.join(',')
89
     const argsStr = argsArr.map(stripAfterEquals).join(',')
89
     const argsStr = argsArr.map(stripAfterEquals).join(',')
92
         try{
92
         try{
93
             return await rpcFunction(`+argsStr+`)
93
             return await rpcFunction(`+argsStr+`)
94
         }catch(e){
94
         }catch(e){
95
-            errorHandler(socket)(e)
95
+            errorHandler(socket)(e, rpcName, [`+args+`])
96
         }
96
         }
97
     }`)
97
     }`)
98
 }
98
 }
126
                 })
126
                 })
127
                 return res
127
                 return res
128
             }catch(e){
128
             }catch(e){
129
-                errorHandler(socket, e)
129
+                errorHandler(socket)(e, rpc.name, [`+args+`])
130
             }
130
             }
131
         }`)
131
         }`)
132
     }
132
     }
139
             })
139
             })
140
             return res
140
             return res
141
         }catch(e){
141
         }catch(e){
142
-            errorHandler(socket, e)
142
+            errorHandler(socket)(e, rpc.name, [`+args+`])
143
         }
143
         }
144
     }`)
144
     }`)
145
 }
145
 }

+ 87
- 13
test/Test.ts 查看文件

38
             function triggerCallback(...messages:any[]):number {return subcallback.apply({}, messages)},
38
             function triggerCallback(...messages:any[]):number {return subcallback.apply({}, messages)},
39
         ]
39
         ]
40
     }],{
40
     }],{
41
-        connectionHandler: (socket) => { console.log("connectionHandler OK") },
42
-        closeHandler: (socket) => { console.log("closeHandler OK") },
43
-        errorHandler: (socket, err) => { console.error("errorHandler OK SO YOU SHOULDN'T SEE THIS"); throw err }
41
+        connectionHandler: (socket) => {  },
42
+        closeHandler: (socket) => { },
43
+        errorHandler: (socket, err) => { throw err }
44
     })
44
     })
45
 }
45
 }
46
 
46
 
193
                 }
193
                 }
194
             }, 
194
             }, 
195
             function checkCandy():string { cb(candy); return candy },
195
             function checkCandy():string { cb(candy); return candy },
196
-            function stealCandy():string { candy = "_OK"; cb(candy); cb = (...any) => console.log.apply(console,["Server:", ...any]); return candy }
196
+            function stealCandy():string { candy = "_OK"; cb(candy); cb = () => {}; return candy }
197
         ]
197
         ]
198
         }],{
198
         }],{
199
-            connectionHandler: (socket) => { console.log("connectionHandler OK") },
200
-            closeHandler: (socket) => { console.log("closeHandler OK") },
201
-            errorHandler: (socket, err) => { console.error("errorHandler OK SO YOU SHOULDN'T SEE THIS"); throw err }
199
+            connectionHandler: (socket) => { },
200
+            closeHandler: (socket) => { },
201
+            errorHandler: (socket, err) => { throw err }
202
         })
202
         })
203
         client = new RPCSocket(21000, "localhost")
203
         client = new RPCSocket(21000, "localhost")
204
         return await client.connect()
204
         return await client.connect()
210
     })
210
     })
211
 
211
 
212
     it('Unhook+unsubscribe should stop callbacks', (done) => {
212
     it('Unhook+unsubscribe should stop callbacks', (done) => {
213
-        client['test'].subscribe(c => console.log("Client: "+c)).then( async (res: SubscriptionResponse) => {
213
+        client['test'].subscribe(c => {}).then( async (res: SubscriptionResponse) => {
214
             const r1 = await client['test'].checkCandy()
214
             const r1 = await client['test'].checkCandy()
215
             const r3 = await client['test'].stealCandy()
215
             const r3 = await client['test'].stealCandy()
216
             client.unhook(res.uuid)
216
             client.unhook(res.uuid)
217
-            console.log("---- No client output below this line")
218
             const r2 = await client['test'].checkCandy()
217
             const r2 = await client['test'].checkCandy()
219
             const r4 = await client['test'].checkCandy()
218
             const r4 = await client['test'].checkCandy()
220
-            console.log("---- More output below")
221
 
219
 
222
             if(r1 === "OK" && r3 === "_OK" && r2 === "_OK" && r4 === "_OK")
220
             if(r1 === "OK" && r3 === "_OK" && r2 === "_OK" && r4 === "_OK")
223
                 done()
221
                 done()
382
         })
380
         })
383
     })
381
     })
384
 
382
 
385
-    it("RPC throws server with handler", (done)=>{
383
+    it("RPC throws on server with handler", (done)=>{
386
         let server = new RPCServer(21004, [ {
384
         let server = new RPCServer(21004, [ {
387
             name: 'createUser' as 'createUser',
385
             name: 'createUser' as 'createUser',
388
             exportRPCs: () => [{
386
             exportRPCs: () => [{
389
                 name: 'createUser' as 'createUser',
387
                 name: 'createUser' as 'createUser',
390
                 call: createUser
388
                 call: createUser
391
         }]}], {
389
         }]}], {
392
-            errorHandler: (socket, e) => {
393
-                //console.log("EXPECTED SERVER EXCEPTION", String(e));
390
+            errorHandler: (socket, e, rpcName, args) => {
394
                 done()
391
                 done()
395
             }
392
             }
396
         })
393
         })
416
         })
413
         })
417
     })
414
     })
418
 })
415
 })
416
+
417
+describe("Errorhandler functionality", ()=>{
418
+    let createUser = async( user: {a:any,b:any}) => {
419
+        throw new Error("BAD BAD BAD")
420
+    }
421
+
422
+    it("correct values are passed to the handler", (done)=>{
423
+        let server = new RPCServer(21004, [ {
424
+            name: 'createUser' as 'createUser',
425
+            exportRPCs: () => [{
426
+                name: 'createUser' as 'createUser',
427
+                call: createUser
428
+        }]}], {
429
+            errorHandler: (socket, e, rpcName, args) => {
430
+                if(e.message === "BAD BAD BAD" && rpcName === "createUser" && args[0]['a'] === 'a' && args[0]['b'] === 'b')
431
+                    done()
432
+            }
433
+        })
434
+
435
+        let sock = new RPCSocket(21004, 'localhost')
436
+        sock.connect().then((cli) => {
437
+            cli["createUser"]["createUser"]({
438
+                a:'a',
439
+                b:'b'
440
+            })
441
+            .then(r => {
442
+                if(r != null)
443
+                done("UNEXPECTED RESULT " + r)
444
+            })
445
+            .catch((e) => {
446
+                done("UNEXPECTED CLIENT ERROR " + e)
447
+                done(e)
448
+            })
449
+            .finally(() => {
450
+                sock.destroy()
451
+                server.destroy()
452
+            })
453
+        })
454
+    })
455
+
456
+    it("handler sees sesame", (done)=>{
457
+        let sesame = "AAAAAAAAAAAAAAA"
458
+        let server = new RPCServer(21004, [ {
459
+            name: 'createUser' as 'createUser',
460
+            exportRPCs: () => [{
461
+                name: 'createUser' as 'createUser',
462
+                call: createUser
463
+        }]}], {
464
+            sesame: sesame,
465
+            errorHandler: (socket, e, rpcName, args) => {
466
+                if(e.message === "BAD BAD BAD" && rpcName === "createUser" && args[0] === sesame && args[1]['a'] === 'a' && args[1]['b'] === 'b')
467
+                    done()
468
+            }
469
+            
470
+        })
471
+
472
+        let sock = new RPCSocket(21004, 'localhost')
473
+        sock.connect(sesame).then((cli) => {
474
+            cli["createUser"]["createUser"]({
475
+                a:'a',
476
+                b:'b'
477
+            })
478
+            .then(r => {
479
+                if(r != null)
480
+                done("UNEXPECTED RESULT " + r)
481
+            })
482
+            .catch((e) => {
483
+                done("UNEXPECTED CLIENT ERROR " + e)
484
+                done(e)
485
+            })
486
+            .finally(() => {
487
+                sock.destroy()
488
+                server.destroy()
489
+            })
490
+        })
491
+    })
492
+})

Loading…
取消
儲存