async-stack-tagging-api-v2.ts

// @ts-ignore
const scheduleAsyncTask = console.scheduleAsyncTask ?? (() => {});
// @ts-ignore
const startAsyncTask = console.startAsyncTask ?? (() => {});
// @ts-ignore
const finishAsyncTask = console.finishAsyncTask ?? (() => {});

interface ConsoleTask {
  run<T>(f: () => T): T;
}

function scheduleTask(name: string): ConsoleTask {
  const id = scheduleAsyncTask(name, true);
  return {
    run(f) {
      try {
        startAsyncTask(id);
        const value = f();
        finishAsyncTask(id);
        return value;
      } catch (e) {
        finishAsyncTask(id);
        throw e;
      }
    },
  };
}