refactor(store): allow for custom action types with createReducer (#2021

)
ngrx · Jul 16, 2019 · 9d413c9 · 9d413c9
1 parent 57fd3d7
commit 9d413c9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/modules/store/src/reducer_creator.ts b/modules/store/src/reducer_creator.ts
@@ -225,18 +225,18 @@ export function on(
  * }
  * ```
  */
-export function createReducer<S>(
+export function createReducer<S, A extends Action = Action>(
   initialState: S,
   ...ons: On<S>[]
-): ActionReducer<S> {
-  const map = new Map<string, ActionReducer<S>>();
+): ActionReducer<S, A> {
+  const map = new Map<string, ActionReducer<S, A>>();
   for (let on of ons) {
     for (let type of on.types) {
       map.set(type, on.reducer);
     }
   }
 
-  return function(state: S = initialState, action: Action): S {
+  return function(state: S = initialState, action: A): S {
     const reducer = map.get(action.type);
     return reducer ? reducer(state, action) : state;
   };