在Node.js中也能实现一样的编程姿势——WebSocket

2023-12-09 12:26:48

WebSocket，作为一种基于TCP全双工通信协议，在各种领域的应用可谓是十分广泛。在上一篇文章中，我们已经介绍了如何使用浏览器端的WebSocket。今天，我们继续以一种相似的编程姿势，利用Node.js来实现一个WebSocket服务端。

在Node.js生态中，拥有许多成熟稳定的WebSocket库，而其中较受欢迎的两大库是socket.io和ws。

socket.io

socket.io是由Automattic开发的一个非常强大的WebSocket库。它允许你使用中间件、路由以及自定义事件轻松创建WebSocket服务器。此外，socket.io还支持多种传输协议，包括WebSocket、HTTP长轮询、轮询以及Flash Socket，这使得它非常适合需要跨浏览器支持的应用程序。

优点：

强大且全面： socket.io是一个非常全面的WebSocket库，它提供了许多有用的功能，如中间件、路由、自定义事件以及多种传输协议支持。
活跃的社区： socket.io拥有一个活跃且庞大的社区，这使得你可以在遇到问题时获得帮助。
易于使用： socket.io非常易于使用，即使你对WebSocket一无所知，也可以在短时间内学会如何使用它。

缺点：

体积庞大： socket.io的体积相对较大，这可能会增加你的应用程序的加载时间。
较重的依赖关系： socket.io依赖于许多其他的库，这可能会增加你的应用程序的复杂性。

ws

ws是由Brian McKelvey开发的一个轻量级的WebSocket库。它提供了基本的WebSocket功能，如发送和接收消息，以及对WebSocket事件的监听。与socket.io不同的是，ws只支持WebSocket协议。

优点：

轻量级： ws非常轻量级，这使得它非常适合资源受限的应用程序。
简单的API： ws拥有非常简单的API，这使得它非常易于使用。
良好的文档： ws拥有非常全面的文档，这使得你可以在遇到问题时轻松找到解决方案。

缺点：

功能有限： ws的功能相对有限，如果你需要使用更高级的功能，如中间件、路由或自定义事件，则需要使用其他WebSocket库。
较少的支持： ws的社区相对较小，这可能会让你在遇到问题时难以获得帮助。

总结

socket.io和ws都是非常优秀的WebSocket库，它们都有着各自的优缺点。如果你需要一个功能强大且全面的WebSocket库，那么socket.io是一个不错的选择。如果你需要一个轻量级且简单的WebSocket库，那么ws是一个不错的选择。

如何使用socket.io创建WebSocket服务端

// 引入socket.io
const socketIO = require("socket.io");

// 创建一个HTTP服务器
const httpServer = require("http").createServer();

// 将socket.io应用到HTTP服务器
const io = socketIO(httpServer);

// 监听客户端连接事件
io.on("connection", (socket) => {
  console.log("客户端已连接");

  // 监听客户端发送的消息
  socket.on("message", (message) => {
    console.log("收到客户端消息：", message);

    // 将消息发送回客户端
    socket.emit("message", "你好，客户端");
  });

  // 监听客户端断开连接事件
  socket.on("disconnect", () => {
    console.log("客户端已断开连接");
  });
});

// 启动HTTP服务器
httpServer.listen(3000, () => {
  console.log("HTTP服务器已启动，端口号为3000");
});

如何使用ws创建WebSocket服务端

// 引入ws
const WebSocket = require("ws");

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

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

  // 监听客户端发送的消息
  ws.on("message", (message) => {
    console.log("收到客户端消息：", message);

    // 将消息发送回客户端
    ws.send("你好，客户端");
  });

  // 监听客户端断开连接事件
  ws.on("close", () => {
    console.log("客户端已断开连接");
  });
});

console.log("WebSocket服务器已启动，端口号为3000");