@effect-agent/storage-memory 0.1.0-beta.49 → 0.1.0-beta.51

package/dist/MemorySubmissionLedger.mjs

@@ -1,8 +1,8 @@
 import { t as __exportAll } from "./rolldown-runtime-D7D4PA-g.mjs";
-import { Clock, DateTime, Duration, Effect, Layer, Option, Ref, Schema, Stream } from "effect";
+import { Cause, Clock, DateTime, Duration, Effect, Exit, Fiber, Layer, Option, Ref, Schema, Stream } from "effect";
 import { AttemptId, ReceiptId, SubmissionId } from "@effect-agent/core/Identifiers";
 import { PersistedJson, ProducerEpoch } from "@effect-agent/thread/Records";
-import { AbortCommand, AbortIntent, AbortIntentRequest, AdmissionAdmitted, AdmissionConflict, AdmissionIndeterminate, AdmissionNotAdmitted, AdmissionRequest, AdmissionResult, ApprovalConflict, ApprovalDecisionCommand, ApprovalDecisionIntent, AttachChildToReservationRequest, BeginChildBudgetReleaseRequest, ChildAttachmentSnapshot, ChildBudgetReservationRequest, ChildBudgetReservationSnapshot, ChildReservationConflict, ChildSettledNotification, Claim, ClaimJoiningRequest, ClaimRequest, DEFAULT_OWNERSHIP_LEASE_DURATION, InputAppliedMarker, JoinSnapshot, JoinedToHost, JoiningClaim, LedgerCapabilities, LedgerError, MarkInputAppliedRequest, MarkJoinedRequest, MarkReadyRequest, MarkUnknownRequest, OwnershipLost, OwnershipRenewal, OwnershipSnapshot, OwnershipToken, QueueSequence, RecoverySnapshot, RecoverySnapshotRequest, ReleaseChildBudgetRequest, ReleaseOwnershipRequest, RenewOwnershipRequest, ReservedChildBudget, ReservedSettlement, RevertJoiningRequest, Settlement, SettlementConflict, SettlementFinalization, SettlementReservation, SettlementReservationSnapshot, SubmissionLedger, SubmissionLookup, SubmissionLookupByKey, SubmissionSnapshot, SuspendRequest, SuspensionSnapshot, UnknownResolution, UnknownResolutionCommand, UnknownResolutionConflict, UnknownResolutionIntent, settlementFailureFromRecord } from "@effect-agent/thread/SubmissionLedger";
+import { AbortCommand, AbortIntent, AbortIntentRequest, AdmissionAdmitted, AdmissionConflict, AdmissionIndeterminate, AdmissionNotAdmitted, AdmissionPolicyError, AdmissionRequest, AdmissionResult, ApprovalConflict, ApprovalDecisionCommand, ApprovalDecisionIntent, AttachChildToReservationRequest, BeginChildBudgetReleaseRequest, ChildAttachmentSnapshot, ChildBudgetReservationRequest, ChildBudgetReservationSnapshot, ChildReservationConflict, ChildSettledNotification, Claim, ClaimJoiningRequest, ClaimRequest, DEFAULT_OWNERSHIP_LEASE_DURATION, InputAppliedMarker, JoinSnapshot, JoinedToHost, JoiningClaim, LedgerCapabilities, LedgerError, MarkInputAppliedRequest, MarkJoinedRequest, MarkReadyRequest, MarkUnknownRequest, OwnershipLost, OwnershipRenewal, OwnershipSnapshot, OwnershipToken, QueueSequence, RecoverySnapshot, RecoverySnapshotRequest, ReleaseChildBudgetRequest, ReleaseOwnershipRequest, RenewOwnershipRequest, ReservedChildBudget, ReservedSettlement, RevertJoiningRequest, Settlement, SettlementConflict, SettlementFinalization, SettlementReservation, SettlementReservationSnapshot, SubmissionAdmissionFence, SubmissionLedger, SubmissionLookup, SubmissionLookupByKey, SubmissionSnapshot, SuspendRequest, SuspensionSnapshot, UnknownResolution, UnknownResolutionCommand, UnknownResolutionConflict, UnknownResolutionIntent, settlementFailureFromRecord } from "@effect-agent/thread/SubmissionLedger";
 //#region src/MemorySubmissionLedger.ts
 var MemorySubmissionLedger_exports = /* @__PURE__ */ __exportAll({
 	MemorySubmissionLedgerLive: () => MemorySubmissionLedgerLive,
@@ -77,6 +77,8 @@ const toSnapshot = (row) => SubmissionSnapshot.make({
 	receiptId: row.receiptId,
 	state: row.state,
 	createdAt: utc(row.createdAtMillis),
+	...row.admissionGroup === void 0 ? {} : { admissionGroup: row.admissionGroup },
+	...row.admissionFence === void 0 ? {} : { admissionFence: row.admissionFence },
 	...row.settledOutcome === void 0 ? {} : { settledOutcome: row.settledOutcome },
 	...row.readyAtMillis === void 0 ? {} : { readyAt: utc(row.readyAtMillis) },
 	...row.parentLinkage === void 0 ? {} : { parentLinkage: row.parentLinkage }
@@ -161,18 +163,20 @@ const makeSubmissionLedger = (options = {}) => Effect.gen(function* () {
 		childReservations: /* @__PURE__ */ new Map(),
 		mintCounter: 0
 	});
+	const admissionFence = yield* SubmissionAdmissionFence;
 	const leaseMillis = Duration.toMillis(DEFAULT_OWNERSHIP_LEASE_DURATION);
 	const capabilities = Effect.succeed(LedgerCapabilities.make({ durability: "non-durable" }));
 	const admit = Effect.fn("MemorySubmissionLedger.admit")((unvalidated) => Effect.gen(function* () {
 		const request = yield* validate(AdmissionRequest, "admit", unvalidated);
 		const nowMillis = yield* Clock.currentTimeMillis;
+		const services = yield* Effect.context();
 		const decision = yield* Ref.modify(state, (current) => {
 			const key = admissionKey(request.threadId, request.principal, request.idempotencyKey);
 			const existingId = current.admissionIndex.get(key);
 			if (existingId !== void 0) {
 				const existing = current.submissions.get(existingId);
 				if (existing === void 0) return [failure(ledgerError("admit", "Admission index references a missing Submission")), current];
-				if (existing.row.inputDigest !== request.inputDigest || !sameParentLinkage(existing.row.parentLinkage, request.parentLinkage)) return [failure(AdmissionConflict.make({
+				if (existing.row.inputDigest !== request.inputDigest || !sameParentLinkage(existing.row.parentLinkage, request.parentLinkage) || existing.row.admissionGroup !== request.admissionGroup || !Schema.toEquivalence(Schema.optional(Schema.Json))(existing.row.admissionFence, request.admissionFence)) return [failure(AdmissionConflict.make({
 					threadId: request.threadId,
 					principal: request.principal,
 					idempotencyKey: request.idempotencyKey,
@@ -187,6 +191,15 @@ const makeSubmissionLedger = (options = {}) => Effect.gen(function* () {
 					replayed: true
 				})), current];
 			}
+			const checked = Effect.runSyncExitWith(services)(admissionFence.check(request));
+			if (Exit.isFailure(checked)) return [{
+				_tag: "cause",
+				cause: checked.cause
+			}, current];
+			if (request.admissionGroup !== void 0 && [...current.submissions.values()].some(({ row }) => row.threadId === request.threadId && row.admissionGroup === request.admissionGroup && row.state !== "settled")) return [failure(AdmissionPolicyError.make({
+				reason: "occupied",
+				code: "admission-group"
+			})), current];
 			if (current.submissions.size >= MAX_SUBMISSIONS) return [failure(ledgerError("admit", `In-memory submission limit ${MAX_SUBMISSIONS} exceeded`)), current];
 			const lane = current.lanes.get(request.threadId) ?? {
 				nextQueueSequence: 1,
@@ -209,7 +222,9 @@ const makeSubmissionLedger = (options = {}) => Effect.gen(function* () {
 				settledOutcome: void 0,
 				createdAtMillis: nowMillis,
 				readyAtMillis: void 0,
-				parentLinkage: request.parentLinkage
+				parentLinkage: request.parentLinkage,
+				...request.admissionGroup === void 0 ? {} : { admissionGroup: request.admissionGroup },
+				...request.admissionFence === void 0 ? {} : { admissionFence: request.admissionFence }
 			};
 			const submissions = new Map(current.submissions).set(row.submissionId, {
 				row,
@@ -243,8 +258,18 @@ const makeSubmissionLedger = (options = {}) => Effect.gen(function* () {
 			}];
 		});
 		if (decision._tag === "failure") return yield* decision.error;
+		if (decision._tag === "cause") {
+			for (const reason of decision.cause.reasons) if (Cause.isDieReason(reason) && Cause.isAsyncFiberError(reason.defect)) {
+				yield* Fiber.interrupt(reason.defect.fiber);
+				return yield* AdmissionPolicyError.make({
+					reason: "unavailable",
+					code: "synchronous-memory-policy-required"
+				});
+			}
+			return yield* Effect.failCause(decision.cause);
+		}
 		return decision.value;
-	}));
+	}), Effect.uninterruptible);
 	const markReady = Effect.fn("MemorySubmissionLedger.markReady")((unvalidated) => Effect.gen(function* () {
 		const request = yield* validate(MarkReadyRequest, "markReady", unvalidated);
 		const nowMillis = yield* Clock.currentTimeMillis;