async_hook_context_store.js

const asyncHooks = require('async_hooks');

const contextStore = new Map();

class Context {
  constructor() {
    this.id = Math.random();
  }
}

const asyncHook = asyncHooks.createHook({
  init(asyncId, type, triggerAsyncId, resource) {
    const parentContext = contextStore.get(triggerAsyncId);
    if (parentContext) {
      contextStore.set(asyncId, parentContext);
    }
  },
  destroy(asyncId) {
    contextStore.delete(asyncId);
  }
});
asyncHook.enable();

function setContext() {
  contextStore.set(asyncHooks.executionAsyncId(), new Context());
}

function getContext() {
  return contextStore.get(asyncHooks.executionAsyncId());
}