effortless-aws 0.28.0 → 0.30.0

package/dist/runtime/wrap-fifo-queue.js

@@ -1,6 +1,6 @@
 import {
   createHandlerRuntime
-} from "../chunk-EZG3NX42.js";
+} from "../chunk-ISJC6CHC.js";
 
 // src/runtime/wrap-fifo-queue.ts
 var parseMessages = (rawRecords, schema) => {
@@ -29,19 +29,22 @@ var parseMessages = (rawRecords, schema) => {
   return messages;
 };
 var wrapFifoQueue = (handler) => {
-  if (!handler.onMessage && !handler.onBatch) {
-    throw new Error("wrapFifoQueue requires a handler with onMessage or onBatch defined");
+  if (!handler.onMessage && !handler.onMessageBatch) {
+    throw new Error("wrapFifoQueue requires a handler with onMessage or onMessageBatch defined");
   }
   const rt = createHandlerRuntime(handler, "fifo-queue", handler.__spec.lambda?.logLevel ?? "info");
   const handleError = handler.onError ?? (({ error }) => console.error(`[effortless:${rt.handlerName}]`, error));
   return async (event) => {
     const startTime = Date.now();
     rt.patchConsole();
-    let shared;
+    let ctxProps = {};
     try {
       const rawRecords = event.Records ?? [];
       const input = { messageCount: rawRecords.length };
-      shared = await rt.commonArgs();
+      const common = await rt.commonArgs();
+      const ctx = common.ctx;
+      ctxProps = ctx && typeof ctx === "object" ? { ...ctx } : {};
+      const shared = { ...ctxProps };
       let messages;
       try {
         messages = parseMessages(rawRecords, handler.schema);
@@ -53,9 +56,14 @@ var wrapFifoQueue = (handler) => {
         };
       }
       const batchItemFailures = [];
-      if (handler.onBatch) {
+      if (handler.onMessageBatch) {
         try {
-          await handler.onBatch({ messages, ...shared });
+          const result = await handler.onMessageBatch({ messages, ...shared });
+          if (result?.failures) {
+            for (const id of result.failures) {
+              batchItemFailures.push({ itemIdentifier: id });
+            }
+          }
         } catch (error) {
           handleError({ error, ...shared });
           for (const message of messages) {
@@ -80,11 +88,11 @@ var wrapFifoQueue = (handler) => {
       }
       return { batchItemFailures };
     } finally {
-      if (handler.onAfterInvoke && shared) {
+      if (handler.onCleanup) {
         try {
-          await handler.onAfterInvoke(shared);
+          await handler.onCleanup(ctxProps);
         } catch (e) {
-          console.error(`[effortless:${rt.handlerName}] onAfterInvoke error`, e);
+          console.error(`[effortless:${rt.handlerName}] onCleanup error`, e);
         }
       }
       rt.restoreConsole();

package/dist/runtime/wrap-table-stream.js

@@ -1,7 +1,7 @@
 import {
   createHandlerRuntime,
   createTableClient
-} from "../chunk-EZG3NX42.js";
+} from "../chunk-ISJC6CHC.js";
 
 // src/runtime/wrap-table-stream.ts
 import { unmarshall } from "@aws-sdk/util-dynamodb";
@@ -36,20 +36,13 @@ var parseRecords = (rawRecords, schema) => {
   }
   return { records, sequenceNumbers };
 };
-var collectFailures = (records, sequenceNumbers) => {
-  const failures = [];
-  for (const record of records) {
-    const seq = sequenceNumbers.get(record);
-    if (seq) failures.push({ itemIdentifier: seq });
-  }
-  return failures;
-};
 var ENV_DEP_SELF = "EFF_DEP_SELF";
 var wrapTableStream = (handler) => {
-  if (!handler.onRecord && !handler.onBatch) {
-    throw new Error("wrapTableStream requires a handler with onRecord or onBatch defined");
+  if (!handler.onRecord && !handler.onRecordBatch) {
+    throw new Error("wrapTableStream requires a handler with onRecord or onRecordBatch defined");
   }
   const tagField = handler.__spec.tagField ?? "tag";
+  const concurrency = handler.__spec.concurrency ?? 1;
   let selfClient = null;
   const getSelfClient = () => {
     if (selfClient) return selfClient;
@@ -67,11 +60,14 @@ var wrapTableStream = (handler) => {
   return async (event) => {
     const startTime = Date.now();
     rt.patchConsole();
-    let shared;
+    let ctxProps = {};
     try {
       const rawRecords = event.Records ?? [];
       const input = { recordCount: rawRecords.length };
-      shared = { ...await rt.commonArgs(), table: getSelfClient() };
+      const common = await rt.commonArgs();
+      const ctx = common.ctx;
+      ctxProps = ctx && typeof ctx === "object" ? { ...ctx } : {};
+      const shared = { ...ctxProps };
       let records;
       let sequenceNumbers;
       try {
@@ -82,37 +78,47 @@ var wrapTableStream = (handler) => {
         return { batchItemFailures: rawRecords.map((r) => r.dynamodb?.SequenceNumber).filter((s) => !!s).map((seq) => ({ itemIdentifier: seq })) };
       }
       const batchItemFailures = [];
-      if (handler.onBatch) {
+      const frozenBatch = Object.freeze(records);
+      if (handler.onRecordBatch) {
         try {
-          await handler.onBatch({ records, ...shared });
+          const result = await handler.onRecordBatch({ records: frozenBatch, ...shared });
+          if (result?.failures) {
+            for (const seq of result.failures) {
+              batchItemFailures.push({ itemIdentifier: seq });
+            }
+          }
         } catch (error) {
           handleError({ error, ...shared });
-          batchItemFailures.push(...collectFailures(records, sequenceNumbers));
-        }
-      } else {
-        const results = [];
-        const failures = [];
-        const onRecord = handler.onRecord;
-        for (const record of records) {
-          try {
-            const result = await onRecord({ record, ...shared });
-            if (result !== void 0) results.push(result);
-          } catch (error) {
-            handleError({ error, ...shared });
-            failures.push({ record, error });
+          for (const record of records) {
             const seq = sequenceNumbers.get(record);
             if (seq) batchItemFailures.push({ itemIdentifier: seq });
           }
         }
-        if (handler.onBatchComplete) {
-          try {
-            await handler.onBatchComplete({ results, failures, ...shared });
-          } catch (error) {
-            handleError({ error, ...shared });
-            for (const record of records) {
+      } else {
+        const onRecord = handler.onRecord;
+        if (concurrency <= 1) {
+          for (const record of records) {
+            try {
+              await onRecord({ record, batch: frozenBatch, ...shared });
+            } catch (error) {
+              handleError({ error, ...shared });
               const seq = sequenceNumbers.get(record);
-              if (seq && !batchItemFailures.some((f) => f.itemIdentifier === seq)) {
-                batchItemFailures.push({ itemIdentifier: seq });
+              if (seq) batchItemFailures.push({ itemIdentifier: seq });
+            }
+          }
+        } else {
+          for (let i = 0; i < records.length; i += concurrency) {
+            const chunk = records.slice(i, i + concurrency);
+            const results = await Promise.allSettled(
+              chunk.map((record) => onRecord({ record, batch: frozenBatch, ...shared }))
+            );
+            for (let j = 0; j < results.length; j++) {
+              const result = results[j];
+              const record = chunk[j];
+              if (result.status === "rejected") {
+                handleError({ error: result.reason, ...shared });
+                const seq = sequenceNumbers.get(record);
+                if (seq) batchItemFailures.push({ itemIdentifier: seq });
               }
             }
           }
@@ -125,11 +131,11 @@ var wrapTableStream = (handler) => {
       }
       return { batchItemFailures };
     } finally {
-      if (handler.onAfterInvoke && shared) {
+      if (handler.onCleanup) {
         try {
-          await handler.onAfterInvoke(shared);
+          await handler.onCleanup(ctxProps);
         } catch (e) {
-          console.error(`[effortless:${rt.handlerName}] onAfterInvoke error`, e);
+          console.error(`[effortless:${rt.handlerName}] onCleanup error`, e);
         }
       }
       rt.restoreConsole();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "effortless-aws",
-  "version": "0.28.0",
+  "version": "0.30.0",
   "license": "MIT",
   "type": "module",
   "description": "Code-first AWS Lambda framework. Export handlers, deploy with one command.",
@@ -45,10 +45,13 @@
     "@types/node": "^22.19.9",
     "tsup": "^8.5.1",
     "type-fest": "^4.22.1",
-    "typescript": "^5.9.3"
+    "typescript": "^5.9.3",
+    "vite-tsconfig-paths": "^6.1.1",
+    "vitest": "^4.0.18"
   },
   "scripts": {
     "build": "tsup",
-    "typecheck": "tsc --noEmit"
+    "typecheck": "tsc --noEmit",
+    "test": "vitest run"
   }
 }