-
Notifications
You must be signed in to change notification settings - Fork 1
/
wsServer.js
33 lines (26 loc) · 985 Bytes
/
wsServer.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import express from "express";
import { WebSocketServer } from "ws";
const wsServer = (handle) => ({
listen(port, path, callback) {
const app = express();
const webSocketServer = new WebSocketServer({ noServer: true });
webSocketServer.on('connection', (client) => {
client.on('message', (message) => {
handle(message.toString())
.then((out) => {
if (out) {
client.send(JSON.stringify(out));
}
});
});
});
const httpServer = app.listen(port, callback);
httpServer.on('upgrade', (request, socket, head) => {
webSocketServer.handleUpgrade(request, socket, head, socket => {
webSocketServer.emit('connection', socket, request);
});
});
return { app, webSocketServer, httpServer };
},
});
export default wsServer;