diff options
Diffstat (limited to 'websocket_server.go')
| -rw-r--r-- | websocket_server.go | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/websocket_server.go b/websocket_server.go index 0c1d965..5928c73 100644 --- a/websocket_server.go +++ b/websocket_server.go @@ -53,16 +53,23 @@ func (trunks *Trunks) initWebSocketServer() (err error) { trunks.Wsd = websocket.NewServer(opts) - trunks.Wsd.RegisterTextHandler( + err = trunks.Wsd.RegisterTextHandler( "POST", apiAttackHttp, trunks.handleWsAttackHttp, ) - trunks.Wsd.RegisterTextHandler( + if err != nil { + return err + } + + err = trunks.Wsd.RegisterTextHandler( "DELETE", apiAttackHttp, trunks.handleWsAttackHttpCancel, ) + if err != nil { + return err + } return nil } |