fix(formula): getStateBySelector to object param

barajs · Nov 29, 2019 · 82a296f · 82a296f
1 parent bb72f82
commit 82a296f
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/src/formula/state.ts b/src/formula/state.ts
@@ -9,14 +9,19 @@ export const getState = withStore(
   },
 )
 
+export interface GetStateBySelectorProps {
+  param: Formula
+  selector: (state: any) => any
+}
 export const getStateBySelector = withStore(
-  (paramFormula: Formula, selector: (state: any) => any) => async (
-    _: any,
+  (config: GetStateBySelectorProps) => async (
+    payload: any,
     store: Store,
     ...rest: any[]
   ) => {
-    const param = await Promise.resolve(paramFormula(_, ...rest))
+    const { param, selector } = config
+    const arg = await Promise.resolve(param(payload, ...rest))
     const state = store.getState()
-    return selector(param)(state)
+    return selector(arg)(state)
   },
 )