NodeJS中如何区分并处理带参与无参的模块require调用？

你的核心问题在于：Node.js模块在被require时会立即执行所有顶层代码，但你需要区分两种场景——用户只是加载模块（require('my-module')），还是加载后立即调用导出的函数（require('my-module')('red')）。之前的写法要么会重复执行逻辑，要么完全不触发默认行为，这里给你一个可靠的解决方案：

解决方案思路 #

利用Node.js的事件循环机制，在模块加载完成后延迟判断导出函数是否被调用过。如果没被调用，说明用户只是加载了模块，此时执行默认的打印逻辑；如果被调用过，就只执行带参数的目标逻辑。

完整代码实现 #

修改你的my-module.js如下：

let wasFunctionCalled = false;

// 颜色处理函数，用ANSI转义码实现控制台彩色输出
function displayMessage(clr) {
  // 定义颜色映射表，可根据需求扩展更多颜色
  const colorMap = {
    red: '\x1b[31m',
    green: '\x1b[32m',
    blue: '\x1b[34m',
    white: '\x1b[37m',
    reset: '\x1b[0m' // 重置颜色，避免影响后续控制台输出
  };
  // 若未传入有效颜色，默认使用白色
  const targetColor = colorMap[clr] || colorMap.white;
  console.log(`${targetColor}Hello World${colorMap.reset}`);
}

// 导出带参数的函数，调用时标记已执行
module.exports = function(color) {
  wasFunctionCalled = true;
  displayMessage(color);
};

// 利用setImmediate在模块加载完成后判断是否需要执行默认逻辑
setImmediate(() => {
  if (!wasFunctionCalled) {
    displayMessage(''); // 触发默认白色打印
  }
});

为什么这个方案有效？ #

• 当用户执行require('my-module')('red')时：
  1. 模块先加载完成，导出函数；
  2. 立即调用导出的函数，wasFunctionCalled被设为true；
  3. 进入事件循环的下一轮，setImmediate的回调执行时，检测到wasFunctionCalled为true，跳过默认逻辑。
• 当用户只执行require('my-module')时：
  1. 模块加载完成，导出函数但未被调用；
  2. 事件循环下一轮的setImmediate回调执行，检测到wasFunctionCalled仍为false，触发默认的白色打印。

额外说明 #

• ANSI转义码是控制台彩色输出的标准方式，几乎所有终端都支持；
• setImmediate比setTimeout(fn, 0)更可靠，因为它专门用于在当前事件循环结束后立即执行回调，不会有不必要的延迟；
• 如果你需要支持更多颜色，只需要在colorMap中添加对应的转义码即可。

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