fix: should export createAsyncCtxStorageMiddleware function on applic…

…ation (#1724)
koajs · Dec 7, 2022 · cb92bc9 · cb92bc9
1 parent 382aa08
commit cb92bc9
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 9 deletions.
diff --git a/__tests__/application/index.js b/__tests__/application/index.js
@@ -86,4 +86,9 @@ describe('app', () => {
     assert.deepStrictEqual(Koa.HttpError, CreateError.HttpError);
     assert.throws(() => { throw new CreateError(500, 'test error'); }, Koa.HttpError);
   });
+
+  it('should export createAsyncCtxStorageMiddleware function', () => {
+    const app = new Koa();
+    assert.strictEqual(typeof app.createAsyncCtxStorageMiddleware, 'function');
+  });
 });
diff --git a/lib/application.js b/lib/application.js
@@ -69,7 +69,7 @@ module.exports = class Application extends Emitter {
       const { AsyncLocalStorage } = require('async_hooks');
       assert(AsyncLocalStorage, 'Requires node 12.17.0 or higher to enable asyncLocalStorage');
       this.ctxStorage = new AsyncLocalStorage();
-      this.use(createAsyncCtxStorage(this));
+      this.use(this.createAsyncCtxStorageMiddleware());
     }
   }
 
@@ -234,15 +234,16 @@ module.exports = class Application extends Emitter {
   static get default() {
     return Application;
   }
-};
 
-function createAsyncCtxStorage(app) {
-  return async function asyncCtxStorage(ctx, next) {
-    await app.ctxStorage.run(ctx, async() => {
-      return await next();
-    });
-  };
-}
+  createAsyncCtxStorageMiddleware() {
+    const app = this;
+    return async function asyncCtxStorage(ctx, next) {
+      await app.ctxStorage.run(ctx, async() => {
+        return await next();
+      });
+    };
+  }
+};
 
 /**
  * Response helper.