@effect-agent/storage-memory 0.1.0-beta.9 → 0.1.0-beta.91

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (34) hide show
  1. package/dist/MemoryMessageDeliveryStore.d.mts +15 -0
  2. package/dist/MemoryMessageDeliveryStore.mjs +147 -0
  3. package/dist/MemoryMessageDeliveryStore.mjs.map +1 -0
  4. package/dist/MemoryScheduleStore.d.mts +10 -0
  5. package/dist/MemoryScheduleStore.mjs +169 -0
  6. package/dist/MemoryScheduleStore.mjs.map +1 -0
  7. package/dist/MemorySemanticIndex.d.mts +21 -0
  8. package/dist/MemorySemanticIndex.mjs +222 -0
  9. package/dist/MemorySemanticIndex.mjs.map +1 -0
  10. package/dist/MemorySubmissionLedger.d.mts +24 -0
  11. package/dist/MemorySubmissionLedger.mjs +1058 -0
  12. package/dist/MemorySubmissionLedger.mjs.map +1 -0
  13. package/dist/MemorySubscriptionStore.d.mts +9 -0
  14. package/dist/MemorySubscriptionStore.mjs +849 -0
  15. package/dist/MemorySubscriptionStore.mjs.map +1 -0
  16. package/dist/MemoryThreadStore.d.mts +17 -0
  17. package/dist/MemoryThreadStore.mjs +377 -0
  18. package/dist/MemoryThreadStore.mjs.map +1 -0
  19. package/dist/index.d.mts +7 -30
  20. package/dist/index.mjs +7 -1298
  21. package/dist/rolldown-runtime-D7D4PA-g.mjs +13 -0
  22. package/package.json +1 -45
  23. package/src/MemoryMessageDeliveryStore.ts +293 -0
  24. package/src/MemoryScheduleStore.ts +328 -0
  25. package/src/MemorySemanticIndex.ts +357 -0
  26. package/src/{memory-ledger.ts → MemorySubmissionLedger.ts} +478 -86
  27. package/src/MemorySubscriptionStore.ts +1549 -0
  28. package/src/MemoryThreadStore.ts +766 -0
  29. package/src/index.ts +6 -2
  30. package/dist/index.mjs.map +0 -1
  31. package/dist/testing.d.mts +0 -2
  32. package/dist/testing.mjs +0 -2
  33. package/src/memory-storage.ts +0 -614
  34. package/src/testing.ts +0 -10
