Nodejs websocket 模块 ws 有用过的朋友讨论下

Node.js Websocket 模块 ws 讨论

GitHub 地址

https://github.com/einaros/ws

WebSocket 模块 ws 使用讨论

在使用 Node.js 的 WebSocket 模块 ws 进行开发时，有时会遇到内存泄漏的问题。以下是一个简单的示例，展示了如何使用 ws 模块，并且如何通过一些手段来监控内存使用情况。

服务端代码

首先，我们来看一下服务端的代码：

var WebSocketServer = require('ws').Server;
var m = require("memwatch");
var wss = new WebSocketServer({ port: 8101 });

// 监控内存使用情况
setInterval(function () {
    var mem = process.memoryUsage();
    console.log(`clients: ${wss.clients.length} rss:`, Math.round((mem.rss / 1024 / 1024)) + "MB");
}, 5000);

wss.on('connection', function (ws) {
    // 当连接建立时
    ws.on('close', function (err) {
        console.log('Client disconnected');
    });
});

这段代码中，我们创建了一个 WebSocket 服务器，并且每 5 秒输出一次当前的内存使用情况。这样可以方便地监控内存使用情况，从而发现是否存在内存泄漏的问题。

客户端代码

接下来，我们看一下客户端的代码：

var WebSocket = require('ws');

let count = 0;

function run() {
    if (count > 1000) return;

    var ws = new WebSocket('ws://localhost:8101');

    ws.on('open', function () {
        count++;
        console.log('count=' + count);

        setInterval(function () {
            let online = { uid: count };
            ws.send(JSON.stringify(online), function (err) {
                if (err) console.log(err);
                else console.log('send success');
            });
        }, 1000); // 每秒发送一次消息

        run(); // 递归调用，不断创建连接
    });

    ws.on('message', function (data, flags) {
        console.log('Received:', data);
    });
}

run();

在这段代码中，客户端不断地尝试连接到服务器，并且每秒向服务器发送一条消息。然而，如果连接不断被创建而没有正确关闭，这会导致内存泄漏。

解决方案

为了解决内存泄漏的问题，我们需要确保每次连接断开时都能正确地清理资源。可以通过以下方式改进客户端代码：

function run() {
    if (count > 1000) return;

    var ws = new WebSocket('ws://localhost:8101');

    ws.on('open', function () {
        count++;
        console.log('count=' + count);

        setInterval(function () {
            let online = { uid: count };
            ws.send(JSON.stringify(online), function (err) {
                if (err) console.log(err);
                else console.log('send success');
            });
        }, 1000); // 每秒发送一次消息
    });

    ws.on('close', function () {
        console.log('Client disconnected');
        ws = null; // 清理资源
    });

    ws.on('message', function (data, flags) {
        console.log('Received:', data);
    });
}

run();

通过在 ws 对象上设置 ws = null，我们可以确保当连接断开时，相关的资源能够被正确地释放。

希望这些示例代码能帮助你更好地理解和使用 ws 模块，并解决可能遇到的内存泄漏问题。