Include async functions in origins promise detection

koajs · Dec 20, 2019 · faf4415 · faf4415
1 parent 71c4d00
commit faf4415
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -56,7 +56,8 @@ module.exports = function(options) {
     let origin;
     if (typeof options.origin === 'function') {
       origin = options.origin(ctx);
-      if (origin instanceof Promise) origin = await origin;
+      const isAsync = origin.constructor.name === "AsyncFunction";
+      if (origin instanceof Promise || isAsync) origin = await origin;
       if (!origin) return await next();
     } else {
       origin = options.origin || requestOrigin;