Run support.

Author: odeadglaz

README.md

@@ -22,13 +22,7 @@ const app = express();
 const port = 3000;
 
 const ContextMiddleware = (req, res, next) => {
-	const requestResource = new AsyncResource('REQUEST_CONTEXT');
-	requestResource.runInAsyncScope(() => {
-		Context.create({
-			val: true
-		});
-		next();
-	});
+    Context.run(next, { val: true });
 };
 
 app.use('/', ContextMiddleware);

example/middleware.js

@@ -1,12 +1,5 @@
-const {  AsyncResource  } = require('async_hooks');
 const Context = require('../src');
 
 module.exports = (req, res, next) => {
-	const asyncResource = new AsyncResource('REQUEST_CONTEXT');
-	return asyncResource.runInAsyncScope(() => {
-		Context.create({
-			val: true
-		});
-		next();
-	});
+	Context.run(next, { val: true });
 };

src/hooks/index.js

@@ -11,7 +11,7 @@ const getContextRef = (parentContext, triggerAsyncId) => (
 );
 
 /**
- * Suspends a given function execution over process next tick.
+ * Suspends a given f   unction execution over process next tick.
  * @param {Function} fn - The function to trigger upon next tick.
  * @param {...any} args - The function arguments to trigger with.
  * @return {any}

src/index.js

@@ -1,4 +1,5 @@
 const asyncHooks = require('async_hooks');
+const ExecutionContextResource = require('./lib/ExecutionContextResource')
 const { isProduction } = require('./lib');
 const { create: createHooks } = require('./hooks');
 const { ExecutionContextErrors } = require('./constants');
@@ -32,7 +33,7 @@ const createExecutionContext = () => {
         createHooks(executionContextMap)
     ).enable();
 
-    return {
+    const Context = {
 
         /**
          * Creates an execution context for the current asyncId process.
@@ -41,7 +42,7 @@ const createExecutionContext = () => {
          * @returns void
          */
         create: (initialContext = {}) => {
-            console.log('Creating execution context : ', executionContextMap);
+            console.log('Ceating: ', executionContextMap);
             const asyncId = asyncHooks.executionAsyncId();
 
             // Creation is allowed once per execution context
@@ -90,8 +91,25 @@ const createExecutionContext = () => {
 
             // Root context
             return context;
+        },
+
+        /**
+         * Runs a given function within "AsyncResource" context, this will ensure the function executed within a uniq execution context.
+         * @param {Function} fn - The function to run.
+         * @param {Object} initialContext - The initial context to expose to the function execution
+         */
+        run: (fn, initialContext) => {
+            const resource = new ExecutionContextResource();
+
+            resource.runInAsyncScope(() => {
+                Context.create(initialContext);
+
+                fn();
+            });
         }
     };
+
+    return Context;
 }
 
 global.ExecutionContext = global.ExecutionContext || createExecutionContext();

src/lib/ExecutionContextResource.js

@@ -0,0 +1,11 @@
+const { AsyncResource } = require('async_hooks');
+
+const ASYNC_RESOURCE_TYPE = 'REQUEST_CONTEXT';
+
+class ExecutionContextResource extends AsyncResource {
+    constructor() {
+        super(ASYNC_RESOURCE_TYPE);
+    }
+}
+
+module.exports = ExecutionContextResource;