@yieldstar/core 0.4.5 → 0.5.0

package/src/base/workflow.ts

@@ -1,6 +1,7 @@
 import type { Logger } from "pino";
 import type { EventParams, WorkflowEvent, EventContext } from "./event";
 import { HeapClient } from "./heap";
+import { StoreClient } from "./store";
 import { StepResponse, WorkflowResult } from "./step";
 
 export type WorkflowGeneratorParams<
@@ -9,6 +10,7 @@ export type WorkflowGeneratorParams<
 > = {
   event: WorkflowEvent<Params, Context>;
   heapClient: HeapClient;
+  storeClient: StoreClient;
   logger: Logger;
 };
 

package/src/index.ts

@@ -2,6 +2,42 @@ export type { HeapClient, HeapRecord } from "./base/heap";
 export type { WorkflowInvoker } from "./base/invoker";
 export type { EventProcessor } from "./base/runtime";
 export type { SchedulerClient } from "./base/scheduler";
+export type { StandardSchemaV1 } from "@standard-schema/spec";
+export type {
+  Draft,
+  RuntimeStore,
+  StoreDefinition,
+  StoreDeleteFromResult,
+  StoreKey,
+  StorePath,
+  StoreSelector,
+  StoreSnapshot,
+  StoreState,
+  StoreStepId,
+  StoreTakeResult,
+  StoreUpdateFromResult,
+  StoreUpdateResult,
+  StoreVersion,
+  StoreWaiter,
+  StoreMutation,
+  StoreMutationCommitResult,
+  TrackedStoreUpdater,
+} from "./base/store";
+export {
+  StoreClient,
+  CasStoreClient,
+  assertSynchronousClaim,
+  assertSynchronousUpdater,
+  cloneStoreState,
+  defineStore,
+  diffStorePaths,
+  isStoreSelectorMatch,
+  storePathsIntersect,
+  trackStoreSelector,
+  trackStoreUpdater,
+  unwrapTrackedValue,
+  validateStoreState,
+} from "./base/store";
 export type {
   EventParams,
   EventContext,
@@ -29,6 +65,7 @@ export {
   StepKey,
   StepResponse,
   StepResult,
+  StepStoreWait,
   WorkflowDelay,
   WorkflowRestart,
   WorkflowResult,

package/src/lib/workflow-runner.ts

@@ -3,11 +3,12 @@ import type {
   WorkflowGenerator,
   HeapClient,
   SchedulerClient,
+  StoreClient,
   EventProcessor,
   WorkflowEvent,
   EventParams,
 } from "..";
-import { StepDelay, WorkflowDelay, WorkflowResult } from "..";
+import { StepDelay, StepStoreWait, WorkflowDelay, WorkflowResult } from "..";
 import { EventContext } from "../base/event";
 
 export class WorkflowRunner<
@@ -15,16 +16,19 @@ export class WorkflowRunner<
 > {
   private heapClient: HeapClient;
   private schedulerClient: SchedulerClient;
+  private storeClient: StoreClient;
   private router: Router;
 
   constructor(params: {
     heapClient: HeapClient;
     schedulerClient: SchedulerClient;
+    storeClient: StoreClient;
     router: Router;
     logger: Logger;
   }) {
     this.heapClient = params.heapClient;
     this.schedulerClient = params.schedulerClient;
+    this.storeClient = params.storeClient;
     this.router = params.router;
   }
 
@@ -49,6 +53,9 @@ export class WorkflowRunner<
         case "workflow-delay":
           this.schedulerClient.requestWakeUp(event, response.resumeIn);
           break;
+
+        case "store-wait":
+          break;
       }
     } catch (err) {
       // todo: distinguish between a workflow error and a system error
@@ -64,12 +71,13 @@ export class WorkflowRunner<
     workflow: WorkflowGenerator<Params, Result, Context>;
     event: WorkflowEvent<Params, Context>;
     logger: Logger;
-  }): Promise<WorkflowResult<Result> | WorkflowDelay> {
+  }): Promise<WorkflowResult<Result> | WorkflowDelay | StepStoreWait> {
     const { workflow, event, logger } = params;
 
     const workflowIterator = workflow({
       event,
       heapClient: this.heapClient,
+      storeClient: this.storeClient,
       logger,
     });
 
@@ -85,6 +93,10 @@ export class WorkflowRunner<
       return new WorkflowDelay(stageResponse.resumeIn - Date.now());
     }
 
+    if (stageResponse instanceof StepStoreWait) {
+      return stageResponse;
+    }
+
     throw new Error(
       "Workflow runner critical error. This is likely because your worklow and worker are referencing different versions of @yieldstar/core."
     );

package/src/utils/map.test.ts

@@ -1,4 +1,4 @@
-import { expect, test } from "bun:test";
+import { expect, test } from "vitest";
 import { ReadOnlyMap, FreezableMap } from "./map";
 
 test("ReadOnlyMap", () => {

package/src/utils/map.ts

@@ -3,8 +3,14 @@ import { errorWithOriginalStack } from "./error";
 export class ReadOnlyMap<K, V> implements ReadonlyMap<K, V> {
   private sourceMap: Map<K, V>;
 
-  constructor(sourceMap: Map<K, V>) {
-    this.sourceMap = sourceMap;
+  constructor(sourceMap: Map<K, V> | Record<any, any> | undefined) {
+    if (sourceMap instanceof Map) {
+      this.sourceMap = sourceMap;
+    } else if (typeof sourceMap === "object" && sourceMap !== null) {
+      this.sourceMap = new Map(Object.entries(sourceMap)) as Map<K, V>;
+    } else {
+      this.sourceMap = new Map();
+    }
   }
 
   get(key: K): V | undefined {