@openfn/ws-worker 0.2.11 → 0.2.12

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # ws-worker
 
+## 0.2.12
+
+### Patch Changes
+
+- 6c3e9e42: Ensure capacity is also set on the engine
+- Updated dependencies [05ccc10b]
+- Updated dependencies [7235bf5e]
+  - @openfn/engine-multi@0.2.3
+
 ## 0.2.11
 
 ### Patch Changes

package/dist/index.js

@@ -409,7 +409,7 @@ function onJobStart({ channel, state }, event) {
   });
 }
 function onJobError(context, event) {
-  const { state, error, jobId } = event;
+  const { state = {}, error, jobId } = event;
   if (state.errors?.[jobId]?.message === error.message) {
     onJobComplete(context, event);
   } else {
@@ -462,9 +462,13 @@ async function onWorkflowComplete({ state, channel, onFinish }, _event) {
   });
   onFinish({ reason, state: result });
 }
-async function onWorkflowError({ state, channel, logger, onFinish }, event) {
+async function onWorkflowError(context, event) {
+  const { state, channel, logger, onFinish } = context;
   try {
     const reason = calculateJobExitReason("", { data: {} }, event);
+    if (state.activeJob) {
+      await onJobError(context, { error: event });
+    }
     await sendEvent(channel, ATTEMPT_COMPLETE, {
       final_dataclip_id: state.lastDataclipId,
       ...reason
@@ -473,6 +477,7 @@ async function onWorkflowError({ state, channel, logger, onFinish }, event) {
   } catch (e) {
     logger.error("ERROR in workflow-error handler:", e.message);
     logger.error(e);
+    onFinish({});
   }
 }
 function onJobLog({ channel, state }, event) {

package/dist/start.js

@@ -4954,6 +4954,7 @@ async function createMock() {
       dispatch("workflow-start", { workflowId: id, threadId });
       try {
         await run(xplan, void 0, opts);
+        dispatch("workflow-complete", { workflowId: id, threadId });
       } catch (e) {
         dispatch("workflow-error", {
           threadId,
@@ -4961,9 +4962,9 @@ async function createMock() {
           type: e.name,
           message: e.message
         });
+      } finally {
+        delete activeWorkflows[id];
       }
-      delete activeWorkflows[id];
-      dispatch("workflow-complete", { workflowId: id, threadId });
     }, 1);
   };
   const getStatus = () => {
@@ -5394,7 +5395,7 @@ function onJobStart({ channel, state }, event) {
   });
 }
 function onJobError(context, event) {
-  const { state, error, jobId } = event;
+  const { state = {}, error, jobId } = event;
   if (state.errors?.[jobId]?.message === error.message) {
     onJobComplete(context, event);
   } else {
@@ -5447,9 +5448,13 @@ async function onWorkflowComplete({ state, channel, onFinish }, _event) {
   });
   onFinish({ reason, state: result });
 }
-async function onWorkflowError({ state, channel, logger: logger2, onFinish }, event) {
+async function onWorkflowError(context, event) {
+  const { state, channel, logger: logger2, onFinish } = context;
   try {
     const reason = calculateJobExitReason("", { data: {} }, event);
+    if (state.activeJob) {
+      await onJobError(context, { error: event });
+    }
     await sendEvent(channel, ATTEMPT_COMPLETE, {
       final_dataclip_id: state.lastDataclipId,
       ...reason
@@ -5458,6 +5463,7 @@ async function onWorkflowError({ state, channel, logger: logger2, onFinish }, ev
   } catch (e) {
     logger2.error("ERROR in workflow-error handler:", e.message);
     logger2.error(e);
+    onFinish({});
   }
 }
 function onJobLog({ channel, state }, event) {
@@ -5790,12 +5796,14 @@ if (args.mock) {
     engineReady(engine);
   });
 } else {
-  createRTE({ repoDir: args.repoDir, memoryLimitMb: args.runMemory }).then(
-    (engine) => {
-      logger.debug("engine created");
-      engineReady(engine);
-    }
-  );
+  createRTE({
+    repoDir: args.repoDir,
+    memoryLimitMb: args.runMemory,
+    maxWorkers: args.capacity
+  }).then((engine) => {
+    logger.debug("engine created");
+    engineReady(engine);
+  });
 }
 /**
  * @fileoverview Main entrypoint for libraries using yargs-parser in Node.js

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@openfn/ws-worker",
-  "version": "0.2.11",
+  "version": "0.2.12",
   "description": "A Websocket Worker to connect Lightning to a Runtime Engine",
   "main": "dist/index.js",
   "type": "module",
@@ -21,7 +21,7 @@
     "koa-logger": "^3.2.1",
     "phoenix": "^1.7.7",
     "ws": "^8.14.1",
-    "@openfn/engine-multi": "0.2.2",
+    "@openfn/engine-multi": "0.2.3",
     "@openfn/logger": "0.0.19",
     "@openfn/runtime": "0.2.1"
   },
@@ -40,7 +40,7 @@
     "tsup": "^6.2.3",
     "typescript": "^4.6.4",
     "yargs": "^17.6.2",
-    "@openfn/lightning-mock": "1.1.4"
+    "@openfn/lightning-mock": "1.1.5"
   },
   "files": [
     "dist",