@yieldstar/core 0.1.0 → 0.1.4

package/dist/index.js

@@ -96,14 +96,18 @@ class WorkflowRunner {
     this.schedulerClient = params.schedulerClient;
     this.router = params.router;
   }
-  run = async (task) => {
+  run = async (task, logger) => {
     const { workflowId, executionId } = task;
     const workflow = this.router[workflowId];
     if (!workflow) {
       throw new Error(`No workflow registered for "${workflowId}"`);
     }
     try {
-      const response = await this.runWorkflows({ executionId, workflow });
+      const response = await this.runWorkflows({
+        executionId,
+        workflow,
+        logger
+      });
       switch (response.type) {
         case "workflow-result":
           return response;
@@ -120,10 +124,11 @@ class WorkflowRunner {
     }
   };
   async runWorkflows(params) {
-    const { executionId, workflow } = params;
+    const { executionId, workflow, logger } = params;
     const workflowIterator = workflow({
       heapClient: this.heapClient,
-      executionId
+      executionId,
+      logger
     });
     const iteratorResult = await workflowIterator.next();
     const stageResponse = iteratorResult.value;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@yieldstar/core",
-  "version": "0.1.0",
+  "version": "0.1.4",
   "type": "module",
   "main": "dist/index.js",
   "types": "src/index.ts",

package/src/base/runtime.ts

@@ -1,7 +1,9 @@
+import type { Logger } from "pino";
 import type { WorkflowResult } from "./step";
 
 export type Task = { workflowId: string; executionId: string; params?: any };
 
 export type TaskProcessor = <T = any>(
-  task: Task
+  task: Task,
+  logger: Logger
 ) => Promise<void | WorkflowResult<T>>;

package/src/base/workflow.ts

@@ -1,9 +1,11 @@
+import type { Logger } from "pino";
 import { HeapClient } from "./heap";
 import { StepResponse, WorkflowResult } from "./step";
 
 export type WorkflowGenerator<T = any> = (params: {
   executionId: string;
   heapClient: HeapClient;
+  logger: Logger;
 }) => AsyncGenerator<StepResponse, WorkflowResult<T>, StepResponse>;
 
 export type WorkflowGeneratorReturnType<CG> = CG extends WorkflowGenerator<

package/src/lib/workflow-runner.ts

@@ -1,3 +1,4 @@
+import type { Logger } from "pino";
 import type {
   WorkflowGenerator,
   HeapClient,
@@ -17,13 +18,14 @@ export class WorkflowRunner<
     heapClient: HeapClient;
     schedulerClient: SchedulerClient;
     router: Router;
+    logger: Logger;
   }) {
     this.heapClient = params.heapClient;
     this.schedulerClient = params.schedulerClient;
     this.router = params.router;
   }
 
-  run: TaskProcessor = async (task) => {
+  run: TaskProcessor = async (task, logger) => {
     const { workflowId, executionId } = task;
     const workflow = this.router[workflowId];
 
@@ -32,7 +34,11 @@ export class WorkflowRunner<
     }
 
     try {
-      const response = await this.runWorkflows({ executionId, workflow });
+      const response = await this.runWorkflows({
+        executionId,
+        workflow,
+        logger,
+      });
 
       switch (response.type) {
         case "workflow-result":
@@ -55,12 +61,14 @@ export class WorkflowRunner<
   private async runWorkflows<T>(params: {
     executionId: string;
     workflow: WorkflowGenerator<T>;
+    logger: Logger;
   }): Promise<WorkflowResult<T> | WorkflowDelay> {
-    const { executionId, workflow } = params;
+    const { executionId, workflow, logger } = params;
 
     const workflowIterator = workflow({
       heapClient: this.heapClient,
       executionId,
+      logger,
     });
 
     const iteratorResult = await workflowIterator.next();