@@ -0,0 +1,13 @@
1
+ //#region \0rolldown/runtime.js
2
+ var __defProp = Object.defineProperty;
3
+ var __exportAll = (all, no_symbols) => {
4
+ let target = {};
5
+ for (var name in all) __defProp(target, name, {
6
+ get: all[name],
7
+ enumerable: true
8
+ });
9
+ if (!no_symbols) __defProp(target, Symbol.toStringTag, { value: "Module" });
10
+ return target;
11
+ };
12
+ //#endregion
13
+ export { __exportAll as t };
package/package.json CHANGED
@@ -1,45 +1 @@
1
- {
2
- "name": "@effect-agent/storage-memory",
3
- "version": "0.1.0-beta.9",
4
- "exports": {
5
- ".": {
6
- "types": "./dist/index.d.mts",
7
- "default": "./dist/index.mjs"
8
- },
9
- "./testing": {
10
- "types": "./dist/testing.d.mts",
11
- "default": "./dist/testing.mjs"
12
- }
13
- },
14
- "dependencies": {
15
- "@effect-agent/core": "0.1.0-beta.9",
16
- "@effect-agent/session": "0.1.0-beta.9",
17
- "effect": "4.0.0-beta.107"
18
- },
19
- "description": "In-memory Conversation store and submission ledger reference adapters for Effect Agent tests and conformance suites.",
20
- "license": "MIT",
21
- "repository": {
22
- "type": "git",
23
- "url": "git+https://github.com/danieljvdm/effect-agent.git",
24
- "directory": "packages/storage-memory"
25
- },
26
- "files": [
27
- "dist",
28
- "src"
29
- ],
30
- "type": "module",
31
- "publishConfig": {
32
- "access": "public"
33
- },
34
- "scripts": {
35
- "build": "vp pack",
36
- "check": "tsc --noEmit -p tsconfig.json",
37
- "test": "vp test --passWithNoTests"
38
- },
39
- "devDependencies": {
40
- "@effect/platform-node": "4.0.0-beta.107",
41
- "@effect/vitest": "4.0.0-beta.107",
42
- "typescript": "7.0.2",
43
- "vite-plus": "0.2.6"
44
- }
45
- }
1
+ {"name":"@effect-agent/storage-memory","version":"0.1.0-beta.91","dependencies":{"effect-agent":"0.1.0-beta.91"},"devDependencies":{"@effect/platform-node":"4.0.0-rc.112","@effect/vitest":"4.0.0-rc.112","effect":"4.0.0-rc.112","typescript":"7.0.2","vite-plus":"0.3.0"},"peerDependencies":{"effect":"^4.0.0-rc.112"},"exports":{".":{"types":"./dist/index.d.mts","default":"./dist/index.mjs"},"./memory-message-delivery-store":{"types":"./dist/MemoryMessageDeliveryStore.d.mts","default":"./dist/MemoryMessageDeliveryStore.mjs"},"./memory-schedule-store":{"types":"./dist/MemoryScheduleStore.d.mts","default":"./dist/MemoryScheduleStore.mjs"},"./memory-semantic-index":{"types":"./dist/MemorySemanticIndex.d.mts","default":"./dist/MemorySemanticIndex.mjs"},"./memory-submission-ledger":{"types":"./dist/MemorySubmissionLedger.d.mts","default":"./dist/MemorySubmissionLedger.mjs"},"./memory-subscription-store":{"types":"./dist/MemorySubscriptionStore.d.mts","default":"./dist/MemorySubscriptionStore.mjs"},"./memory-thread-store":{"types":"./dist/MemoryThreadStore.d.mts","default":"./dist/MemoryThreadStore.mjs"}},"description":"In-memory Thread store and submission ledger reference adapters for Effect Agent tests and conformance suites.","license":"MIT","repository":{"type":"git","url":"git+https://github.com/danieljvdm/effect-agent.git","directory":"packages/storage-memory"},"files":["dist","src"],"type":"module","sideEffects":[],"publishConfig":{"access":"public"},"scripts":{"build":"vp pack","check":"tsc --noEmit -p tsconfig.json","test":"vp test --passWithNoTests"}}
@@ -0,0 +1,293 @@
1
+ import { Effect, Layer, Ref, Schema, Semaphore } from "effect";
2
+ import { ThreadId } from "effect-agent/identifiers";
3
+ import {
4
+ applyMessageDeliveryChange,
5
+ defaultMessageDeliveryStoreLimits,
6
+ MessageDeliveryChange,
7
+ MessageDeliveryError,
8
+ MessageDeliveryFailpoint,
9
+ MessageDeliveryKey,
10
+ MessageDeliveryPageRequest,
11
+ MessageDeliveryRecord,
12
+ MessageDeliveryStore,
13
+ MessageDeliveryStoreLimits,
14
+ messageDeliveryDeadline,
15
+ messageDeliveryKeyString,
16
+ messageDeliveryUsesCapacity,
17
+ isWorkerUpdateDelivery,
18
+ messageDeliveryCapacity,
19
+ sameMessageDeliveryIdentity,
20
+ validateMessageDelivery,
21
+ } from "effect-agent/message-delivery";
22
+ import { ScheduleInstant } from "effect-agent/schedule";
23
+
24
+ const codec = Schema.fromJsonString(MessageDeliveryRecord);
25
+
26
+ const decode = (text: string) =>
27
+ Schema.decodeEffect(codec)(text).pipe(
28
+ Effect.mapError(() => MessageDeliveryError.make({ reason: "corrupt", operation: "decode" })),
29
+ );
30
+
31
+ const Scan = Schema.Struct({
32
+ nowMillis: ScheduleInstant,
33
+ limit: Schema.Int.check(Schema.isGreaterThan(0), Schema.isLessThanOrEqualTo(100)),
34
+ ownerThreadId: Schema.optionalKey(ThreadId),
35
+ });
36
+
37
+ export interface MemoryMessageDeliveryStoreOptions {
38
+ /** UTF-8 bound on the complete stored record, including a processed Settlement. */
39
+ readonly maxStoredValueBytes?: number;
40
+ }
41
+
42
+ /** Reference adapter. All retained rows are bounded; completed deduplication evidence is never evicted. */
43
+ export const memoryMessageDeliveryStoreLayer = (
44
+ limits: MessageDeliveryStoreLimits = defaultMessageDeliveryStoreLimits,
45
+ options: MemoryMessageDeliveryStoreOptions = {},
46
+ ): Layer.Layer<MessageDeliveryStore, MessageDeliveryError> =>
47
+ Layer.effect(
48
+ MessageDeliveryStore,
49
+ Effect.gen(function* () {
50
+ const config = yield* validateMessageDelivery(MessageDeliveryStoreLimits, limits, "limits");
51
+
52
+ const maxStoredValueBytes = yield* validateMessageDelivery(
53
+ Schema.Int.check(Schema.isGreaterThan(0)),
54
+ options.maxStoredValueBytes ?? 16 * 1_024 * 1_024,
55
+ "stored-value-limit",
56
+ );
57
+
58
+ const state = yield* Ref.make<ReadonlyMap<string, string>>(new Map());
59
+ const lock = yield* Semaphore.make(1);
60
+ const failpoint = yield* MessageDeliveryFailpoint;
61
+
62
+ const encode = Effect.fn("MemoryMessageDeliveryStore.encode")(function* (
63
+ record: MessageDeliveryRecord,
64
+ ) {
65
+ const text = yield* Schema.encodeEffect(codec)(record).pipe(
66
+ Effect.mapError(() =>
67
+ MessageDeliveryError.make({ reason: "corrupt", operation: "encode" }),
68
+ ),
69
+ );
70
+
71
+ if (new TextEncoder().encode(text).byteLength > maxStoredValueBytes) {
72
+ return yield* MessageDeliveryError.make({
73
+ reason: "capacity",
74
+ operation: "stored-value-bytes",
75
+ });
76
+ }
77
+
78
+ return text;
79
+ });
80
+
81
+ const all = Effect.fn("MemoryMessageDeliveryStore.all")(function* () {
82
+ return yield* Effect.forEach((yield* Ref.get(state)).values(), decode);
83
+ });
84
+
85
+ const insert: MessageDeliveryStore["Service"]["insert"] = Effect.fn(
86
+ "MemoryMessageDeliveryStore.insert",
87
+ )(function* (record) {
88
+ const encoded = yield* encode(record);
89
+ const input = yield* decode(encoded);
90
+
91
+ if (
92
+ input.version !== 1 ||
93
+ input.status !== "pending" ||
94
+ input.leaseUntilMillis !== null ||
95
+ input.retry.attempts !== 0 ||
96
+ input.retry.generation !== 0 ||
97
+ input.retry.automaticAttempts !== 0 ||
98
+ input.retry.nextAttemptAtMillis !== input.createdAtMillis ||
99
+ input.retry.lastAttemptAtMillis !== null ||
100
+ input.retry.lastFailure !== null ||
101
+ input.deadlineAtMillis !== input.initialDeadlineAtMillis
102
+ ) {
103
+ return yield* MessageDeliveryError.make({
104
+ reason: "validation",
105
+ operation: "insert-state",
106
+ });
107
+ }
108
+ if (
109
+ new TextEncoder().encode(JSON.stringify(input.envelope)).byteLength >
110
+ config.maxEnvelopeBytes
111
+ ) {
112
+ return yield* MessageDeliveryError.make({
113
+ reason: "capacity",
114
+ operation: "envelope-bytes",
115
+ });
116
+ }
117
+ yield* failpoint.hit("message-delivery:insert:before");
118
+
119
+ const inserted = yield* lock.withPermit(
120
+ Effect.uninterruptible(
121
+ Effect.gen(function* () {
122
+ const records = yield* Ref.get(state);
123
+ const key = messageDeliveryKeyString(input.key);
124
+ const existingText = records.get(key);
125
+
126
+ if (existingText !== undefined) {
127
+ const existing = yield* decode(existingText);
128
+
129
+ if (!sameMessageDeliveryIdentity(existing, input))
130
+ return yield* MessageDeliveryError.make({
131
+ reason: "conflict",
132
+ operation: "insert",
133
+ });
134
+
135
+ return existing;
136
+ }
137
+
138
+ const owned = (yield* all()).filter(
139
+ (record) =>
140
+ record.key.ownerThreadId === input.key.ownerThreadId &&
141
+ isWorkerUpdateDelivery(record) === isWorkerUpdateDelivery(input),
142
+ );
143
+
144
+ const capacity = messageDeliveryCapacity(config, isWorkerUpdateDelivery(input));
145
+
146
+ if (
147
+ owned.length >= capacity.retained ||
148
+ owned.filter(messageDeliveryUsesCapacity).length >= capacity.pending
149
+ ) {
150
+ return yield* MessageDeliveryError.make({
151
+ reason: "capacity",
152
+ operation: "insert",
153
+ });
154
+ }
155
+ yield* Ref.set(state, new Map(records).set(key, encoded));
156
+
157
+ return input;
158
+ }),
159
+ ),
160
+ );
161
+
162
+ yield* failpoint.hit("message-delivery:insert:after");
163
+
164
+ return yield* decode(yield* encode(inserted));
165
+ });
166
+
167
+ const get: MessageDeliveryStore["Service"]["get"] = Effect.fn(
168
+ "MemoryMessageDeliveryStore.get",
169
+ )(function* (key) {
170
+ const input = yield* validateMessageDelivery(MessageDeliveryKey, key, "get");
171
+ const text = (yield* Ref.get(state)).get(messageDeliveryKeyString(input));
172
+
173
+ return text === undefined ? null : yield* decode(text);
174
+ });
175
+
176
+ const change: MessageDeliveryStore["Service"]["change"] = Effect.fn(
177
+ "MemoryMessageDeliveryStore.change",
178
+ )(function* (key, change) {
179
+ const input = yield* validateMessageDelivery(MessageDeliveryChange, change, "change");
180
+ const decodedKey = yield* validateMessageDelivery(MessageDeliveryKey, key, "change");
181
+ const point = `message-delivery:${input._tag.toLowerCase()}`;
182
+
183
+ yield* failpoint.hit(`${point}:before`);
184
+
185
+ const changed = yield* lock.withPermit(
186
+ Effect.uninterruptible(
187
+ Effect.gen(function* () {
188
+ const existing = yield* get(decodedKey);
189
+
190
+ if (existing === null)
191
+ return yield* MessageDeliveryError.make({
192
+ reason: "not-found",
193
+ operation: "change",
194
+ });
195
+ const next = yield* Effect.fromResult(applyMessageDeliveryChange(existing, input));
196
+ const encoded = yield* encode(next);
197
+
198
+ yield* Ref.update(state, (records) =>
199
+ new Map(records).set(messageDeliveryKeyString(decodedKey), encoded),
200
+ );
201
+
202
+ return next;
203
+ }),
204
+ ),
205
+ );
206
+
207
+ yield* failpoint.hit(`${point}:after`);
208
+
209
+ return yield* decode(yield* encode(changed));
210
+ });
211
+
212
+ return MessageDeliveryStore.of({
213
+ limits: config,
214
+ maxStoredValueBytes,
215
+ insert,
216
+ get,
217
+ change,
218
+ list: Effect.fn("MemoryMessageDeliveryStore.list")(function* (request) {
219
+ const input = yield* validateMessageDelivery(MessageDeliveryPageRequest, request, "list");
220
+
221
+ const records = (yield* all())
222
+ .filter(
223
+ (record) =>
224
+ record.key.ownerThreadId === input.ownerThreadId &&
225
+ (input.after === undefined || record.key.messageId > input.after),
226
+ )
227
+ .sort((left, right) =>
228
+ left.key.messageId < right.key.messageId
229
+ ? -1
230
+ : left.key.messageId > right.key.messageId
231
+ ? 1
232
+ : 0,
233
+ );
234
+
235
+ const items = records.slice(0, input.limit);
236
+
237
+ return {
238
+ items,
239
+ next: records.length > input.limit ? (items.at(-1)?.key.messageId ?? null) : null,
240
+ };
241
+ }),
242
+ due: Effect.fn("MemoryMessageDeliveryStore.due")(
243
+ function* (nowMillis, limit, ownerThreadId) {
244
+ const input = yield* validateMessageDelivery(
245
+ Scan,
246
+ { nowMillis, limit, ...(ownerThreadId === undefined ? {} : { ownerThreadId }) },
247
+ "due",
248
+ );
249
+
250
+ return (yield* all())
251
+ .flatMap((record) => {
252
+ const deadline = messageDeliveryDeadline(record);
253
+
254
+ return deadline !== null &&
255
+ deadline <= input.nowMillis &&
256
+ (input.ownerThreadId === undefined ||
257
+ input.ownerThreadId === record.key.ownerThreadId)
258
+ ? [{ key: record.key, deadline }]
259
+ : [];
260
+ })
261
+ .sort(
262
+ (left, right) =>
263
+ left.deadline - right.deadline ||
264
+ (messageDeliveryKeyString(left.key) < messageDeliveryKeyString(right.key)
265
+ ? -1
266
+ : 1),
267
+ )
268
+ .slice(0, input.limit)
269
+ .map((record) => record.key);
270
+ },
271
+ ),
272
+ nextDeadline: Effect.fn("MemoryMessageDeliveryStore.nextDeadline")(
273
+ function* (ownerThreadId) {
274
+ if (ownerThreadId !== undefined)
275
+ yield* validateMessageDelivery(ThreadId, ownerThreadId, "nextDeadline");
276
+ let next: number | null = null;
277
+
278
+ for (const record of yield* all()) {
279
+ if (ownerThreadId !== undefined && ownerThreadId !== record.key.ownerThreadId)
280
+ continue;
281
+ const deadline = messageDeliveryDeadline(record);
282
+
283
+ if (deadline !== null && (next === null || deadline < next)) next = deadline;
284
+ }
285
+
286
+ return next;
287
+ },
288
+ ),
289
+ });
290
+ }),
291
+ );
292
+
293
+ export const MemoryMessageDeliveryStoreLive = memoryMessageDeliveryStoreLayer();
@@ -0,0 +1,328 @@
1
+ import { Effect, Layer, Ref, Result, Schema } from "effect";
2
+ import {
3
+ ScheduleCapacityError,
4
+ ScheduleDueCursor,
5
+ defaultSchedulingLimits,
6
+ ScheduleChange,
7
+ ScheduleConflict,
8
+ ScheduleFailpoint,
9
+ ScheduleKey,
10
+ ScheduleNotFound,
11
+ ScheduleOwner,
12
+ SchedulePageRequest,
13
+ ScheduleRecord,
14
+ ScheduleStorageError,
15
+ ScheduleStore,
16
+ } from "effect-agent/schedule";
17
+ import {
18
+ scheduleUsesCapacity,
19
+ applyScheduleChange,
20
+ compareScheduleKeys,
21
+ compareScheduleNames,
22
+ scheduleDeadline,
23
+ scheduleKeyString,
24
+ scheduleKeyOf,
25
+ scheduleOwnerKey,
26
+ } from "effect-agent/schedule-transition";
27
+
28
+ interface MemoryScheduleState {
29
+ readonly records: ReadonlyMap<string, string>;
30
+ }
31
+
32
+ const storageError = (operation: string, reason: "unavailable" | "corrupt") =>
33
+ ScheduleStorageError.make({ operation, reason });
34
+
35
+ const encodeRecord = (operation: string, record: ScheduleRecord) =>
36
+ Effect.try({
37
+ try: () => Schema.encodeSync(Schema.fromJsonString(ScheduleRecord))(record),
38
+ catch: () => storageError(operation, "corrupt"),
39
+ });
40
+
41
+ const decodeRecord = (operation: string, encoded: string) =>
42
+ Effect.try({
43
+ try: () => Schema.decodeSync(Schema.fromJsonString(ScheduleRecord))(encoded),
44
+ catch: () => storageError(operation, "corrupt"),
45
+ });
46
+
47
+ const decodeInput = <A, I>(operation: string, schema: Schema.Codec<A, I>, value: unknown) =>
48
+ Effect.try({
49
+ try: () => Schema.decodeUnknownSync(schema)(value),
50
+ catch: () => storageError(operation, "corrupt"),
51
+ });
52
+
53
+ const sameOwner = (record: ScheduleRecord, owner: ScheduleOwner): boolean =>
54
+ scheduleOwnerKey(record.owner) === scheduleOwnerKey(owner);
55
+
56
+ const makeScheduleStore = Effect.gen(function* () {
57
+ const state = yield* Ref.make<MemoryScheduleState>({ records: new Map() });
58
+ const failpoint = yield* ScheduleFailpoint;
59
+
60
+ const insert: ScheduleStore["Service"]["insert"] = Effect.fn("MemoryScheduleStore.insert")(
61
+ (record, ownerLimit) =>
62
+ Effect.gen(function* () {
63
+ const encoded = yield* encodeRecord("insert", record);
64
+
65
+ yield* failpoint.hit("schedule:insert:before");
66
+
67
+ const decision = yield* Effect.uninterruptible(
68
+ Ref.modify(
69
+ state,
70
+ (
71
+ current,
72
+ ): readonly [
73
+ Result.Result<
74
+ ScheduleRecord,
75
+ ScheduleConflict | ScheduleCapacityError | ScheduleStorageError
76
+ >,
77
+ MemoryScheduleState,
78
+ ] => {
79
+ const key = scheduleKeyString(record);
80
+ const existingText = current.records.get(key);
81
+
82
+ if (existingText !== undefined) {
83
+ const decoded = Result.try({
84
+ try: () => Schema.decodeSync(Schema.fromJsonString(ScheduleRecord))(existingText),
85
+ catch: () => storageError("insert", "corrupt"),
86
+ });
87
+
88
+ if (Result.isFailure(decoded)) {
89
+ return [Result.fail(decoded.failure), current];
90
+ }
91
+ if (decoded.success.creationFingerprint === record.creationFingerprint) {
92
+ return [Result.succeed(decoded.success), current];
93
+ }
94
+
95
+ return [
96
+ Result.fail(
97
+ ScheduleConflict.make({ reason: "creation", key: scheduleKeyOf(record) }),
98
+ ),
99
+ current,
100
+ ];
101
+ }
102
+
103
+ let ownerCount = 0;
104
+
105
+ for (const text of current.records.values()) {
106
+ const decoded = Result.try({
107
+ try: () => Schema.decodeSync(Schema.fromJsonString(ScheduleRecord))(text),
108
+ catch: () => storageError("insert", "corrupt"),
109
+ });
110
+
111
+ if (Result.isFailure(decoded)) {
112
+ return [Result.fail(decoded.failure), current];
113
+ }
114
+ if (
115
+ sameOwner(decoded.success, record.owner) &&
116
+ scheduleUsesCapacity(decoded.success)
117
+ )
118
+ ownerCount += 1;
119
+ }
120
+ if (ownerCount >= ownerLimit) {
121
+ return [Result.fail(ScheduleCapacityError.make({ limit: ownerLimit })), current];
122
+ }
123
+ const records = new Map(current.records);
124
+
125
+ records.set(key, encoded);
126
+ const next = { records };
127
+
128
+ return [Result.succeed(record), next];
129
+ },
130
+ ),
131
+ );
132
+
133
+ const inserted = yield* Effect.fromResult(decision);
134
+
135
+ yield* failpoint.hit("schedule:insert:after");
136
+
137
+ return yield* decodeRecord("insert", yield* encodeRecord("insert", inserted));
138
+ }),
139
+ );
140
+
141
+ const get: ScheduleStore["Service"]["get"] = Effect.fn("MemoryScheduleStore.get")(
142
+ function* (key) {
143
+ const decodedKey = yield* decodeInput("get", ScheduleKey, key);
144
+ const text = (yield* Ref.get(state)).records.get(scheduleKeyString(decodedKey));
145
+
146
+ return text === undefined ? null : yield* decodeRecord("get", text);
147
+ },
148
+ );
149
+
150
+ const list: ScheduleStore["Service"]["list"] = Effect.fn("MemoryScheduleStore.list")(
151
+ function* (request) {
152
+ const decodedRequest = yield* decodeInput("list", SchedulePageRequest, request);
153
+ const records: Array<ScheduleRecord> = [];
154
+
155
+ for (const text of (yield* Ref.get(state)).records.values()) {
156
+ const record = yield* decodeRecord("list", text);
157
+
158
+ if (
159
+ sameOwner(record, decodedRequest.owner) &&
160
+ (decodedRequest.after === undefined ||
161
+ compareScheduleNames(record.scheduleId, decodedRequest.after) > 0)
162
+ ) {
163
+ records.push(record);
164
+ }
165
+ }
166
+ records.sort((left, right) => compareScheduleNames(left.scheduleId, right.scheduleId));
167
+ const hasNext = records.length > decodedRequest.limit;
168
+ const items = records.slice(0, decodedRequest.limit);
169
+
170
+ return {
171
+ items,
172
+ next: hasNext ? (items.at(-1)?.scheduleId ?? null) : null,
173
+ };
174
+ },
175
+ );
176
+
177
+ const change: ScheduleStore["Service"]["change"] = Effect.fn("MemoryScheduleStore.change")(
178
+ (key, command, ownerLimit = defaultSchedulingLimits.maxSchedulesPerOwner) =>
179
+ Effect.gen(function* () {
180
+ const decodedKey = yield* decodeInput("change", ScheduleKey, key);
181
+ const decodedCommand = yield* decodeInput("change", ScheduleChange, command);
182
+
183
+ yield* failpoint.hit(`schedule:${decodedCommand._tag.toLowerCase()}:before`);
184
+
185
+ const decision = yield* Effect.uninterruptible(
186
+ Ref.modify(
187
+ state,
188
+ (
189
+ current,
190
+ ): readonly [
191
+ Result.Result<
192
+ ScheduleRecord,
193
+ ScheduleConflict | ScheduleStorageError | ScheduleNotFound | ScheduleCapacityError
194
+ >,
195
+ MemoryScheduleState,
196
+ ] => {
197
+ const storageKey = scheduleKeyString(decodedKey);
198
+ const text = current.records.get(storageKey);
199
+
200
+ if (text === undefined) {
201
+ return [Result.fail(ScheduleNotFound.make({ key: decodedKey })), current];
202
+ }
203
+
204
+ const decoded = Result.try({
205
+ try: () => Schema.decodeSync(Schema.fromJsonString(ScheduleRecord))(text),
206
+ catch: () => storageError("change", "corrupt"),
207
+ });
208
+
209
+ if (Result.isFailure(decoded)) {
210
+ return [Result.fail(decoded.failure), current];
211
+ }
212
+ const applied = applyScheduleChange(decoded.success, decodedCommand);
213
+
214
+ if (Result.isFailure(applied)) {
215
+ return [Result.fail(applied.failure), current];
216
+ }
217
+ if (applied.success === decoded.success) {
218
+ return [Result.succeed(decoded.success), current];
219
+ }
220
+ if (!scheduleUsesCapacity(decoded.success) && scheduleUsesCapacity(applied.success)) {
221
+ let count = 0;
222
+
223
+ for (const text of current.records.values()) {
224
+ const candidate = Schema.decodeResult(Schema.fromJsonString(ScheduleRecord))(
225
+ text,
226
+ );
227
+
228
+ if (Result.isFailure(candidate))
229
+ return [Result.fail(storageError("change", "corrupt")), current];
230
+ if (
231
+ sameOwner(candidate.success, decodedKey.owner) &&
232
+ scheduleUsesCapacity(candidate.success)
233
+ )
234
+ count += 1;
235
+ }
236
+ if (count >= ownerLimit)
237
+ return [Result.fail(ScheduleCapacityError.make({ limit: ownerLimit })), current];
238
+ }
239
+
240
+ const encoded = Result.try({
241
+ try: () =>
242
+ Schema.encodeSync(Schema.fromJsonString(ScheduleRecord))(applied.success),
243
+ catch: () => storageError("change", "corrupt"),
244
+ });
245
+
246
+ if (Result.isFailure(encoded)) {
247
+ return [Result.fail(encoded.failure), current];
248
+ }
249
+ const records = new Map(current.records);
250
+
251
+ records.set(storageKey, encoded.success);
252
+ const next = { records };
253
+
254
+ return [Result.succeed(applied.success), next];
255
+ },
256
+ ),
257
+ );
258
+
259
+ const changed = yield* Effect.fromResult(decision);
260
+
261
+ yield* failpoint.hit(`schedule:${decodedCommand._tag.toLowerCase()}:after`);
262
+
263
+ return yield* decodeRecord("change", yield* encodeRecord("change", changed));
264
+ }),
265
+ );
266
+
267
+ const due: ScheduleStore["Service"]["due"] = Effect.fn("MemoryScheduleStore.due")(
268
+ function* (nowMillis, limit, owner, after) {
269
+ const decodedOwner =
270
+ owner === undefined ? undefined : yield* decodeInput("due", ScheduleOwner, owner);
271
+
272
+ const cursor =
273
+ after === undefined ? undefined : yield* decodeInput("due", ScheduleDueCursor, after);
274
+
275
+ const records: Array<ScheduleDueCursor> = [];
276
+
277
+ for (const text of (yield* Ref.get(state)).records.values()) {
278
+ const record = yield* decodeRecord("due", text);
279
+ const deadline = scheduleDeadline(record);
280
+
281
+ if (
282
+ deadline !== null &&
283
+ deadline <= nowMillis &&
284
+ (decodedOwner === undefined || sameOwner(record, decodedOwner)) &&
285
+ (cursor === undefined ||
286
+ deadline > cursor.deadlineAtMillis ||
287
+ (deadline === cursor.deadlineAtMillis && compareScheduleKeys(record, cursor) > 0))
288
+ ) {
289
+ records.push({ ...scheduleKeyOf(record), deadlineAtMillis: deadline });
290
+ }
291
+ }
292
+ records.sort((left, right) => {
293
+ const byDeadline = left.deadlineAtMillis - right.deadlineAtMillis;
294
+
295
+ return byDeadline !== 0 ? byDeadline : compareScheduleKeys(left, right);
296
+ });
297
+
298
+ return records.slice(0, limit);
299
+ },
300
+ );
301
+
302
+ const nextDeadline: ScheduleStore["Service"]["nextDeadline"] = Effect.fn(
303
+ "MemoryScheduleStore.nextDeadline",
304
+ )(function* (owner) {
305
+ const decodedOwner =
306
+ owner === undefined ? undefined : yield* decodeInput("nextDeadline", ScheduleOwner, owner);
307
+
308
+ let earliest: number | null = null;
309
+
310
+ for (const text of (yield* Ref.get(state)).records.values()) {
311
+ const record = yield* decodeRecord("nextDeadline", text);
312
+
313
+ if (decodedOwner !== undefined && !sameOwner(record, decodedOwner)) continue;
314
+ const deadline = scheduleDeadline(record);
315
+
316
+ if (deadline !== null && (earliest === null || deadline < earliest)) earliest = deadline;
317
+ }
318
+
319
+ return earliest;
320
+ });
321
+
322
+ return ScheduleStore.of({ insert, get, list, change, due, nextDeadline });
323
+ });
324
+
325
+ export const memoryScheduleStoreLayer = (): Layer.Layer<ScheduleStore> =>
326
+ Layer.effect(ScheduleStore, makeScheduleStore);
327
+
328
+ export const MemoryScheduleStoreLive: Layer.Layer<ScheduleStore> = memoryScheduleStoreLayer();