Socket.IO连接后在REST API中操作Socket的问题求助

核心问题根源 #

你当前的实现存在两个致命逻辑错误：

1. 单个socket存储覆盖问题：用app.set('socket', socket)存储的永远是最后一个建立连接的用户socket，而非当前请求用户的socket，导致无法对应用户加入房间。
2. socket实例错误导致推送逻辑失效：因为使用了错误的socket实例调用to()，排除的是最后一个连接的用户而非消息发送者本身，所以发送者仍会收到推送。

解决方案步骤 #

1. 建立Session与Socket的映射关系 #

创建一个全局映射表，将用户的sessionID与对应的socket实例绑定，确保能通过REST请求的sessionID找到对应用户的所有socket连接。

修改app.js：

// 导入依赖（根据你的session配置调整，这里以express-session为例）
const cookieParser = require('cookie-parser');

// 全局存储sessionID到socket的映射（支持多标签页连接）
const sessionSocketMap = new Map();

io.on('connection', (socket) => {
    // 从socket握手的cookie中解析sessionID
    const cookies = cookieParser.parse(socket.handshake.headers.cookie || '');
    const sessionID = cookies['connect.sid']; // 这里的key需与你的session配置一致

    if (sessionID) {
        // 处理用户多连接场景，用数组存储同一个session的所有socket
        if (sessionSocketMap.has(sessionID)) {
            sessionSocketMap.get(sessionID).push(socket);
        } else {
            sessionSocketMap.set(sessionID, [socket]);
        }
    }

    // 断开连接时清理映射，避免内存泄漏
    socket.on('disconnect', () => {
        if (sessionID) {
            const sockets = sessionSocketMap.get(sessionID);
            if (sockets) {
                const index = sockets.indexOf(socket);
                if (index !== -1) {
                    sockets.splice(index, 1);
                    if (sockets.length === 0) {
                        sessionSocketMap.delete(sessionID);
                    }
                }
            }
        }
        socket.disconnect(true);
    });
});

// 将映射表存入app，供后续模块调用
app.set('sessionSocketMap', sessionSocketMap);

2. 调整控制器传递参数 #

将映射表传递给服务类，替代原来的单个socket实例：

const controller = (ServiceClass, params) => {
    return async (req, res, next) => {
        const service = new ServiceClass({
            session: req.session,
            sessionId: req.sessionID,
            cookies: req.cookies,
            socketIo: req.app.get('socketio'),
            sessionSocketMap: req.app.get('sessionSocketMap'), // 传递映射表
        });

        const response = await service.run(params(req));
        res.json(response);
    };
}

export default controller;

3. 修正用户加入房间的逻辑 #

在IsLoggedService中通过sessionID找到对应用户的socket，加入目标房间：

export default class IsLoggedService extends Service {
    constructor(context) {
        super(context);
    }

    async execute() {
        const { sessionId, session, sessionSocketMap } = this.context;
        const userSockets = sessionSocketMap.get(sessionId);
        
        if (userSockets && userSockets.length > 0) {
            // 将用户的所有socket连接加入对应房间
            await Promise.all(userSockets.map(socket => socket.join(`room${session.roleId}`)));
        } else {
            console.warn('未找到当前用户的有效socket连接');
        }
    }
}

4. 修正消息推送逻辑 #

使用发送者自己的socket调用to()，确保排除发送者：

// 新增任务的服务类中的推送代码
async pushTaskNotification() {
    const { sessionId, session, sessionSocketMap } = this.context;
    const userSockets = sessionSocketMap.get(sessionId);
    const targetRoom = `room${session.roleId}`;

    if (userSockets && userSockets.length > 0) {
        // 对每个socket连接执行推送，确保排除当前连接的用户
        userSockets.forEach(socket => {
            socket.to(targetRoom).emit('taskCreated', {
                message: '新任务已创建',
                taskId: '任务ID示例'
            });
        });
    } else {
        console.warn('未找到当前用户的有效socket连接');
    }
}

关键说明 #

• 多连接场景处理：映射表用数组存储同一个用户的多个socket，确保用户打开多标签页时，所有连接都能加入房间并收到推送。
• sessionID获取适配：如果你的项目使用token鉴权而非cookie，可让前端在socket连接时通过query参数传递token，后端解析token后关联用户ID与socket。

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