koa 2.13.3 → 2.14.0

package/History.md

@@ -1,4 +1,16 @@
 
+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
+==================
+
+**fixes**
+  * [[`dbc9c5a`](http://github.com/koajs/koa/commit/dbc9c5a47e2b2799cab403186fdb010df5df6f67)] - fix: Do not response Content-Length if Transfer-Encoding is defined (#1602) (Yiyu He <<dead_horse@qq.com>>)
+
 2.13.3 / 2021-09-24
 ==================
 

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(createAsyncCtxStorage(this));
+    }
   }
 
   /**
@@ -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.
    *
@@ -222,6 +236,14 @@ module.exports = class Application extends Emitter {
   }
 };
 
+function createAsyncCtxStorage(app) {
+  return async function asyncCtxStorage(ctx, next) {
+    await app.ctxStorage.run(ctx, async() => {
+      return await next();
+    });
+  };
+}
+
 /**
  * Response helper.
  */

package/lib/response.js

@@ -191,7 +191,9 @@ module.exports = {
    */
 
   set length(n) {
-    this.set('Content-Length', n);
+    if (!this.has('Transfer-Encoding')) {
+      this.set('Content-Length', n);
+    }
   },
 
   /**

package/package.json

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