feat: throw error after effect hooks

daskyrk · Jan 2, 2020 · 9a871b0 · 9a871b0
1 parent 122fcca
commit 9a871b0
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/src/index.ts b/src/index.ts
@@ -102,7 +102,13 @@ function createStore<
           }
         });
         await Promise.all(ps);
-        const result = await originalEffect(effectFn, payload, extra);
+        let result = null;
+        let error = null;
+        try {
+          result = await originalEffect(effectFn, payload, extra);
+        } catch (e) {
+          error = e;
+        }
         ps = [];
         produce<any, any>(result, (res: any) => {
           for (const afterEffect of hookMap.afterEffect as Array<
@@ -119,6 +125,9 @@ function createStore<
           }
         });
         await Promise.all(ps);
+        if (error) {
+          throw error;
+        }
         return result;
       };
     });