如何引入外部文件函数至Socket回调并避免socket未定义错误

原本的Socket.io代码在connection回调内部定义操作函数，函数可直接访问回调内的socket变量：

io.on('connection', (socket) => { 
  function login() { 
    socket.name = 'myname';
  }
  // 此处还有更多类似的操作函数
});

为优化代码结构，将内部函数移至外部文件后出现socket is not defined错误：

外部文件（../chat/functions.js）

login = function() { 
  socket.name = 'myname';
}

主文件

io.on('connection', (socket) => {
  require("../chat/functions"); 
  login();
  // 此处还有更多类似函数调用
});

需求：不修改原有函数内部逻辑（不给函数加参数），解决socket未定义的问题。

闭包封装函数组（推荐） #

将外部文件改为返回一个接收socket的工厂函数，内部封装所有操作函数，通过闭包让函数自然共享socket上下文，完全保留原函数逻辑：

修改后的外部文件（../chat/functions.js）

module.exports = function(socket) {
  return {
    login: function() {
      socket.name = 'myname';
    },
    // 其他需要的操作函数都放在这个对象里
    sendMessage: function(content) {
      socket.emit('message', content);
    }
  }
}

主文件调用方式

io.on('connection', (socket) => {
  const chatActions = require("../chat/functions")(socket); 
  chatActions.login();
  // 调用其他函数示例：chatActions.sendMessage('hello');
});

这种方式本质是把原回调内的函数集合迁移到外部闭包中，既实现了代码拆分，又让函数能像原来一样直接访问socket变量，无需修改任何函数内部的逻辑。

不推荐的方案（仅供参考） #

如果强行不想封装对象，也可以在主文件回调内将socket挂载到临时全局变量，但这种方式容易引发并发冲突，仅适用于单连接测试场景：

主文件修改

io.on('connection', (socket) => {
  // 临时挂载socket到全局
  global.__tempSocket = socket;
  require("../chat/functions"); 
  login();
  // 调用后立即删除避免污染
  delete global.__tempSocket;
});

外部文件修改

login = function() { 
  __tempSocket.name = 'myname';
}

此方案在多用户同时连接时会互相覆盖socket变量，生产环境禁止使用。

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