Node.js与Socket.io：如何在其他文件中访问Socket实例

咱们先拆解下核心问题：你的DoFuncOnSecondaryFile函数定义在io.on('connection')回调外部，自然拿不到回调内部的socket变量；如果是多客户端场景，还要考虑如何管理多个在线socket实例。下面给你几个实用的解决方案：

方案一：直接传递Socket实例（单客户端/指定客户端场景） #

如果你的应用只需要针对当前连接的客户端执行定时任务，只需要把socket作为参数传入定时函数即可：

修改main.js：

var sec = require('./secondary.js');

// 新增socket参数，接收外部传入的连接实例
function DoFuncOnSecondaryFile(socket){
    setInterval(function(){
        sec.myExternalFunc(socket);
    }, 5000);
}

io.on('connection', function(socket){
    // 客户端连接时，把当前socket传入启动定时任务
    DoFuncOnSecondaryFile(socket);

    // 所有socket事件处理逻辑在此处
});

同步修改secondary.js，接收并使用传入的socket：

module.exports = {
    myExternalFunc: function(socket){
        socket.emit("message", "I can send messages now!");
    }
}

方案二：维护Socket实例集合（多客户端广播场景） #

如果需要给所有在线客户端定时发送消息，建议维护一个存储在线socket的集合：

修改main.js：

var sec = require('./secondary.js');
// 用Set存储在线socket，自动去重且便于增删
const connectedSockets = new Set();

function DoFuncOnSecondaryFile(){
    setInterval(function(){
        // 遍历所有在线socket，调用次级文件的方法
        connectedSockets.forEach(socket => {
            sec.myExternalFunc(socket);
        });
    }, 5000);
}

// 服务器启动时就启动定时任务
DoFuncOnSecondaryFile();

io.on('connection', function(socket){
    // 客户端连接时加入集合
    connectedSockets.add(socket);

    // 客户端断开时从集合移除，避免内存泄漏
    socket.on('disconnect', () => {
        connectedSockets.delete(socket);
    });

    // 所有socket事件处理逻辑在此处
});

secondary.js的修改和方案一致，保持接收socket参数即可：

module.exports = {
    myExternalFunc: function(socket){
        socket.emit("message", "Hello from secondary file!");
    }
}

方案三：将Socket集合注入次级模块（高封装性场景） #

如果希望次级模块的逻辑更独立，可以把socket集合注入到次级模块中，避免每次调用都传参数：

修改main.js：

var sec = require('./secondary.js');
const connectedSockets = new Set();

// 把socket集合传递给次级模块
sec.setConnectedSockets(connectedSockets);

function DoFuncOnSecondaryFile(){
    setInterval(function(){
        // 直接调用次级模块的广播方法，无需传参
        sec.broadcastMessage();
    }, 5000);
}

DoFuncOnSecondaryFile();

io.on('connection', function(socket){
    connectedSockets.add(socket);
    socket.on('disconnect', () => {
        connectedSockets.delete(socket);
    });

    // 所有socket事件处理逻辑在此处
});

修改secondary.js，添加注入方法和封装的广播逻辑：

let connectedSockets = new Set();

module.exports = {
    // 接收外部传入的socket集合
    setConnectedSockets: function(sockets){
        connectedSockets = sockets;
    },
    // 封装广播逻辑，内部直接使用集合
    broadcastMessage: function(){
        connectedSockets.forEach(socket => {
            socket.emit("message", "Broadcast from secondary module!");
        });
    }
}

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