fix: implement awaitPromise capability

opensumi · Sep 8, 2023 · e22b8fa · e22b8fa
1 parent 05de75e
commit e22b8fa
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/helper/compose.ts b/src/helper/compose.ts
@@ -36,15 +36,15 @@ function dispatch<C>(
   if (idx < middlewareList.length) {
     const middleware = middlewareList[idx];
 
-    maybePromise = middleware({
+    const t = middleware({
       ...ctx,
-      proceed: () => {
-        return dispatch(middlewareList, idx + 1, stack, ctx);
-      },
+      proceed: () => dispatch(middlewareList, idx + 1, stack, ctx),
     } as Context<C>);
 
     if (middleware.awaitPromise) {
-      maybePromise = Promise.resolve(maybePromise);
+      maybePromise = Promise.resolve(t);
+    } else {
+      maybePromise = t;
     }
   } else if (ctx.proceed) {
     maybePromise = ctx.proceed();