fix(core): generic type of payload action

satanTime · Jun 2, 2021 · c39764a · c39764a
1 parent 2824bb5
commit c39764a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/projects/ngrx-correlation-id/src/lib/actions.ts b/projects/ngrx-correlation-id/src/lib/actions.ts
@@ -28,16 +28,16 @@ export const cidEnd: {
     type: 'ngrx-correlation-id-end';
 } = cidEndInternal;
 
-export class CidPayload implements Action {
+export class CidPayload<T = any> implements Action {
     public readonly type: 'ngrx-correlation-id-payload' = 'ngrx-correlation-id-payload';
 
-    constructor(public readonly cid: string, public readonly payload?: CidTask['payload']) {}
+    constructor(public readonly cid: string, public readonly payload?: CidTask<T>['payload']) {}
 }
 
 export const cidPayloadInternal: any = (props: any) => new CidPayload(props.cid, props.payload);
 cidPayloadInternal.type = 'ngrx-correlation-id-payload';
 export const cidPayload: {
-    (props: {cid: string; payload: CidTask['payload']}): CidPayload;
+    <T = any>(props: {cid: string; payload: CidTask<T>['payload']}): CidPayload;
     type: 'ngrx-correlation-id-payload';
 } = cidPayloadInternal;
 

diff --git a/projects/ngrx-correlation-id/src/lib/reducer.ts b/projects/ngrx-correlation-id/src/lib/reducer.ts
@@ -8,10 +8,10 @@ export interface CidTask<T = any> {
     payload?: T;
 }
 
-export interface State {
+export interface State<T = any> {
     tasks: Array<string>;
     payloads: {
-        [taskId: string]: CidTask['payload'];
+        [taskId: string]: CidTask<T>['payload'];
     };
 }