@cadenza.io/core 3.26.0 → 3.27.0

package/dist/index.mjs

@@ -3853,6 +3853,7 @@ var Actor = class {
     this.stateByKey = /* @__PURE__ */ new Map();
     this.sessionByKey = /* @__PURE__ */ new Map();
     this.idempotencyByKey = /* @__PURE__ */ new Map();
+    this.pendingHydrationByKey = /* @__PURE__ */ new Map();
     this.writeQueueByKey = /* @__PURE__ */ new Map();
     this.nextTaskBindingIndex = 0;
     if (!spec.name || typeof spec.name !== "string") {
@@ -3864,6 +3865,7 @@ var Actor = class {
     }
     this.kind = options.isMeta || spec.kind === "meta" ? "meta" : "standard";
     this.sourceDefinition = options.definitionSource;
+    this.hydrateDurableState = options.hydrateDurableState;
     this.spec = {
       ...spec,
       defaultKey: normalizedDefaultKey,
@@ -3887,9 +3889,11 @@ var Actor = class {
         bindingOptions.touchSession
       );
       const actorKey = this.resolveActorKey(normalizedInput, invocationOptions);
-      this.touchSession(actorKey, invocationOptions.touchSession, Date.now());
+      const now2 = Date.now();
+      this.pruneExpiredActorKeys(now2);
+      this.touchSession(actorKey, invocationOptions.touchSession, now2);
       const runTask = async () => {
-        const stateRecord = this.ensureStateRecord(actorKey);
+        const stateRecord = await this.maybeHydrateStateRecord(actorKey);
         stateRecord.lastReadAt = Date.now();
         let durableStateChanged = false;
         let runtimeStateChanged = false;
@@ -4058,6 +4062,7 @@ var Actor = class {
    */
   getDurableState(actorKey) {
     const key = normalizeActorKey(actorKey) ?? this.spec.defaultKey;
+    this.pruneExpiredActorKeys(Date.now());
     return cloneForDurableState(this.ensureStateRecord(key).durableState);
   }
   /**
@@ -4065,6 +4070,7 @@ var Actor = class {
    */
   getRuntimeState(actorKey) {
     const key = normalizeActorKey(actorKey) ?? this.spec.defaultKey;
+    this.pruneExpiredActorKeys(Date.now());
     return this.ensureStateRecord(key).runtimeState;
   }
   /**
@@ -4078,6 +4084,7 @@ var Actor = class {
    */
   getDurableVersion(actorKey) {
     const key = normalizeActorKey(actorKey) ?? this.spec.defaultKey;
+    this.pruneExpiredActorKeys(Date.now());
     return this.ensureStateRecord(key).version;
   }
   /**
@@ -4085,6 +4092,7 @@ var Actor = class {
    */
   getRuntimeVersion(actorKey) {
     const key = normalizeActorKey(actorKey) ?? this.spec.defaultKey;
+    this.pruneExpiredActorKeys(Date.now());
     return this.ensureStateRecord(key).runtimeVersion;
   }
   /**
@@ -4127,6 +4135,7 @@ var Actor = class {
       this.stateByKey.clear();
       this.sessionByKey.clear();
       this.idempotencyByKey.clear();
+      this.pendingHydrationByKey.clear();
       if ((this.spec.loadPolicy ?? "eager") === "eager") {
         this.ensureStateRecord(this.spec.defaultKey);
       }
@@ -4138,6 +4147,7 @@ var Actor = class {
     }
     this.stateByKey.delete(normalizedKey);
     this.sessionByKey.delete(normalizedKey);
+    this.pendingHydrationByKey.delete(normalizedKey);
     for (const key of this.idempotencyByKey.keys()) {
       if (key.startsWith(`${normalizedKey}:`)) {
         this.idempotencyByKey.delete(key);
@@ -4223,6 +4233,7 @@ var Actor = class {
       runtimeState: this.resolveInitialRuntimeState(),
       version: 0,
       runtimeVersion: 0,
+      hydrationResolved: this.hydrateDurableState === void 0,
       createdAt: now2,
       lastReadAt: now2,
       lastDurableWriteAt: now2,
@@ -4232,6 +4243,54 @@ var Actor = class {
     this.touchSession(actorKey, true, now2);
     return record;
   }
+  async maybeHydrateStateRecord(actorKey) {
+    const record = this.ensureStateRecord(actorKey);
+    if (record.hydrationResolved || !this.hydrateDurableState) {
+      return record;
+    }
+    const pending = this.pendingHydrationByKey.get(actorKey);
+    if (pending) {
+      return pending;
+    }
+    let hydrationPromise;
+    hydrationPromise = (async () => {
+      const current = this.ensureStateRecord(actorKey);
+      if (current.hydrationResolved || !this.hydrateDurableState) {
+        return current;
+      }
+      const snapshot = await this.hydrateDurableState(actorKey);
+      const latest = this.ensureStateRecord(actorKey);
+      if (latest.hydrationResolved) {
+        return latest;
+      }
+      if (snapshot) {
+        const durableVersion = Number(snapshot.durableVersion);
+        if (!Number.isInteger(durableVersion) || durableVersion < 0) {
+          throw new Error(
+            `Actor "${this.spec.name}" received invalid hydrated durable version for key "${actorKey}"`
+          );
+        }
+        if (!isObject2(snapshot.durableState) || Array.isArray(snapshot.durableState)) {
+          throw new Error(
+            `Actor "${this.spec.name}" received invalid hydrated durable state for key "${actorKey}"`
+          );
+        }
+        const hydratedAt = Date.now();
+        latest.durableState = cloneForDurableState(snapshot.durableState);
+        latest.version = durableVersion;
+        latest.lastReadAt = hydratedAt;
+        latest.lastDurableWriteAt = hydratedAt;
+      }
+      latest.hydrationResolved = true;
+      return latest;
+    })().finally(() => {
+      if (this.pendingHydrationByKey.get(actorKey) === hydrationPromise) {
+        this.pendingHydrationByKey.delete(actorKey);
+      }
+    });
+    this.pendingHydrationByKey.set(actorKey, hydrationPromise);
+    return hydrationPromise;
+  }
   touchSession(actorKey, shouldTouch, touchedAt) {
     if (!this.spec.session?.enabled || !shouldTouch) {
       return;
@@ -4255,6 +4314,30 @@ var Actor = class {
       existing.absoluteExpiresAt = touchedAt + absoluteTtlMs;
     }
   }
+  pruneExpiredActorKeys(now2) {
+    if (!this.spec.session?.enabled || this.sessionByKey.size === 0) {
+      return;
+    }
+    for (const [actorKey, session] of this.sessionByKey.entries()) {
+      if (!this.isSessionExpired(session, now2)) {
+        continue;
+      }
+      if (this.writeQueueByKey.has(actorKey)) {
+        continue;
+      }
+      this.stateByKey.delete(actorKey);
+      this.sessionByKey.delete(actorKey);
+    }
+  }
+  isSessionExpired(session, now2) {
+    if (session.absoluteExpiresAt !== null && now2 >= session.absoluteExpiresAt) {
+      return true;
+    }
+    if (session.idleExpiresAt !== null && now2 >= session.idleExpiresAt) {
+      return true;
+    }
+    return false;
+  }
   runWithOptionalIdempotency(taskBindingId, actorKey, options, runTask) {
     const idempotencyPolicy = this.spec.idempotency;
     const enabled = idempotencyPolicy?.enabled ?? false;
@@ -7305,7 +7388,8 @@ var Cadenza = class {
     };
     const actorOptions = {
       isMeta: options.isMeta,
-      definitionSource: definition
+      definitionSource: definition,
+      hydrateDurableState: options.hydrateDurableState
     };
     return this.createActor(spec, actorOptions);
   }