From cbd6bd47916639437ff87918c432bf7c55987085 Mon Sep 17 00:00:00 2001 From: Roberto Tonino Date: Tue, 11 May 2021 21:06:45 +0200 Subject: [PATCH] feat: added on connection websocket server callback --- server/src/app.ts | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/server/src/app.ts b/server/src/app.ts index 59c0193..3c82699 100644 --- a/server/src/app.ts +++ b/server/src/app.ts @@ -15,7 +15,7 @@ const PORT = normalizePort(process.env.PORT || '6595') const debug = initDebug('deemix-gui:server') export const app: Application = express() -const ws = new WsServer({ noServer: true }) +const wss = new WsServer({ noServer: true }) const server = http.createServer(app) /* === Middlewares === */ @@ -35,10 +35,16 @@ if (process.env.NODE_ENV !== 'test') { server.listen(PORT) } +wss.on('connection', ws => { + ws.on('message', message => { + console.log('received: %s', message) + }) +}) + /* === Server callbacks === */ server.on('upgrade', (request, socket, head) => { - ws.handleUpgrade(request, socket, head, socket => { - ws.emit('connection', socket, request) + wss.handleUpgrade(request, socket, head, socket => { + wss.emit('connection', socket, request) }) }) server.on('error', getErrorCb(PORT))