feat(context): remove namespace destroy from middleware and enable setting values on context

Author: sonicoder86

index.js

@@ -21,6 +21,7 @@ logFactory.Timer = Timer;
 logFactory.getNamespaces = function() {
   return process.env.DEBUG || '';
 };
-logFactory.getMiddleware = contextMiddlewareFactory.getMiddleware;
+logFactory.getMiddleware = contextMiddlewareFactory.getMiddleware.bind(contextMiddlewareFactory);
+logFactory.setOnContext = contextMiddlewareFactory.setOnContext.bind(contextMiddlewareFactory);
 
 module.exports = logFactory;

koa-example.js

@@ -1,21 +1,28 @@
 'use strict';
 
 const Koa = require('koa');
+const uuid = require('uuid');
 const logFactory = require('./index');
 const logger = logFactory('example');
 const port = 3000;
 
 const app = new Koa();
 
 app.use(async (ctx, next) => {
-  ctx.request.header['x-request-id'] = 'uuid';
+  ctx.request.header['x-request-id'] = uuid.v4();
   await next();
 });
+
 app.use(logFactory.getMiddleware());
+
 app.use(async (ctx) => {
-  logger.info('works');
+  logger.info('before');
+
+  logFactory.setOnContext('customer_id', Math.round(Math.random() * 1000));
+
+  logger.info('after');
   ctx.body = 'It works';
 });
 
 app.listen(port);
-console.log('listening on port: ' + port);
+console.log('listening on port: ' + port);

package-lock.json

@@ -37,9 +37,11 @@
     "json"
   ],
   "dependencies": {
-    "chalk": "2.1.0",
+    "chalk": "2.3.0",
     "cls-hooked": "4.2.2",
-    "event-stream": "3.3.2"
+    "event-stream": "3.3.4",
+    "lodash": "4.17.4",
+    "uuid": "3.1.0"
   },
   "devDependencies": {
     "@bubltechnology/customizable-commit-analyzer": "1.0.2-0",

package.json

@@ -1,25 +1,41 @@
 'use strict';
 
 const continuationLocalStorage = require('cls-hooked');
+const uuid = require('uuid');
 
 class ContextMiddlewareFactory {
   static getMiddleware() {
+    const namespace = this._createNamespace();
+
     return async function(ctx, next) {
-      const namespace = continuationLocalStorage.createNamespace('session');
-
-      try{
-        await new Promise(namespace.bind(function(resolve, reject) {
-          namespace.set('request_id', ctx.request.header['x-request-id']);
-
-          next().then(resolve).catch(reject);
-        }));
-      } catch(error) {
-        throw error;
-      } finally {
-        continuationLocalStorage.destroyNamespace('session');
-      }
+      await new Promise(namespace.bind(function(resolve, reject) {
+        namespace.set(
+          'request_id',
+          ctx.request.header['x-request-id'] || uuid.v4()
+        );
+
+        next().then(resolve).catch(reject);
+      }));
     };
   }
+
+  static setOnContext(key, value) {
+    const namespace = this._createNamespace();
+    namespace.set(key, value);
+  }
+
+  static destroyNamespace() {
+    if (this._namespace) {
+      continuationLocalStorage.destroyNamespace('session');
+    }
+  }
+
+  static _createNamespace() {
+    if (!this._namespace) {
+      this._namespace = continuationLocalStorage.createNamespace('session');
+    }
+    return this._namespace;
+  }
 }
 
 module.exports = ContextMiddlewareFactory;