异步函数外部栈上下文：异步函数运行状态检测需求及问题

嘿，我来帮你搞定这个函数运行状态检测的问题！

有时候我们确实需要知道某个特定函数（或是它的子函数）是否正在运行——比如Node.js的domains模块就能处理部分异步操作（不过不确定对async函数的支持情况）。但传统的简单方案在遇到async函数时会直接失效，其实回调式函数原本也存在这个问题，下面给你几个适配不同场景的解决方案：

一、核心思路：主动标记状态 #

同步函数可以直接通过调用栈判断，但异步/async函数因为执行时机的问题，调用栈会被打断，所以我们需要给目标函数做一层包装，主动标记它的运行状态。

1. 适配async/await函数的包装方案 #

// 包装函数：给目标异步函数添加运行状态追踪
function trackAsyncFunction(fn) {
  let isRunning = false;

  // 返回包装后的async函数
  const wrappedFn = async function(...args) {
    isRunning = true;
    try {
      // 执行原函数并返回结果
      return await fn.apply(this, args);
    } finally {
      // 无论成功失败，都重置运行状态
      isRunning = false;
    }
  };

  // 给包装后的函数挂载可访问的状态属性
  Object.defineProperty(wrappedFn, 'isRunning', {
    get: () => isRunning,
    enumerable: true
  });

  return wrappedFn;
}

// 使用示例：包装你的async函数
const trackedDoUpdate = trackAsyncFunction(async function doUpdate() {
  // 这里是你的异步操作逻辑
  await new Promise(resolve => setTimeout(resolve, 1500));
  console.log('更新完成');
});

// 检测函数是否正在运行
function checkDoUpdateStatus() {
  console.log('doUpdate是否在运行：', trackedDoUpdate.isRunning);
}

// 测试调用
checkDoUpdateStatus(); // 输出：false
trackedDoUpdate();
checkDoUpdateStatus(); // 输出：true
setTimeout(() => checkDoUpdateStatus(), 2000); // 输出：false

2. 适配传统回调式异步函数的方案 #

如果你的函数是传统的回调风格，也可以用类似的包装逻辑：

function trackCallbackFunction(fn) {
  let isRunning = false;

  const wrappedFn = function(...args) {
    isRunning = true;
    // 取出原函数的回调参数
    const originalCallback = args.pop();
    // 包装回调函数，完成后重置状态
    const wrappedCallback = (...callbackArgs) => {
      isRunning = false;
      originalCallback.apply(this, callbackArgs);
    };
    args.push(wrappedCallback);
    // 执行原函数
    return fn.apply(this, args);
  };

  // 挂载状态属性
  Object.defineProperty(wrappedFn, 'isRunning', {
    get: () => isRunning,
    enumerable: true
  });

  return wrappedFn;
}

// 使用示例
const trackedCallbackUpdate = trackCallbackFunction(function doUpdate(callback) {
  setTimeout(() => {
    callback(null, '更新成功');
  }, 1000);
});

// 检测状态
console.log('初始状态：', trackedCallbackUpdate.isRunning); // false
trackedCallbackUpdate((err, res) => {
  console.log(res);
  console.log('回调完成后状态：', trackedCallbackUpdate.isRunning); // false
});
console.log('调用后状态：', trackedCallbackUpdate.isRunning); // true

二、进阶：追踪嵌套子函数的执行上下文 #

如果需要检测“当前代码是否处于目标函数的调用链中”（比如子函数里判断自己是不是在doUpdate的执行流程里），可以用Node.js的AsyncLocalStorage（Node.js 14+）来实现上下文追踪：

const { AsyncLocalStorage } = require('async_hooks');

// 创建异步本地存储实例
const updateContext = new AsyncLocalStorage();

// 包装目标函数，注入上下文
function wrapWithContext(fn) {
  return async function(...args) {
    // 运行时注入上下文标记
    return updateContext.run({ isInDoUpdate: true }, () => {
      return fn.apply(this, args);
    });
  };
}

// 在任意位置检测是否处于目标函数的上下文中
function isInDoUpdateFlow() {
  const context = updateContext.getStore();
  return context?.isInDoUpdate || false;
}

// 测试示例
async function subFunction() {
  console.log('subFunction是否在doUpdate流程中：', isInDoUpdateFlow()); // true
}

const trackedDoUpdate = wrapWithContext(async function doUpdate() {
  await subFunction();
});

trackedDoUpdate();
console.log('外部是否在doUpdate流程中：', isInDoUpdateFlow()); // false

三、注意事项 #

• 所有调用目标函数的地方必须使用包装后的版本，否则状态追踪会失效；
• 如果目标函数有多个实例，要注意状态的隔离，避免不同实例的状态互相干扰；
• AsyncLocalStorage是Node.js的内置API，浏览器环境可以用类似的库（如zone.js）来实现类似功能。

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