Nodejs 能否探测是不是脚本从 require 运行的?

Nodejs 能否探测是不是脚本从 require 运行的?

脚本如果能探测是不是直接运行的话, 一些测试的函数可以默认隐藏了…

6 回复

当然可以！在 Node.js 中，你可以通过检查当前模块的 module.parent 属性来判断一个脚本是否是通过 require 运行的。如果模块是直接运行的，那么 module.parent 将为 null；如果模块是通过 require 引入的，则 module.parent 会指向引入该模块的模块。

以下是一个简单的示例代码来演示如何实现这一点：

// main.js
const myModule = require('./myModule');

console.log(myModule.checkIfRequired());

// myModule.js
function checkIfRequired() {
    if (module.parent) {
        return "This script is running via 'require'.";
    } else {
        return "This script is running directly.";
    }
}

module.exports = {
    checkIfRequired: checkIfRequired
};

在这个例子中，我们创建了两个文件：main.js 和 myModule.js。myModule.js 包含一个函数 checkIfRequired，该函数根据 module.parent 是否存在来返回不同的消息。

当我们在 main.js 中通过 require 引入 myModule.js 并调用 checkIfRequired 函数时，myModule.js 会检测到它不是直接运行的，并返回相应的消息。

示例输出

假设你运行 node main.js，你会看到以下输出：

This script is running via 'require'.

而如果你直接运行 myModule.js（例如使用 node myModule.js），你会看到：

This script is running directly.

通过这种方式，你可以在 Node.js 中轻松地判断一个脚本是通过 require 引入的还是直接运行的。这样你就可以根据需要决定是否隐藏某些测试函数或执行特定逻辑。


求助了 StackOverflow… 答案是检测 module.parent 是否存在 http://stackoverflow.com/questions/12283213/how-can-my-node-script-detect-whether-it-is-being-called-by-require

受教了

有用的信息

受教

当然可以。你可以通过检查当前模块的调用栈来判断脚本是否是通过 require 加载的，还是直接执行的。在 Node.js 中，当一个模块被 require 加载时，其调用栈会有所不同。

下面是一个简单的示例代码，用于检测当前模块是否是通过 require 加载的：

function isRequired() {
    const originalPrepareStackTrace = Error.prepareStackTrace;
    Error.prepareStackTrace = (_, stack) => stack;

    const currentStack = new Error().stack;
    Error.prepareStackTrace = originalPrepareStackTrace;

    // 检查调用栈中是否有 'module.js'，表示该模块是通过 require 加载的
    return currentStack.some(frame => frame.getFileName().includes('module.js'));
}

if (isRequired()) {
    console.log("This script is being required.");
} else {
    console.log("This script is running directly.");
}

解释

  1. Error.prepareStackTrace: 这个方法允许我们自定义错误堆栈跟踪的格式。
  2. new Error().stack: 创建一个新的错误对象并获取其堆栈跟踪信息。
  3. 检查堆栈跟踪: 我们检查堆栈跟踪中的文件名，如果包含 module.js，则说明该模块是通过 require 加载的。

这样你就可以根据不同的情况执行不同的逻辑，例如隐藏一些测试函数或进行其他处理。

回到顶部