使用 Node.js 开发 TCP 客户端并拆分连续消息包

在网络编程中，TCP 是一种可靠的传输协议，它确保数据的完整性和顺序性。然而，在实际开发 TCP 客户端时，我们常常会遇到一个挑战：由于 TCP 是面向流的协议，接收到的数据可能并不总是按预期的消息边界到达，尤其是在高速数据传输时，多个消息包可能会连续到达。这时，我们需要自己处理数据的拆分和重组，以确保消息的完整性。

问题描述

假设我们在开发一个 TCP 客户端，该客户端接收到的每条消息都有固定的起始标志符和结束标志符。当收到的数据流中包含多个连续的消息包时，我们需要将它们正确地拆分并处理。

解决方案

为了解决这个问题，我们可以通过以下步骤来处理连续的消息包：

维护一个缓存区：使用一个缓存区来存储接收到的数据。这样即使消息包分成了多次接收，我们也可以将它们重组在一起。
检测起始标志符和结束标志符：在接收到的数据中查找起始标志符和结束标志符，并将完整的消息提取出来。
处理并移除已处理的消息：一旦我们找到了完整的消息，就可以处理它，并从缓存区中移除已经处理的部分。

下面是一个示例代码，展示了如何实现这些步骤：

import net from "net";
import fse from "fs-extra";

// 定义起始标志符和结束标志符
const START_FLAG = Buffer.from([0xeb, 0x90]);
const END_FLAG = Buffer.from([0xeb, 0x90]);

class TcpClient {
  client: net.Socket;
  buffer: Buffer;

  constructor() {
    this.client = new net.Socket();
    this.buffer = Buffer.alloc(0);
    this.client.on("data", this.handleResponse.bind(this));
  }

  handleResponse(Buffer) {
    // 将接收到的数据追加到缓存区
    this.buffer = Buffer.concat([this.buffer, data]);

    let start, end;
    // 查找并处理完整的消息
    while ((start = this.buffer.indexOf(START_FLAG)) !== -1 && (end = this.buffer.indexOf(END_FLAG, start + START_FLAG.length)) !== -1) {
      // 提取消息内容
      const messageBuffer = this.buffer.slice(start + START_FLAG.length, end);

      // 处理消息内容 (示例：输出消息)
      console.log("Received message:", messageBuffer.toString());

      // 移除已处理的部分
      this.buffer = this.buffer.slice(end + END_FLAG.length);
    }

    // 记录原始数据以备调试
    fse.appendFileSync("buffer.txt", data.toString("hex") + "\n", "utf8");
  }

  connectToServer(port: number, host: string) {
    this.client.connect(port, host, () => {
      console.log("Connected to server");
    });
  }
}

const tcpClientInstance = new TcpClient();
tcpClientInstance.connectToServer(12345, "127.0.0.1");

代码解析

缓存区：this.buffer 用来缓存接收到的数据。因为在实际网络传输中，数据可能会被分段接收，因此我们需要一个缓存来存储这些片段，直到我们能拼凑出完整的消息。
标志符的检测与消息提取：通过 buffer.indexOf 方法，我们可以找到起始标志符和结束标志符的位置。找到后，我们就能提取出完整的消息内容。
处理并移除已处理的消息：一旦消息被提取出来，我们将其从缓存中移除，以便处理下一条消息。