feat(sdk): Add summary generators for map and parallel operations (#91)

*Description of changes:*

- Add summaryGenerator field to ConcurrencyConfig interface
- Create predefined summary generators for map and parallel operations
- Update ConcurrencyController to pass summaryGenerator to top-level
runInChildContext only
- Update map and parallel handlers to use predefined summary generators
- Update tests to expect new configuration fields

This enables summaries for large BatchResult payloads in map and
parallel operations.

*Issue #, if available:*


By submitting this pull request, I confirm that you can use, modify,
copy, and redistribute this contribution, under the terms of your
choice.

Co-authored-by: Pooya Paridel <parpooya@amazon.com>

Author: ParidelPooya

packages/aws-durable-execution-sdk-js/bundle-size-history.json

@@ -1,14 +1,4 @@
 [
-  {
-    "timestamp": "2025-06-27T16:47:20.156Z",
-    "size": 888765,
-    "gitCommit": "6c98b2ec2c523a92309a7b8ee00e9c61f83392cf"
-  },
-  {
-    "timestamp": "2025-07-09T20:06:07.839Z",
-    "size": 891952,
-    "gitCommit": "bf7ad0b2cbd6aa0a0bf265e67a013bdf1dec4854"
-  },
   {
     "timestamp": "2025-07-10T01:35:30.375Z",
     "size": 891887,
@@ -248,5 +238,15 @@
     "timestamp": "2025-09-29T21:53:05.018Z",
     "size": 361581,
     "gitCommit": "1eba8c6d9f097275746307697346a5f2caf27c23"
+  },
+  {
+    "timestamp": "2025-10-02T19:08:08.763Z",
+    "size": 357086,
+    "gitCommit": "77e3ad5a04f502ab227318b89b47b5ac8b63630f"
+  },
+  {
+    "timestamp": "2025-10-02T19:56:14.710Z",
+    "size": 357050,
+    "gitCommit": "7934b1ddf1b32907d5a20f1995a49bfa22845821"
   }
-]
+]

packages/aws-durable-execution-sdk-js/src/handlers/concurrent-execution-handler/concurrent-execution-handler.ts

@@ -18,6 +18,7 @@ export interface ConcurrencyConfig {
   maxConcurrency?: number;
   topLevelSubType?: string;
   iterationSubType?: string;
+  summaryGenerator?: (result: any) => string;
   completionConfig?: {
     minSuccessful?: number;
     toleratedFailureCount?: number;
@@ -300,6 +301,7 @@ export const createConcurrentExecutionHandler = (
 
     return await runInChildContext(name, executeOperation, {
       subType: config?.topLevelSubType,
+      summaryGenerator: config?.summaryGenerator,
     });
   };
 };

packages/aws-durable-execution-sdk-js/src/handlers/invoke-handler/invoke-handler.test.ts

@@ -268,9 +268,13 @@ describe("InvokeHandler", () => {
     });
 
     it("should wait when operation is in progress and other operations are running", async () => {
-      const mockGetStepData = jest.fn()
+      const mockGetStepData = jest
+        .fn()
         .mockReturnValueOnce({ Status: OperationStatus.STARTED })
-        .mockReturnValueOnce({ Status: OperationStatus.SUCCEEDED, InvokeDetails: { Result: '{"result":"success"}' } });
+        .mockReturnValueOnce({
+          Status: OperationStatus.SUCCEEDED,
+          InvokeDetails: { Result: '{"result":"success"}' },
+        });
 
       mockContext.getStepData = mockGetStepData;
       mockHasRunningOperations.mockReturnValue(true); // Other operations running
@@ -303,7 +307,8 @@ describe("InvokeHandler", () => {
     });
 
     it("should create checkpoint and terminate for new invoke without name", async () => {
-      const mockGetStepData = jest.fn()
+      const mockGetStepData = jest
+        .fn()
         .mockReturnValueOnce(undefined) // First call - no step data
         .mockReturnValueOnce({ Status: OperationStatus.STARTED }); // After checkpoint
 
@@ -358,7 +363,8 @@ describe("InvokeHandler", () => {
     });
 
     it("should create checkpoint and terminate for new invoke with name", async () => {
-      const mockGetStepData = jest.fn()
+      const mockGetStepData = jest
+        .fn()
         .mockReturnValueOnce(undefined) // First call - no step data
         .mockReturnValueOnce({ Status: OperationStatus.STARTED }); // After checkpoint
 
@@ -397,7 +403,8 @@ describe("InvokeHandler", () => {
     });
 
     it("should handle invoke with options", async () => {
-      const mockGetStepData = jest.fn()
+      const mockGetStepData = jest
+        .fn()
         .mockReturnValueOnce(undefined) // First call - no step data
         .mockReturnValueOnce({ Status: OperationStatus.STARTED }); // After checkpoint
 

packages/aws-durable-execution-sdk-js/src/handlers/invoke-handler/invoke-handler.ts

@@ -106,14 +106,18 @@ export const createInvokeHandler = (
           });
           continue; // Re-evaluate status after waiting
         }
-        
+
         // No other operations running, safe to terminate
         log(
           context.isVerbose,
           "⏳",
           `Invoke ${name || funcId} still in progress, terminating`,
         );
-        return terminate(context, TerminationReason.OPERATION_TERMINATED, stepId);
+        return terminate(
+          context,
+          TerminationReason.OPERATION_TERMINATED,
+          stepId,
+        );
       }
 
       // Execute with potential interception (testing)

packages/aws-durable-execution-sdk-js/src/handlers/map-handler/map-handler.test.ts

@@ -47,7 +47,15 @@ describe("Map Handler", () => {
           { id: "map-item-1", data: "item2", index: 1 },
         ],
         expect.any(Function),
-        TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+        {
+          ...{
+            ...TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+            summaryGenerator: expect.any(Function),
+            completionConfig: undefined,
+          },
+          summaryGenerator: expect.any(Function),
+          completionConfig: undefined,
+        },
       );
     });
 
@@ -70,7 +78,11 @@ describe("Map Handler", () => {
           { id: "map-item-1", data: "item2", index: 1 },
         ],
         expect.any(Function),
-        TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+        {
+          ...TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+          summaryGenerator: expect.any(Function),
+          completionConfig: undefined,
+        },
       );
     });
 
@@ -89,15 +101,23 @@ describe("Map Handler", () => {
         undefined,
         [{ id: "map-item-0", data: "item", index: 0 }],
         expect.any(Function),
-        TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+        {
+          ...TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+          summaryGenerator: expect.any(Function),
+          completionConfig: undefined,
+        },
       );
     });
 
     it("should parse parameters with config", async () => {
       const items = ["item1", "item2"];
       const mapFunc: MapFunc<string> = jest.fn().mockResolvedValue("result");
       const config = {
-        ...TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+        ...{
+          ...TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+          summaryGenerator: expect.any(Function),
+          completionConfig: undefined,
+        },
         maxConcurrency: 2,
       };
 
@@ -116,7 +136,14 @@ describe("Map Handler", () => {
           { id: "map-item-1", data: "item2", index: 1 },
         ],
         expect.any(Function),
-        { ...TEST_CONSTANTS.DEFAULT_MAP_CONFIG, maxConcurrency: 2 },
+        {
+          ...{
+            ...TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+            summaryGenerator: expect.any(Function),
+            completionConfig: undefined,
+          },
+          maxConcurrency: 2,
+        },
       );
     });
   });
