fix(context): fixed runPromise mem leak and incorrect propagation

Author: eliabecardoso

context-legacy.js

@@ -127,22 +127,24 @@ Namespace.prototype.runPromise = function runPromise(fn) {
   this.enter(context);
 
   let promise = fn(context);
+
   if (!promise || !promise.then || !promise.catch) {
     throw new Error('fn must return a promise.');
   }
 
   if (DEBUG_CLS_HOOKED) {
     debug2(' BEFORE runPromise: ' + this.name + ' uid:' + currentUid + ' len:' + this._set.length + ' ' +
-      util.inspect(context));
+    util.inspect(context));
   }
 
+  this.exit(context);
+
   return promise
     .then(result => {
       if (DEBUG_CLS_HOOKED) {
         debug2(' AFTER runPromise: ' + this.name + ' uid:' + currentUid + ' len:' + this._set.length + ' ' +
           util.inspect(context));
       }
-      this.exit(context);
       return result;
     })
     .catch(err => {
@@ -151,7 +153,6 @@ Namespace.prototype.runPromise = function runPromise(fn) {
         debug2(' AFTER runPromise: ' + this.name + ' uid:' + currentUid + ' len:' + this._set.length + ' ' +
           util.inspect(context));
       }
-      this.exit(context);
       throw err;
     });
 };

context.js

@@ -138,20 +138,20 @@ Namespace.prototype.runPromise = function runPromise(fn) {
     debug2('CONTEXT-runPromise BEFORE: (' + this.name + ') currentUid:' + currentUid + ' len:' + this._set.length + ' ' + util.inspect(context));
   }
 
+  this.exit(context);
+
   return promise
     .then(result => {
       if (DEBUG_CLS_HOOKED) {
         debug2('CONTEXT-runPromise AFTER then: (' + this.name + ') currentUid:' + currentUid + ' len:' + this._set.length + ' ' + util.inspect(context));
       }
-      this.exit(context);
       return result;
     })
     .catch(err => {
       err[ERROR_SYMBOL] = context;
       if (DEBUG_CLS_HOOKED) {
         debug2('CONTEXT-runPromise AFTER catch: (' + this.name + ') currentUid:' + currentUid + ' len:' + this._set.length + ' ' + util.inspect(context));
       }
-      this.exit(context);
       throw err;
     });
 };