global exposed.

Author: odeadglaz

package.json

@@ -1,6 +1,6 @@
 {
   "name": "node-execution-context",
-  "version": "1.0.5",
+  "version": "1.1.0",
   "description": "Provides execution context wrapper for node JS, can be used to create execution wrapper for handling requests and more",
   "author": "Oded Goldglas <[email protected]>",
   "license": "ISC",

src/index.js

@@ -3,17 +3,6 @@ const { isProduction } = require('./lib');
 const { create: createHooks } = require('./hooks');
 const { ExecutionContextErrors } = require('./constants');
 
-/**
- * The global service context execution map
- * @type ExecutionContextMap
- */
-const executionContextMap = new Map();
-
-// Sets node async hooks setup
-asyncHooks.createHook(
-    createHooks(executionContextMap)
-).enable();
-
 /**
  * Handles execution context error, throws when none production
  * @param code
@@ -28,65 +17,82 @@ const handleError = (code) => {
 
 /**
  * The Execution Context API
+ * @return {ExecutionContextAPI}
  */
-const Context = {
-
-    /**
-     * Creates an execution context for the current asyncId process.
-     * This will expose Context get / update at any point after.
-     * @param {Object} initialContext - The initial context to be used
-     * @returns void
-     */
-    create: (initialContext = {}) => {
-        const asyncId = asyncHooks.executionAsyncId();
-
-        // Creation is allowed once per execution context
-        if (executionContextMap.has(asyncId)) handleError(ExecutionContextErrors.CONTEXT_ALREADY_DECLARED);
-
-        executionContextMap.set(asyncId, {
-            context: { ...initialContext, executionId: asyncId },
-            children: []
-        });
-    },
-
-    /**
-     * Updates the current async process context
-     * @param {Object} update - The update to apply on the current process context
-     * @returns void
-     */
-    update: (update = {}) => {
-        const asyncId = asyncHooks.executionAsyncId();
-
-        if (!executionContextMap.has(asyncId)) handleError(ExecutionContextErrors.CONTEXT_DOES_NOT_EXISTS);
-
-        const contextData = executionContextMap.get(asyncId);
-
-        // Update target is always the root context, ref updates will need to be channeled
-        const targetContextData = contextData.ref
-            ? executionContextMap.get(contextData.ref)
-            : contextData;
-
-        targetContextData.context = { ...targetContextData.context, ...update };
-    },
+const createExecutionContext = () => {
 
     /**
-     * Gets the current async process execution context
-     * @returns {Object}
+     * The global service context execution map
+     * @type ExecutionContextMap
      */
-    get: () => {
-        const asyncId = asyncHooks.executionAsyncId();
-        if (!executionContextMap.has(asyncId)) handleError(ExecutionContextErrors.CONTEXT_DOES_NOT_EXISTS);
-
-        const { context = {}, ref } = executionContextMap.get(asyncId);
-        if (ref) {
-
-            // Ref will be used to point out on the root context
-            return executionContextMap.get(ref).context;
+    const executionContextMap = new Map();
+
+    // Sets node async hooks setup
+    asyncHooks.createHook(
+        createHooks(executionContextMap)
+    ).enable();
+
+    return {
+
+        /**
+         * Creates an execution context for the current asyncId process.
+         * This will expose Context get / update at any point after.
+         * @param {Object} initialContext - The initial context to be used
+         * @returns void
+         */
+        create: (initialContext = {}) => {
+            const asyncId = asyncHooks.executionAsyncId();
+
+            // Creation is allowed once per execution context
+            if (executionContextMap.has(asyncId)) handleError(ExecutionContextErrors.CONTEXT_ALREADY_DECLARED);
+
+            executionContextMap.set(asyncId, {
+                context: { ...initialContext, executionId: asyncId },
+                children: []
+            });
+        },
+
+        /**
+         * Updates the current async process context.
+         * @param {Object} update - The update to apply on the current process context.
+         * @returns void
+         */
+        update: (update = {}) => {
+            const asyncId = asyncHooks.executionAsyncId();
+
+            if (!executionContextMap.has(asyncId)) handleError(ExecutionContextErrors.CONTEXT_DOES_NOT_EXISTS);
+
+            const contextData = executionContextMap.get(asyncId);
+
+            // Update target is always the root context, ref updates will need to be channeled
+            const targetContextData = contextData.ref
+                ? executionContextMap.get(contextData.ref)
+                : contextData;
+
+            targetContextData.context = { ...targetContextData.context, ...update };
+        },
+
+        /**
+         * Gets the current async process execution context.
+         * @returns {Object}
+         */
+        get: () => {
+            const asyncId = asyncHooks.executionAsyncId();
+            if (!executionContextMap.has(asyncId)) handleError(ExecutionContextErrors.CONTEXT_DOES_NOT_EXISTS);
+
+            const { context = {}, ref } = executionContextMap.get(asyncId);
+            if (ref) {
+
+                // Ref will be used to point out on the root context
+                return executionContextMap.get(ref).context;
+            }
+
+            // Root context
+            return context;
         }
+    };
+}
 
-        // Root context
-        return context;
-    }
-};
+global.ExecutionContext = global.ExecutionContext || createExecutionContext();
 
-module.exports = Context;
+module.exports = global.ExecutionContext;

src/types.d.ts

@@ -30,3 +30,30 @@ interface HookCallbacks {
      */
     destroy?(asyncId: number): void;
 }
+
+interface ExecutionContextAPI {
+
+    /**
+     * Creates an execution context for the current asyncId process.
+     * @param initialContext
+     */
+    create(initialContext: object): void;
+
+    /**
+     * Updates the current async process context.
+     * @param update
+     */
+    update(update: object): void;
+
+    /**
+     * Gets the current async process execution context.
+     */
+    get(): object
+
+    /**
+     * Runs a given function within an async resource context
+     * @param fn
+     * @param initialContext
+     */
+    run(fn: Function, initialContext: object): void
+}