feat: added on connection websocket server callback

This commit is contained in:
Roberto Tonino 2021-05-11 21:06:45 +02:00
parent a6fe1bff19
commit cbd6bd4791
1 changed files with 9 additions and 3 deletions

View File

@ -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))