实现 WebSocket 服务端

在 Node.js 中实现 WebSocket 服务端,可以使用 ws 库。ws 是一个简单易用的 WebSocket 模块,提供了 WebSocket Server 和 WebSocket Client 的实现。

以下是使用 ws 库创建 WebSocket 服务端的示例代码:

const WebSocket = require('ws');

// 创建 WebSocket 服务器
const wss = new WebSocket.Server({ port: 8080 });

// 监听客户端连接事件
wss.on('connection', (ws) => {
  console.log('client connected');

  // 监听客户端消息事件
  ws.on('message', (message) => {
    console.log(`received: ${message}`);

    // 将消息广播给所有客户端
    wss.clients.forEach((client) => {
      if (client !== ws && client.readyState === WebSocket.OPEN) {
        client.send(message);
      }
    });
  });

  // 监听客户端断开事件
  ws.on('close', () => {
    console.log('client disconnected');
  });
});

以上示例代码创建了一个 WebSocket 服务器,并监听客户端连接、消息、断开等事件。当客户端发送消息时,服务器会将该消息广播给所有已连接的客户端。

需要注意的是,WebSocket 服务器不同于传统的 HTTP 服务器,它需要在长时间保持连接的情况下运行。因此,在实际应用中,需要考虑到服务器的稳定性和扩展性,并对服务器进行相应的优化配置。

同时,为了能够处理更多连接和高并发请求,可结合 Node.js 的 Cluster 模块或使用 PM2 进行 WebSocket 服务器的集群化部署。

const cluster = require('cluster');
const numCPUs = require('os').cpus().length;
const WebSocket = require('ws');

if (cluster.isMaster) {
  // 创建工作进程
  for (let i = 0; i < numCPUs; i++) {
    cluster.fork();
  }

  // 监听工作进程退出事件
  cluster.on('exit', (worker, code, signal) => {
    console.log(`worker ${worker.process.pid} died`);
    cluster.fork();
  });
} else {
  // 创建 WebSocket 服务器
  const wss = new WebSocket.Server({ port: 8080 });

  // 监听客户端连接事件
  wss.on('connection', (ws) => {
    console.log('client connected');

    // 监听客户端消息事件
    ws.on('message', (message) => {
      console.log(`received: ${message}`);

      // 将消息广播给所有客户端
      wss.clients.forEach((client) => {
        if (client !== ws && client.readyState === WebSocket.OPEN) {
          client.send(message);
        }
      });
    });

    // 监听客户端断开事件
    ws.on('close', () => {
      console.log('client disconnected');
    });
  });
}

相比于单进程部署,多进程部署可以在一定程度上提高服务器的性能和稳定性,同时使用 Cluster 模块也可以让多进程部署变得更加容易。需要注意的是,在实际应用中,需要根据服务器的资源情况和负载量来配置工作进程数,以达到最佳的性能和稳定性。

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值