@@ -154,7 +181,11 @@ describe("Map Handler", () => {
         undefined,
         [],
         expect.any(Function),
-        TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+        {
+          ...TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+          summaryGenerator: expect.any(Function),
+          completionConfig: undefined,
+        },
       );
     });
 
@@ -179,7 +210,11 @@ describe("Map Handler", () => {
           { id: "map-item-2", data: "item3", index: 2 },
         ],
         expect.any(Function),
-        TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+        {
+          ...TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+          summaryGenerator: expect.any(Function),
+          completionConfig: undefined,
+        },
       );
     });
 
@@ -251,7 +286,11 @@ describe("Map Handler", () => {
       const items = ["item1", "item2"];
       const mapFunc: MapFunc<string> = jest.fn().mockResolvedValue("result");
       const config = {
-        ...TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+        ...{
+          ...TEST_CONSTANTS.DEFAULT_MAP_CONFIG,
+          summaryGenerator: expect.any(Function),
+          completionConfig: undefined,
+        },
         maxConcurrency: 5,
       };
 

packages/aws-durable-execution-sdk-js/src/handlers/map-handler/map-handler.ts

@@ -9,6 +9,7 @@ import {
 } from "../../types";
 import { log } from "../../utils/logger/logger";
 import { BatchResult } from "../concurrent-execution-handler/batch-result";
+import { createMapSummaryGenerator } from "../../utils/summary-generators";
 
 export const createMapHandler = (
   context: ExecutionContext,
@@ -74,6 +75,7 @@ export const createMapHandler = (
       maxConcurrency: config?.maxConcurrency,
       topLevelSubType: OperationSubType.MAP,
       iterationSubType: OperationSubType.MAP_ITERATION,
+      summaryGenerator: createMapSummaryGenerator(),
       completionConfig: config?.completionConfig,
     });
 

packages/aws-durable-execution-sdk-js/src/handlers/parallel-handler/parallel-handler.test.ts

@@ -63,6 +63,7 @@ describe("Parallel Handler", () => {
           completionConfig: undefined,
           iterationSubType: "ParallelBranch",
           maxConcurrency: undefined,
+          summaryGenerator: expect.any(Function),
           topLevelSubType: "Parallel",
         },
       );
@@ -91,6 +92,7 @@ describe("Parallel Handler", () => {
           completionConfig: undefined,
           iterationSubType: "ParallelBranch",
           maxConcurrency: 2,
+          summaryGenerator: expect.any(Function),
           topLevelSubType: "Parallel",
         },
       );
@@ -119,6 +121,7 @@ describe("Parallel Handler", () => {
           completionConfig: undefined,
           iterationSubType: "ParallelBranch",
           maxConcurrency: 3,
+          summaryGenerator: expect.any(Function),
           topLevelSubType: "Parallel",
         },
       );

packages/aws-durable-execution-sdk-js/src/handlers/parallel-handler/parallel-handler.ts

@@ -9,6 +9,7 @@ import {
 } from "../../types";
 import { log } from "../../utils/logger/logger";
 import { BatchResult } from "../concurrent-execution-handler/batch-result";
+import { createParallelSummaryGenerator } from "../../utils/summary-generators";
 
 export const createParallelHandler = (
   context: ExecutionContext,
@@ -84,6 +85,7 @@ export const createParallelHandler = (
       maxConcurrency: config?.maxConcurrency,
       topLevelSubType: OperationSubType.PARALLEL,
       iterationSubType: OperationSubType.PARALLEL_BRANCH,
+      summaryGenerator: createParallelSummaryGenerator(),
       completionConfig: config?.completionConfig,
     });