如何在Node.js不同函数中访问Socket.io的WebSocket连接？

你的核心问题是user中间件对象的sendMessage方法无法访问Socket.io的io实例，导致无法触发socket事件。以下是几种实用解决方案：

方案一：直接将io实例注入到user中间件对象 #

这是最直接的实现方式，在服务器初始化时把io传递给user对象，让它持有实例引用。

修改中间件文件 #

const userSocket = (io) => {
    io.on("connection", (socket) => {
        socket.on("join-room", (id) => {
            socket.join(id);
            console.log(`user joined room ${id}`);
        });
    });
};

// 定义user对象并预留io属性
const user = {
    io: null,
    sendMessage(roomId, message) {
        // 给指定房间发送消息，全局发送用this.io.emit("message", message)
        this.io.to(roomId).emit("message", message);
    }
};

module.exports.userSocket = userSocket;
module.exports.user = user;

修改服务器文件 #

const express = require("express");
const app = express();
const { createServer } = require("http");
const httpServer = createServer(app);

const port = 3000;
const io = require("socket.io")(httpServer, { cors: { origin: "*" } });

const { userSocket, user } = require("./resources/users/user");
userSocket(io);
// 将io实例注入到user对象
user.io = io;

httpServer.listen(port);
console.log(`Listening on port ${port}`);

使用时直接调用user.sendMessage("room123", "hello")即可触发socket事件。

方案二：创建独立的Socket.io实例共享模块 #

如果多个中间件都需要访问io，可以抽离单独模块统一管理实例：

新建socket.js文件 #

let ioInstance = null;

module.exports = {
    setIo(io) {
        ioInstance = io;
    },
    getIo() {
        if (!ioInstance) {
            throw new Error("Socket.io instance not initialized");
        }
        return ioInstance;
    }
};

修改中间件文件 #

const socketManager = require("../socket");

const userSocket = (io) => {
    // 将io实例存入共享模块
    socketManager.setIo(io);
    io.on("connection", (socket) => {
        socket.on("join-room", (id) => {
            socket.join(id);
            console.log(`user joined room ${id}`);
        });
    });
};

const user = {
    sendMessage(roomId, message) {
        const io = socketManager.getIo();
        io.to(roomId).emit("message", message);
    }
};

module.exports.userSocket = userSocket;
module.exports.user = user;

服务器文件无需额外修改，userSocket会自动完成实例的共享注册。

方案三：关联单个socket与user中间件（针对特定用户发送） #

如果需要给特定用户发送消息，可以在connection事件中存储socket实例：

修改中间件文件 #

const userSocket = (io) => {
    io.on("connection", (socket) => {
        socket.on("join-room", (id) => {
            socket.join(id);
            console.log(`user joined room ${id}`);
        });
        // 用Map存储所有在线socket
        user.sockets.set(socket.id, socket);
    });
};

const user = {
    sockets: new Map(),
    // 给指定socket用户发消息
    sendMessageToUser(socketId, message) {
        const socket = this.sockets.get(socketId);
        if (socket) socket.emit("message", message);
    },
    // 给指定房间发消息（需结合方案一注入io）
    sendMessageToRoom(roomId, message) {
        this.io.to(roomId).emit("message", message);
    }
};

module.exports.userSocket = userSocket;
module.exports.user = user;

服务器文件仍需注入io到user对象，使用时通过用户的socketId调用对应方法即可。

补充说明 #

你担心的“访问io是否可行”完全多余：userSocket只是复用同一个io实例监听连接事件，并没有创建新的Socket服务，只要是同一个io实例，调用io.emit或io.to(room).emit都是有效的。

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