使用多个服务器实例扩展 websockets/ws

Question

我在单机上使用websockets/ws 。 它的工作正常。 我想在多核和多个实例上水平扩展它。 对于多核，我尝试使用pm2 ，它似乎工作得很好。

第一问：这是最好的方法还是合适的方法？ 这是我使用 pm2 的测试代码

// ws-server.js
const WebSocket = require('ws');

const wss = new WebSocket.Server({ port: 3131 });

var pid = process.pid + ''
console.log('process pid: '+ pid)

wss.on('connection', function connection(ws) {
  ws.on('message', function incoming(message) {
    if (message === 'get-pid') {
      ws.send('pid-' + pid)
    } else {
      var matched = pid === message ? 'old friends' : 'strangers' 
      ws.send([pid, message, 'we are ' + matched].join(', '))
    }
  });
  ws.send('first time')
});

和客户端 websocket 实例

// ws-cient.js
const WebSocket = require('ws');
const ws = new WebSocket('ws://localhost:3131/');

var pid
ws.on('open', function open() {
  ws.send('get-pid');
  setInterval(function() {
    ws.send(pid)
  }, 1000)
});

ws.on('message', function incoming(data) {
  if (/^pid/.test(data)) {
    pid = data.match(/\d+/)[0]
    console.log('got pid: ' + pid) 
  } else {
    console.log(data)
  }
});

只需使用 pm2 运行服务器和客户端

   $ pm2 start ws-server.js -i 50
   $ pm2 start ws-client.js -i 50

如果您现在看到日志pm2 logs ws-client每个客户端每秒都会访问相同的连接（在服务器上）。 因此对于多核 ws 与 PM2 配合良好。

第二个问题：如何扩展多个实例？ 我刚刚看到了用于水平缩放的SocketCluster ，但它可以与 websockets/ws 一起使用，因为我已经用 ws 开发了代码。 什么是水平缩放的其他解决方案。

Answer 1

回答这个问题有点晚了，但对于仍然面临这个问题的任何人，我写了一篇关于在 medium 上集群 WebSocket 服务器的文章。

https://medium.com/@mohsenes/websocket-cluster-with-nestjs-and-redis-a18882d418ed

Answer 2

不幸的是，跨不同进程扩展 Websocket 会很困难，我建议你使用这个库：

https://github.com/ClusterWS/ClusterWS

该库的主要目的是跨进程和机器扩展 WebSocket。 此外，好消息是图书馆小而快。