koa 2.13.4 → 2.14.1

package/History.md

@@ -1,4 +1,16 @@
 
+2.14.1 / 2022-12-07
+==================
+
+**fixes**
+  * [[`cb92bc9`](http://github.com/koajs/koa/commit/cb92bc98939bd71fd3c01e2dc681caf6545baa38)] - fix: should export createAsyncCtxStorageMiddleware function on application (#1724) (fengmk2 <<fengmk2@gmail.com>>)
+
+2.14.0 / 2022-12-06
+==================
+
+**features**
+  * [[`a293122`](http://github.com/koajs/koa/commit/a29312212839c6f0418152d2a2cd3bf3fd3ee5e4)] - feat: support asyncLocalStorage (#1721) (fengmk2 <<fengmk2@gmail.com>>)
+
 2.13.4 / 2021-10-19
 ==================
 

package/lib/application.js

@@ -8,6 +8,7 @@
 const isGeneratorFunction = require('is-generator-function');
 const debug = require('debug')('koa:application');
 const onFinished = require('on-finished');
+const assert = require('assert');
 const response = require('./response');
 const compose = require('koa-compose');
 const context = require('./context');
@@ -64,6 +65,12 @@ module.exports = class Application extends Emitter {
     if (util.inspect.custom) {
       this[util.inspect.custom] = this.inspect;
     }
+    if (options.asyncLocalStorage) {
+      const { AsyncLocalStorage } = require('async_hooks');
+      assert(AsyncLocalStorage, 'Requires node 12.17.0 or higher to enable asyncLocalStorage');
+      this.ctxStorage = new AsyncLocalStorage();
+      this.use(this.createAsyncCtxStorageMiddleware());
+    }
   }
 
   /**
@@ -153,6 +160,13 @@ module.exports = class Application extends Emitter {
     return handleRequest;
   }
 
+  /**
+   * return currnect contenxt from async local storage
+   */
+  get currentContext() {
+    if (this.ctxStorage) return this.ctxStorage.getStore();
+  }
+
   /**
    * Handle request in callback.
    *
@@ -220,6 +234,15 @@ module.exports = class Application extends Emitter {
   static get default() {
     return Application;
   }
+
+  createAsyncCtxStorageMiddleware() {
+    const app = this;
+    return async function asyncCtxStorage(ctx, next) {
+      await app.ctxStorage.run(ctx, async() => {
+        return await next();
+      });
+    };
+  }
 };
 
 /**

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "koa",
-  "version": "2.13.4",
+  "version": "2.14.1",
   "description": "Koa web app framework",
   "main": "lib/application.js",
   "exports": {