effortless-aws 0.30.0 → 0.32.0

package/dist/runtime/wrap-worker.js

@@ -0,0 +1,109 @@
+import {
+  buildDeps,
+  buildParams
+} from "../chunk-HGSMOO4A.js";
+
+// src/runtime/wrap-worker.ts
+import { SQS } from "@aws-sdk/client-sqs";
+import { ECSClient, UpdateServiceCommand } from "@aws-sdk/client-ecs";
+var wrapWorker = (handler) => {
+  return async () => {
+    const queueUrl = process.env.EFF_WORKER_QUEUE_URL;
+    if (!queueUrl) throw new Error("Missing EFF_WORKER_QUEUE_URL env var");
+    const cluster = process.env.EFF_CLUSTER;
+    const service = process.env.EFF_SERVICE;
+    const idleTimeoutSec = process.env.EFF_IDLE_TIMEOUT ? Number(process.env.EFF_IDLE_TIMEOUT) : 300;
+    const concurrency = Math.min(Math.max(handler.__spec.concurrency ?? 1, 1), 10);
+    const sqs = new SQS({});
+    const ecs = new ECSClient({});
+    const deps = buildDeps(handler.deps);
+    const config = await buildParams(handler.config);
+    let ctx = {};
+    if (handler.setup) {
+      const setupArgs = {};
+      if (deps) setupArgs.deps = deps;
+      if (config) setupArgs.config = config;
+      const result = await handler.setup(setupArgs);
+      if (result && typeof result === "object") ctx = result;
+    }
+    const onMessage = handler.onMessage;
+    if (!onMessage) throw new Error("Worker has no onMessage handler");
+    const onError = handler.onError;
+    let lastMessageAt = Date.now();
+    try {
+      while (true) {
+        const resp = await sqs.receiveMessage({
+          QueueUrl: queueUrl,
+          MaxNumberOfMessages: concurrency,
+          WaitTimeSeconds: 20,
+          MessageSystemAttributeNames: ["ApproximateReceiveCount"]
+        });
+        const messages = resp?.Messages;
+        if (!messages || messages.length === 0) {
+          if (Date.now() - lastMessageAt > idleTimeoutSec * 1e3) {
+            console.log("[effortless:worker] Idle timeout reached, shutting down...");
+            break;
+          }
+          continue;
+        }
+        lastMessageAt = Date.now();
+        const results = await Promise.allSettled(
+          messages.map(async (sqsMsg) => {
+            const parsed = JSON.parse(sqsMsg.Body ?? "null");
+            await onMessage(parsed, ctx);
+            return sqsMsg.ReceiptHandle;
+          })
+        );
+        for (let i = 0; i < results.length; i++) {
+          const result = results[i];
+          const sqsMsg = messages[i];
+          if (result.status === "fulfilled") {
+            await sqs.deleteMessage({
+              QueueUrl: queueUrl,
+              ReceiptHandle: result.value
+            });
+          } else {
+            const retryCount = Number(sqsMsg.Attributes?.ApproximateReceiveCount ?? "1");
+            const parsed = JSON.parse(sqsMsg.Body ?? "null");
+            let action = "retry";
+            if (onError) {
+              try {
+                action = await onError({ error: result.reason, msg: parsed, retryCount, ...ctx });
+              } catch (e) {
+                console.error("[effortless:worker] onError threw", e);
+              }
+            } else {
+              console.error("[effortless:worker]", result.reason);
+            }
+            if (action === "delete") {
+              await sqs.deleteMessage({
+                QueueUrl: queueUrl,
+                ReceiptHandle: sqsMsg.ReceiptHandle
+              });
+            }
+          }
+        }
+      }
+    } finally {
+      if (handler.onCleanup) {
+        try {
+          await handler.onCleanup(ctx);
+        } catch (e) {
+          console.error("[effortless:worker] onCleanup error", e);
+        }
+      }
+      if (cluster && service) {
+        try {
+          await ecs.send(new UpdateServiceCommand({ cluster, service, desiredCount: 0 }));
+          console.log("[effortless:worker] Scaled service to 0");
+        } catch (e) {
+          console.error("[effortless:worker] Failed to scale down", e);
+        }
+      }
+      process.exit(0);
+    }
+  };
+};
+export {
+  wrapWorker
+};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "effortless-aws",
-  "version": "0.30.0",
+  "version": "0.32.0",
   "license": "MIT",
   "type": "module",
   "description": "Code-first AWS Lambda framework. Export handlers, deploy with one command.",
@@ -36,6 +36,7 @@
   },
   "devDependencies": {
     "@aws-sdk/client-dynamodb": "^3.975.0",
+    "@aws-sdk/client-ecs": "^3.1014.0",
     "@aws-sdk/client-s3": "^3.989.0",
     "@aws-sdk/client-sesv2": "^3.994.0",
     "@aws-sdk/client-sqs": "3.989.0",