基于Nodejs的长连接socket服务器

基于Node.js的长连接Socket服务器

在开发某些应用时，如Flash游戏或服务器间通信，我们需要实现一种特定的协议，其中每个消息都包含一个4字节的头部，用于指示后续数据包的长度。在Node.js中，我们可以使用head_body_buffers模块来简化这一过程。

示例代码

首先，你需要安装head_body_buffers模块：

npm install head_body_buffers

接下来是服务器端代码：

const net = require('net');
const HeadBodyBuffers = require('./lib/head_body_buffers').HeadBodyBuffers;

// 计算数据包长度的函数
function packetLength(data) {
    return data.readUInt32BE(0);
}

// 创建TCP服务器
const server = net.createServer((connection) => {
    console.log('New socket connected');

    // 设置超时时间
    connection.setTimeout(2 * 60 * 1000);
    connection.setNoDelay(true);

    // 初始化HeadBodyBuffers实例
    const hdb = new HeadBodyBuffers(4, packetLength);

    // 监听packet事件
    hdb.on('packet', (packet) => {
        const head = packet.slice(0, 4);
        const body = packet.slice(4);
        console.log(`Received packet: ${body.toString()} with length ${body.length}`);
        connection.write(packet); // 回复接收到的数据包
    });

    // 监听其他事件
    connection.on('end', () => {
        console.log(`Socket ${connection.remoteAddress}:${connection.remotePort} closed`);
    });

    connection.on('data', (data) => {
        hdb.addBuffer(data);
    });

    connection.on('timeout', () => {
        connection.end();
    });

    connection.on('error', (error) => {
        console.error(error);
        connection.end();
    });
});

// 启动服务器
server.listen(11000, () => {
    console.log('Server started on port 11000');
});

// 处理服务器错误
server.on('error', (error) => {
    console.error(error);
});

客户端代码

接下来是客户端代码：

const net = require('net');
const HeadBodyBuffers = require('./lib/head_body_buffers').HeadBodyBuffers;

// 计算数据包长度的函数
function packetLength(data) {
    return data.readUInt32BE(0);
}

// 创建客户端连接
const client = net.connect(11000, () => {
    console.log('Connected to server');
});

// 初始化HeadBodyBuffers实例
const hdb = new HeadBodyBuffers(4, packetLength);

// 监听packet事件
hdb.on('packet', (packet) => {
    const head = packet.slice(0, 4);
    const body = packet.slice(4);
    console.log(`Received packet from server: ${body.toString()} with length ${body.length}`);
    client.write(packet); // 回复接收到的数据包
});

// 发送保持连接的消息
const keepAlive = '{type:"keepAlive"}';
const packet = Buffer.alloc(4 + Buffer.byteLength(keepAlive));
packet.writeUInt32BE(Buffer.byteLength(keepAlive), 0);
packet.write(keepAlive, 4);
client.write(packet);

// 监听数据接收事件
client.on('data', (data) => {
    hdb.addBuffer(data);
});

解释

• 服务器端：

  • 使用net.createServer创建一个TCP服务器。
  • HeadBodyBuffers用于处理分包逻辑，通过packetLength函数计算包体长度。
  • hdb.on('packet', ...)监听完整数据包，并打印其内容。
  • connection.on('data', ...)将接收到的数据传递给hdb处理。
  • 设置超时和错误处理机制。

• 客户端：

  • 使用net.connect连接到服务器。
  • 发送一条保持连接的消息。
  • 接收并处理来自服务器的数据包。

这种方式可以确保在长连接过程中正确地处理数据包的分隔和重组。