如何在Node.js TCP socket监听器中唯一识别socket id

• Node.js 原生net模块的TCP socket对象没有内置id属性，该属性是socket.io等WebSocket框架提供的封装能力，你直接调用socket.id自然无法获取值。
• 现有代码逻辑存在严重错误：你在data事件回调中遍历连接数组时，只要数组中存在任意一个元素与当前socket不匹配，就会触发新元素插入逻辑，同一个socket每次上报数据都会被多次插入数组，自然会生成大量重复的无效id。
• 连接信息初始化逻辑位置错误：按照你的需求，应该在新连接建立的connection事件中完成id生成、数据库查询、信息存储操作，而不是放到每次数据上报的data事件中处理。

const Net = require('net');
// 用Map存储连接信息比数组查询效率更高，key直接用自定义的socket id
const socketMap = new Map();

const server = new Net.Server();
const HOST = "你的服务IP";
const PORT =  8888;

server.listen(PORT, HOST, () => {
    console.log(`Socket服务已启动，监听${HOST}:${PORT}`);
});

// 连接建立时一次性处理所有初始化逻辑
server.on('connection', function(socket) {
    console.log('新客户端连接建立');
    // 自定义socket唯一id，也可以替换成自增数字、uuid等规则，避免随机数重复
    const socketId = Date.now() + Math.floor(Math.random() * 1000);
    // 把id挂载到socket对象上，后续该连接的所有事件回调里都能直接取
    socket.id = socketId;

    // 这里放置你的数据库查询逻辑，仅在连接建立时执行一次
    // 示例：const deviceInfo = await db.query('select * from device where 设备标识=xxx')
    const deviceInfo = {}; // 替换成你实际查询到的设备信息

    // 存入Map，后续查询的时间复杂度为O(1)，比数组遍历效率高很多
    socketMap.set(socketId, {
        socket: socket,
        deviceInfo: deviceInfo,
        createTime: Date.now()
    });

    // 数据上报事件，直接使用已存储的信息
    socket.on('data', function(chunk) {
        const currentConn = socketMap.get(socket.id);
        if(currentConn) {
            // 直接使用已缓存的设备信息，不需要重复查询数据库
            console.log(`当前连接id：${socket.id}，设备信息：`, currentConn.deviceInfo, `上报数据：${chunk.toString()}`);
            // 这里写你处理上报数据的业务逻辑
        }
    });

    // 连接关闭时从Map删除对应数据，避免内存泄漏
    socket.on('end', function() {
        socketMap.delete(socket.id);
        console.log(`客户端连接关闭，id：${socket.id}`);
    });

    // 错误处理
    socket.on('error', function(err) {
        socketMap.delete(socket.id);
        console.log(`连接出错，id：${socket.id}，错误信息：${err}`);
    });
});

• 用Map替代数组存储连接信息，查询和删除的时间复杂度都是O(1)，尤其在连接数量大的场景下性能优势明显。
• 连接建立阶段就完成id生成、数据库查询、信息存储全流程，完全符合你“仅新连接建立时查一次数据库”的需求，不会每次数据上报重复查询。
• 连接关闭/出错时主动删除存储的连接信息，避免内存泄漏。
• 自定义id直接挂载到socket对象上，在该连接的所有事件回调中都可以直接通过socket.id获取，不需要额外遍历查询。

内容的提问来源于stack exchange，提问作者mynameisbutt