@aztec/world-state 0.0.1-commit.993d52e → 0.0.1-commit.9a89641

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 (76) hide show
  1. package/dest/index.d.ts +2 -1
  2. package/dest/index.d.ts.map +1 -1
  3. package/dest/index.js +1 -0
  4. package/dest/instrumentation/instrumentation.d.ts +4 -3
  5. package/dest/instrumentation/instrumentation.d.ts.map +1 -1
  6. package/dest/instrumentation/instrumentation.js +4 -9
  7. package/dest/native/fork_checkpoint.d.ts +7 -1
  8. package/dest/native/fork_checkpoint.d.ts.map +1 -1
  9. package/dest/native/fork_checkpoint.js +15 -3
  10. package/dest/native/index.d.ts +2 -1
  11. package/dest/native/index.d.ts.map +1 -1
  12. package/dest/native/index.js +1 -0
  13. package/dest/native/ipc_world_state_instance.d.ts +89 -0
  14. package/dest/native/ipc_world_state_instance.d.ts.map +1 -0
  15. package/dest/native/ipc_world_state_instance.js +670 -0
  16. package/dest/native/merkle_trees_facade.d.ts +11 -7
  17. package/dest/native/merkle_trees_facade.d.ts.map +1 -1
  18. package/dest/native/merkle_trees_facade.js +62 -114
  19. package/dest/native/message.d.ts +14 -222
  20. package/dest/native/message.d.ts.map +1 -1
  21. package/dest/native/message.js +0 -42
  22. package/dest/native/native_world_state.d.ts +19 -7
  23. package/dest/native/native_world_state.d.ts.map +1 -1
  24. package/dest/native/native_world_state.js +120 -58
  25. package/dest/native/native_world_state_instance.d.ts +56 -42
  26. package/dest/native/native_world_state_instance.d.ts.map +1 -1
  27. package/dest/native/native_world_state_instance.js +1 -203
  28. package/dest/native/world_state_operation.d.ts +7 -0
  29. package/dest/native/world_state_operation.d.ts.map +1 -0
  30. package/dest/native/world_state_operation.js +5 -0
  31. package/dest/synchronizer/config.d.ts +1 -1
  32. package/dest/synchronizer/config.d.ts.map +1 -1
  33. package/dest/synchronizer/config.js +9 -10
  34. package/dest/synchronizer/errors.d.ts +8 -1
  35. package/dest/synchronizer/errors.d.ts.map +1 -1
  36. package/dest/synchronizer/errors.js +8 -1
  37. package/dest/synchronizer/factory.d.ts +5 -5
  38. package/dest/synchronizer/factory.d.ts.map +1 -1
  39. package/dest/synchronizer/factory.js +7 -6
  40. package/dest/synchronizer/index.d.ts +2 -1
  41. package/dest/synchronizer/index.d.ts.map +1 -1
  42. package/dest/synchronizer/index.js +1 -0
  43. package/dest/synchronizer/server_world_state_synchronizer.d.ts +13 -8
  44. package/dest/synchronizer/server_world_state_synchronizer.d.ts.map +1 -1
  45. package/dest/synchronizer/server_world_state_synchronizer.js +109 -45
  46. package/dest/test/utils.d.ts +1 -1
  47. package/dest/test/utils.d.ts.map +1 -1
  48. package/dest/test/utils.js +6 -1
  49. package/dest/testing.d.ts +4 -3
  50. package/dest/testing.d.ts.map +1 -1
  51. package/dest/testing.js +22 -7
  52. package/dest/world-state-db/merkle_tree_db.d.ts +1 -10
  53. package/dest/world-state-db/merkle_tree_db.d.ts.map +1 -1
  54. package/package.json +13 -11
  55. package/src/index.ts +1 -0
  56. package/src/instrumentation/instrumentation.ts +6 -18
  57. package/src/native/fork_checkpoint.ts +19 -3
  58. package/src/native/index.ts +1 -0
  59. package/src/native/ipc_world_state_instance.ts +847 -0
  60. package/src/native/merkle_trees_facade.ts +88 -108
  61. package/src/native/message.ts +13 -287
  62. package/src/native/native_world_state.ts +144 -97
  63. package/src/native/native_world_state_instance.ts +94 -281
  64. package/src/native/world_state_operation.ts +33 -0
  65. package/src/synchronizer/config.ts +14 -10
  66. package/src/synchronizer/errors.ts +8 -0
  67. package/src/synchronizer/factory.ts +12 -11
  68. package/src/synchronizer/index.ts +1 -0
  69. package/src/synchronizer/server_world_state_synchronizer.ts +127 -44
  70. package/src/test/utils.ts +10 -1
  71. package/src/testing.ts +19 -10
  72. package/src/world-state-db/merkle_tree_db.ts +0 -10
  73. package/dest/native/world_state_ops_queue.d.ts +0 -19
  74. package/dest/native/world_state_ops_queue.d.ts.map +0 -1
  75. package/dest/native/world_state_ops_queue.js +0 -146
  76. package/src/native/world_state_ops_queue.ts +0 -190
@@ -1,291 +1,104 @@
1
- import {
2
- ARCHIVE_HEIGHT,
3
- DomainSeparator,
4
- L1_TO_L2_MSG_TREE_HEIGHT,
5
- MAX_NULLIFIERS_PER_TX,
6
- MAX_TOTAL_PUBLIC_DATA_UPDATE_REQUESTS_PER_TX,
7
- NOTE_HASH_TREE_HEIGHT,
8
- NULLIFIER_TREE_HEIGHT,
9
- PUBLIC_DATA_TREE_HEIGHT,
10
- } from '@aztec/constants';
11
- import { type Logger, type LoggerBindings, createLogger } from '@aztec/foundation/log';
12
- import { NativeWorldState as BaseNativeWorldState, MsgpackChannel } from '@aztec/native';
13
- import { MerkleTreeId } from '@aztec/stdlib/trees';
14
- import type { PublicDataTreeLeaf } from '@aztec/stdlib/trees';
15
-
16
- import assert from 'assert';
17
- import { cpus } from 'os';
18
-
19
- import type { WorldStateInstrumentation } from '../instrumentation/instrumentation.js';
20
- import type { WorldStateTreeMapSizes } from '../synchronizer/factory.js';
21
- import {
22
- WorldStateMessageType,
23
- type WorldStateRequest,
24
- type WorldStateRequestCategories,
25
- type WorldStateResponse,
26
- isWithCanonical,
27
- isWithForkId,
28
- isWithRevision,
1
+ import type { BlockNumber } from '@aztec/foundation/branded-types';
2
+ import type { Fr } from '@aztec/foundation/curves/bn254';
3
+ import type { IndexedTreeId, TreeInfo } from '@aztec/stdlib/interfaces/server';
4
+ import type { MerkleTreeId } from '@aztec/stdlib/trees';
5
+ import type { WorldStateRevision } from '@aztec/stdlib/world-state';
6
+
7
+ import type {
8
+ BlockStateReference,
9
+ SerializedBatchInsertionResult,
10
+ SerializedIndexedLeaf,
11
+ SerializedLeafValue,
12
+ SerializedSequentialInsertionResult,
13
+ SerializedSiblingPathAndIndex,
14
+ TreeStateReference,
15
+ WorldStateStatusFull,
16
+ WorldStateStatusSummary,
29
17
  } from './message.js';
30
- import { WorldStateOpsQueue } from './world_state_ops_queue.js';
31
-
32
- const MAX_WORLD_STATE_THREADS = +(process.env.HARDWARE_CONCURRENCY || '16');
33
18
 
19
+ /** Backend-agnostic handle to a running aztec-wsdb world state, accessed by the TS layer. */
34
20
  export interface NativeWorldStateInstance {
35
- call<T extends WorldStateMessageType>(
36
- messageType: T,
37
- body: WorldStateRequest[T] & WorldStateRequestCategories,
38
- ): Promise<WorldStateResponse[T]>;
39
- // TODO(dbanks12): this returns any type, but we should strongly type it
40
- getHandle(): any;
41
- }
42
-
43
- /**
44
- * Strongly-typed interface to access the WorldState class in the native world_state_napi module.
45
- */
46
- export class NativeWorldState implements NativeWorldStateInstance {
47
- private open = true;
48
-
49
- // We maintain a map of queue to fork
50
- private queues = new Map<number, WorldStateOpsQueue>();
51
-
52
- private instance: MsgpackChannel<WorldStateMessageType, WorldStateRequest, WorldStateResponse>;
53
-
54
- /** Creates a new native WorldState instance */
55
- constructor(
56
- private readonly dataDir: string,
57
- private readonly wsTreeMapSizes: WorldStateTreeMapSizes,
58
- private readonly prefilledPublicData: PublicDataTreeLeaf[] = [],
59
- private readonly instrumentation: WorldStateInstrumentation,
60
- bindings?: LoggerBindings,
61
- private readonly log: Logger = createLogger('world-state:database', bindings),
62
- ) {
63
- const threads = Math.min(cpus().length, MAX_WORLD_STATE_THREADS);
64
- log.info(
65
- `Creating world state data store at directory ${dataDir} with map sizes ${JSON.stringify(
66
- wsTreeMapSizes,
67
- )} and ${threads} threads.`,
68
- );
69
- const prefilledPublicDataBufferArray = prefilledPublicData.map(d => [d.slot.toBuffer(), d.value.toBuffer()]);
70
- const ws = new BaseNativeWorldState(
71
- dataDir,
72
- {
73
- [MerkleTreeId.NULLIFIER_TREE]: NULLIFIER_TREE_HEIGHT,
74
- [MerkleTreeId.NOTE_HASH_TREE]: NOTE_HASH_TREE_HEIGHT,
75
- [MerkleTreeId.PUBLIC_DATA_TREE]: PUBLIC_DATA_TREE_HEIGHT,
76
- [MerkleTreeId.L1_TO_L2_MESSAGE_TREE]: L1_TO_L2_MSG_TREE_HEIGHT,
77
- [MerkleTreeId.ARCHIVE]: ARCHIVE_HEIGHT,
78
- },
79
- {
80
- [MerkleTreeId.NULLIFIER_TREE]: 2 * MAX_NULLIFIERS_PER_TX,
81
- [MerkleTreeId.PUBLIC_DATA_TREE]: 2 * MAX_TOTAL_PUBLIC_DATA_UPDATE_REQUESTS_PER_TX,
82
- },
83
- prefilledPublicDataBufferArray,
84
- DomainSeparator.BLOCK_HEADER_HASH,
85
- {
86
- [MerkleTreeId.NULLIFIER_TREE]: wsTreeMapSizes.nullifierTreeMapSizeKb,
87
- [MerkleTreeId.NOTE_HASH_TREE]: wsTreeMapSizes.noteHashTreeMapSizeKb,
88
- [MerkleTreeId.PUBLIC_DATA_TREE]: wsTreeMapSizes.publicDataTreeMapSizeKb,
89
- [MerkleTreeId.L1_TO_L2_MESSAGE_TREE]: wsTreeMapSizes.messageTreeMapSizeKb,
90
- [MerkleTreeId.ARCHIVE]: wsTreeMapSizes.archiveTreeMapSizeKb,
91
- },
92
- threads,
93
- );
94
- this.instance = new MsgpackChannel(ws);
95
- // Manually create the queue for the canonical fork
96
- this.queues.set(0, new WorldStateOpsQueue());
97
- }
98
-
99
- public getDataDir() {
100
- return this.dataDir;
101
- }
102
-
103
- public clone() {
104
- return new NativeWorldState(
105
- this.dataDir,
106
- this.wsTreeMapSizes,
107
- this.prefilledPublicData,
108
- this.instrumentation,
109
- this.log.getBindings(),
110
- this.log,
111
- );
112
- }
21
+ getTreeInfo(treeId: MerkleTreeId, revision: WorldStateRevision): Promise<TreeInfo>;
22
+ getStateReference(revision: WorldStateRevision): Promise<Record<number, TreeStateReference>>;
23
+ getInitialStateReference(): Promise<Record<number, TreeStateReference>>;
24
+ getLeafValue(
25
+ treeId: MerkleTreeId,
26
+ revision: WorldStateRevision,
27
+ leafIndex: bigint,
28
+ ): Promise<SerializedLeafValue | undefined>;
29
+ getLeafPreimage(
30
+ treeId: IndexedTreeId,
31
+ revision: WorldStateRevision,
32
+ leafIndex: bigint,
33
+ ): Promise<SerializedIndexedLeaf | undefined>;
34
+ getSiblingPath(treeId: MerkleTreeId, revision: WorldStateRevision, leafIndex: bigint): Promise<Buffer[]>;
35
+ getBlockNumbersForLeafIndices(
36
+ treeId: MerkleTreeId,
37
+ revision: WorldStateRevision,
38
+ leafIndices: bigint[],
39
+ ): Promise<(bigint | undefined)[]>;
40
+ findLeafIndices(
41
+ treeId: MerkleTreeId,
42
+ revision: WorldStateRevision,
43
+ leaves: SerializedLeafValue[],
44
+ startIndex: bigint,
45
+ ): Promise<(bigint | undefined)[]>;
46
+ findLowLeaf(
47
+ treeId: IndexedTreeId,
48
+ revision: WorldStateRevision,
49
+ key: Fr,
50
+ ): Promise<{ index: bigint; alreadyPresent: boolean }>;
51
+ findSiblingPaths(
52
+ treeId: MerkleTreeId,
53
+ revision: WorldStateRevision,
54
+ leaves: SerializedLeafValue[],
55
+ ): Promise<(SerializedSiblingPathAndIndex | undefined)[]>;
56
+ updateArchive(forkId: number, blockStateRef: BlockStateReference, blockHeaderHash: Buffer): Promise<void>;
57
+ appendLeaves(treeId: MerkleTreeId, forkId: number, leaves: SerializedLeafValue[]): Promise<void>;
58
+ batchInsert(
59
+ treeId: IndexedTreeId,
60
+ forkId: number,
61
+ leaves: SerializedLeafValue[],
62
+ subtreeDepth: number,
63
+ ): Promise<SerializedBatchInsertionResult>;
64
+ sequentialInsert(
65
+ treeId: IndexedTreeId,
66
+ forkId: number,
67
+ leaves: SerializedLeafValue[],
68
+ ): Promise<SerializedSequentialInsertionResult>;
69
+ syncBlock(input: {
70
+ blockNumber: BlockNumber;
71
+ blockStateRef: BlockStateReference;
72
+ blockHeaderHash: Buffer;
73
+ expectedArchiveRoot: Buffer;
74
+ expectedPreviousArchiveRoot: Buffer;
75
+ paddedNoteHashes: readonly SerializedLeafValue[];
76
+ paddedL1ToL2Messages: readonly SerializedLeafValue[];
77
+ paddedNullifiers: readonly SerializedLeafValue[];
78
+ publicDataWrites: readonly SerializedLeafValue[];
79
+ }): Promise<WorldStateStatusFull>;
80
+ createFork(input: { latest: boolean; blockNumber: BlockNumber }): Promise<number>;
81
+ deleteFork(forkId: number): Promise<void>;
82
+ finalizeBlocks(toBlockNumber: BlockNumber): Promise<WorldStateStatusSummary>;
83
+ unwindBlocks(toBlockNumber: BlockNumber): Promise<WorldStateStatusFull>;
84
+ removeHistoricalBlocks(toBlockNumber: BlockNumber): Promise<WorldStateStatusFull>;
85
+ getStatus(): Promise<WorldStateStatusSummary>;
86
+ createCheckpoint(forkId: number): Promise<number>;
87
+ commitCheckpoint(forkId: number): Promise<void>;
88
+ revertCheckpoint(forkId: number): Promise<void>;
89
+ commitAllCheckpoints(forkId: number, depth: number): Promise<void>;
90
+ revertAllCheckpoints(forkId: number, depth: number): Promise<void>;
91
+ copyStores(dstPath: string, compact: boolean): Promise<void>;
113
92
 
114
93
  /**
115
- * Gets the native WorldState handle from the underlying native instance.
116
- * We call the getHandle() method on the native WorldState to get a NAPI External
117
- * that wraps the underlying C++ WorldState pointer.
118
- * @returns The NAPI External handle to the native WorldState instance,since
119
- * the NAPI external type is opaque, we return any (we could also use an opaque symbol type)
94
+ * IPC path the underlying aztec-wsdb process listens on. The C++ AVM uses this to attach to the
95
+ * same world state instance the TS layer is using.
120
96
  */
121
- public getHandle(): any {
122
- const worldStateWrapper = (this.instance as any).dest;
123
-
124
- if (!worldStateWrapper) {
125
- throw new Error('No WorldStateWrapper found');
126
- }
127
-
128
- if (typeof worldStateWrapper.getHandle !== 'function') {
129
- throw new Error('WorldStateWrapper does not have getHandle method');
130
- }
131
-
132
- // Call getHandle() to get the NAPI External
133
- try {
134
- return worldStateWrapper.getHandle();
135
- } catch (error) {
136
- this.log.error('Failed to get native WorldState handle', error);
137
- }
138
- }
97
+ getIpcPath(): string;
139
98
 
140
99
  /**
141
- * Sends a message to the native instance and returns the response.
142
- * @param messageType - The type of message to send
143
- * @param body - The message body
144
- * @param responseHandler - A callback accepting the response, executed on the job queue
145
- * @param errorHandler - A callback called on request error, executed on the job queue
146
- * @returns The response to the message
100
+ * Shut down the world state instance. Cancels any in-flight queues, closes the IPC channel, and
101
+ * terminates the underlying aztec-wsdb process. Idempotent.
147
102
  */
148
- public async call<T extends WorldStateMessageType>(
149
- messageType: T,
150
- body: WorldStateRequest[T] & WorldStateRequestCategories,
151
- // allows for the pre-processing of responses on the job queue before being passed back
152
- responseHandler = (response: WorldStateResponse[T]): WorldStateResponse[T] => response,
153
- errorHandler = (_: string) => {},
154
- ): Promise<WorldStateResponse[T]> {
155
- // Here we determine which fork the request is being executed against and whether it requires uncommitted data
156
- // We use the fork Id to select the appropriate request queue and the uncommitted data flag to pass to the queue
157
- let forkId = -1;
158
- // We assume it includes uncommitted unless explicitly told otherwise
159
- let committedOnly = false;
160
-
161
- // Canonical requests ALWAYS go against the canonical fork
162
- // These include things like block syncs/unwinds etc
163
- // These requests don't contain a fork ID
164
- if (isWithCanonical(body)) {
165
- forkId = 0;
166
- } else if (isWithForkId(body)) {
167
- forkId = body.forkId;
168
- } else if (isWithRevision(body)) {
169
- forkId = body.revision.forkId;
170
- committedOnly = body.revision.includeUncommitted === false;
171
- } else {
172
- const _: never = body;
173
- throw new Error(`Unable to determine forkId for message=${WorldStateMessageType[messageType]}`);
174
- }
175
-
176
- // Get the queue or create a new one
177
- let requestQueue = this.queues.get(forkId);
178
- if (requestQueue === undefined) {
179
- requestQueue = new WorldStateOpsQueue();
180
- this.queues.set(forkId, requestQueue);
181
- }
182
-
183
- // Enqueue the request and wait for the response
184
- const response = await requestQueue.execute(
185
- async () => {
186
- assert.notEqual(messageType, WorldStateMessageType.CLOSE, 'Use close() to close the native instance');
187
- assert.equal(this.open, true, 'Native instance is closed');
188
- let response: WorldStateResponse[T];
189
- try {
190
- response = await this._sendMessage(messageType, body);
191
- } catch (error: any) {
192
- errorHandler(error.message);
193
- throw error;
194
- }
195
- return responseHandler(response);
196
- },
197
- messageType,
198
- committedOnly,
199
- );
200
-
201
- // If the request was to delete the fork then we clean it up here
202
- if (messageType === WorldStateMessageType.DELETE_FORK) {
203
- await requestQueue.stop();
204
- this.queues.delete(forkId);
205
- }
206
- return response;
207
- }
208
-
209
- /**
210
- * Stops the native instance.
211
- */
212
- public async close(): Promise<void> {
213
- if (!this.open) {
214
- return;
215
- }
216
- this.open = false;
217
- const queue = this.queues.get(0)!;
218
-
219
- await queue.execute(
220
- async () => {
221
- await this._sendMessage(WorldStateMessageType.CLOSE, { canonical: true });
222
- },
223
- WorldStateMessageType.CLOSE,
224
- false,
225
- );
226
- await queue.stop();
227
- }
228
-
229
- private async _sendMessage<T extends WorldStateMessageType>(
230
- messageType: T,
231
- body: WorldStateRequest[T] & WorldStateRequestCategories,
232
- ): Promise<WorldStateResponse[T]> {
233
- let logMetadata: Record<string, any> = {};
234
-
235
- if (body) {
236
- if ('treeId' in body) {
237
- logMetadata['treeId'] = MerkleTreeId[body.treeId];
238
- }
239
-
240
- if ('revision' in body) {
241
- logMetadata = { ...logMetadata, ...body.revision };
242
- }
243
-
244
- if ('forkId' in body) {
245
- logMetadata['forkId'] = body.forkId;
246
- }
247
-
248
- if ('blockNumber' in body) {
249
- logMetadata['blockNumber'] = body.blockNumber;
250
- }
251
-
252
- if ('toBlockNumber' in body) {
253
- logMetadata['toBlockNumber'] = body.toBlockNumber;
254
- }
255
-
256
- if ('leafIndex' in body) {
257
- logMetadata['leafIndex'] = body.leafIndex;
258
- }
259
-
260
- if ('blockHeaderHash' in body) {
261
- logMetadata['blockHeaderHash'] = '0x' + body.blockHeaderHash.toString('hex');
262
- }
263
-
264
- if ('leaves' in body) {
265
- logMetadata['leavesCount'] = body.leaves.length;
266
- }
267
-
268
- // sync operation
269
- if ('paddedNoteHashes' in body) {
270
- logMetadata['notesCount'] = body.paddedNoteHashes.length;
271
- logMetadata['nullifiersCount'] = body.paddedNullifiers.length;
272
- logMetadata['l1ToL2MessagesCount'] = body.paddedL1ToL2Messages.length;
273
- logMetadata['publicDataWritesCount'] = body.publicDataWrites.length;
274
- }
275
- }
276
-
277
- try {
278
- const { duration, response } = await this.instance.sendMessage(messageType, body);
279
- this.log.trace(`Call ${WorldStateMessageType[messageType]} took (ms)`, {
280
- duration,
281
- ...logMetadata,
282
- });
283
-
284
- this.instrumentation.recordRoundTrip(duration.totalUs, messageType);
285
- return response;
286
- } catch (error) {
287
- this.log.error(`Call ${WorldStateMessageType[messageType]} failed: ${error}`, error, logMetadata);
288
- throw error;
289
- }
290
- }
103
+ close(): Promise<void>;
291
104
  }
@@ -0,0 +1,33 @@
1
+ /**
2
+ * Names of the world-state operations, used as labels for instrumentation and
3
+ * logging. (Read/write ordering is enforced server-side by the wsdb scheduler,
4
+ * so there is no longer a client-side operation queue.)
5
+ */
6
+ export type WorldStateOperationName =
7
+ | 'getTreeInfo'
8
+ | 'getStateReference'
9
+ | 'getInitialStateReference'
10
+ | 'getLeafValue'
11
+ | 'getLeafPreimage'
12
+ | 'getSiblingPath'
13
+ | 'getBlockNumbersForLeafIndices'
14
+ | 'findLeafIndices'
15
+ | 'findLowLeaf'
16
+ | 'findSiblingPaths'
17
+ | 'appendLeaves'
18
+ | 'batchInsert'
19
+ | 'sequentialInsert'
20
+ | 'updateArchive'
21
+ | 'syncBlock'
22
+ | 'createFork'
23
+ | 'deleteFork'
24
+ | 'finalizeBlocks'
25
+ | 'unwindBlocks'
26
+ | 'removeHistoricalBlocks'
27
+ | 'getStatus'
28
+ | 'createCheckpoint'
29
+ | 'commitCheckpoint'
30
+ | 'revertCheckpoint'
31
+ | 'commitAllCheckpoints'
32
+ | 'revertAllCheckpoints'
33
+ | 'copyStores';
@@ -1,4 +1,9 @@
1
- import { type ConfigMappingsType, getConfigFromMappings, numberConfigHelper } from '@aztec/foundation/config';
1
+ import {
2
+ type ConfigMappingsType,
3
+ getConfigFromMappings,
4
+ numberConfigHelper,
5
+ optionalNumberConfigHelper,
6
+ } from '@aztec/foundation/config';
2
7
 
3
8
  /** World State synchronizer configuration values. */
4
9
  export interface WorldStateConfig {
@@ -36,47 +41,46 @@ export interface WorldStateConfig {
36
41
  export const worldStateConfigMappings: ConfigMappingsType<WorldStateConfig> = {
37
42
  worldStateBlockCheckIntervalMS: {
38
43
  env: 'WS_BLOCK_CHECK_INTERVAL_MS',
39
- parseEnv: (val: string) => +val,
40
- defaultValue: 100,
44
+ ...numberConfigHelper(100),
41
45
  description: 'The frequency in which to check.',
42
46
  },
43
47
  worldStateBlockRequestBatchSize: {
44
48
  env: 'WS_BLOCK_REQUEST_BATCH_SIZE',
45
- parseEnv: (val: string | undefined) => (val ? +val : undefined),
49
+ ...optionalNumberConfigHelper(),
46
50
  description: 'Size of the batch for each get-blocks request from the synchronizer to the archiver.',
47
51
  },
48
52
  worldStateDbMapSizeKb: {
49
53
  env: 'WS_DB_MAP_SIZE_KB',
50
- parseEnv: (val: string | undefined) => (val ? +val : undefined),
54
+ ...optionalNumberConfigHelper(),
51
55
  description: 'The maximum possible size of the world state DB in KB. Overwrites the general dataStoreMapSizeKb.',
52
56
  },
53
57
  archiveTreeMapSizeKb: {
54
58
  env: 'ARCHIVE_TREE_MAP_SIZE_KB',
55
- parseEnv: (val: string | undefined) => (val ? +val : undefined),
59
+ ...optionalNumberConfigHelper(),
56
60
  description:
57
61
  'The maximum possible size of the world state archive tree in KB. Overwrites the general worldStateDbMapSizeKb.',
58
62
  },
59
63
  nullifierTreeMapSizeKb: {
60
64
  env: 'NULLIFIER_TREE_MAP_SIZE_KB',
61
- parseEnv: (val: string | undefined) => (val ? +val : undefined),
65
+ ...optionalNumberConfigHelper(),
62
66
  description:
63
67
  'The maximum possible size of the world state nullifier tree in KB. Overwrites the general worldStateDbMapSizeKb.',
64
68
  },
65
69
  noteHashTreeMapSizeKb: {
66
70
  env: 'NOTE_HASH_TREE_MAP_SIZE_KB',
67
- parseEnv: (val: string | undefined) => (val ? +val : undefined),
71
+ ...optionalNumberConfigHelper(),
68
72
  description:
69
73
  'The maximum possible size of the world state note hash tree in KB. Overwrites the general worldStateDbMapSizeKb.',
70
74
  },
71
75
  messageTreeMapSizeKb: {
72
76
  env: 'MESSAGE_TREE_MAP_SIZE_KB',
73
- parseEnv: (val: string | undefined) => (val ? +val : undefined),
77
+ ...optionalNumberConfigHelper(),
74
78
  description:
75
79
  'The maximum possible size of the world state message tree in KB. Overwrites the general worldStateDbMapSizeKb.',
76
80
  },
77
81
  publicDataTreeMapSizeKb: {
78
82
  env: 'PUBLIC_DATA_TREE_MAP_SIZE_KB',
79
- parseEnv: (val: string | undefined) => (val ? +val : undefined),
83
+ ...optionalNumberConfigHelper(),
80
84
  description:
81
85
  'The maximum possible size of the world state public data tree in KB. Overwrites the general worldStateDbMapSizeKb.',
82
86
  },
@@ -1,5 +1,13 @@
1
+ /**
2
+ * Thrown by {@link ServerWorldStateSynchronizer.syncImmediate} when world state cannot be synced to the requested
3
+ * target: either the block is not available from the block source (`block_not_available`, e.g. it was pruned away)
4
+ * or the synced block does not match the requested hash (`block_hash_mismatch`, i.e. a reorg). Both causes are
5
+ * transient from the caller's perspective: re-resolving the query against the current chain and retrying may succeed
6
+ * or produce a more precise error.
7
+ */
1
8
  export class WorldStateSynchronizerError extends Error {
2
9
  constructor(message: string, options?: ErrorOptions) {
3
10
  super(message, options);
11
+ this.name = 'WorldStateSynchronizerError';
4
12
  }
5
13
  }
@@ -1,8 +1,8 @@
1
1
  import type { LoggerBindings } from '@aztec/foundation/log';
2
- import type { DataStoreConfig } from '@aztec/kv-store/config';
3
2
  import type { L2BlockSource } from '@aztec/stdlib/block';
3
+ import type { DataStoreConfig } from '@aztec/stdlib/kv-store';
4
4
  import type { L1ToL2MessageSource } from '@aztec/stdlib/messaging';
5
- import type { PublicDataTreeLeaf } from '@aztec/stdlib/trees';
5
+ import { EMPTY_GENESIS_DATA, type GenesisData, isGenesisData } from '@aztec/stdlib/world-state';
6
6
  import { type TelemetryClient, getTelemetryClient } from '@aztec/telemetry-client';
7
7
 
8
8
  import { WorldStateInstrumentation } from '../instrumentation/instrumentation.js';
@@ -21,12 +21,14 @@ export interface WorldStateTreeMapSizes {
21
21
  export async function createWorldStateSynchronizer(
22
22
  config: WorldStateConfig & DataStoreConfig,
23
23
  l2BlockSource: L2BlockSource & L1ToL2MessageSource,
24
- prefilledPublicData: PublicDataTreeLeaf[] = [],
24
+ genesisOrNativeWorldState: GenesisData | NativeWorldStateService,
25
25
  client: TelemetryClient = getTelemetryClient(),
26
26
  bindings?: LoggerBindings,
27
27
  ) {
28
28
  const instrumentation = new WorldStateInstrumentation(client);
29
- const merkleTrees = await createWorldState(config, prefilledPublicData, instrumentation, bindings);
29
+ const merkleTrees = isGenesisData(genesisOrNativeWorldState)
30
+ ? await createWorldState(config, genesisOrNativeWorldState, instrumentation, bindings)
31
+ : genesisOrNativeWorldState;
30
32
  return new ServerWorldStateSynchronizer(merkleTrees, l2BlockSource, config, instrumentation);
31
33
  }
32
34
 
@@ -41,8 +43,8 @@ export async function createWorldState(
41
43
  | 'messageTreeMapSizeKb'
42
44
  | 'publicDataTreeMapSizeKb'
43
45
  > &
44
- Pick<DataStoreConfig, 'dataDirectory' | 'dataStoreMapSizeKb' | 'l1Contracts'>,
45
- prefilledPublicData: PublicDataTreeLeaf[] = [],
46
+ Pick<DataStoreConfig, 'dataDirectory' | 'dataStoreMapSizeKb' | 'rollupAddress'>,
47
+ genesis: GenesisData = EMPTY_GENESIS_DATA,
46
48
  instrumentation: WorldStateInstrumentation = new WorldStateInstrumentation(getTelemetryClient()),
47
49
  bindings?: LoggerBindings,
48
50
  ) {
@@ -56,24 +58,23 @@ export async function createWorldState(
56
58
  publicDataTreeMapSizeKb: config.publicDataTreeMapSizeKb ?? dataStoreMapSizeKb,
57
59
  };
58
60
 
59
- if (!config.l1Contracts?.rollupAddress) {
61
+ if (!config.rollupAddress) {
60
62
  throw new Error('Rollup address is required to create a world state synchronizer.');
61
63
  }
62
64
 
63
65
  // If a data directory is provided in config, then create a persistent store.
64
66
  const merkleTrees = dataDirectory
65
67
  ? await NativeWorldStateService.new(
66
- config.l1Contracts.rollupAddress,
68
+ config.rollupAddress,
67
69
  dataDirectory,
68
70
  wsTreeMapSizes,
69
- prefilledPublicData,
71
+ genesis,
70
72
  instrumentation,
71
73
  bindings,
72
74
  )
73
75
  : await NativeWorldStateService.tmp(
74
- config.l1Contracts.rollupAddress,
75
76
  !['true', '1'].includes(process.env.DEBUG_WORLD_STATE!),
76
- prefilledPublicData,
77
+ genesis,
77
78
  instrumentation,
78
79
  bindings,
79
80
  );
@@ -1,2 +1,3 @@
1
1
  export * from './server_world_state_synchronizer.js';
2
2
  export * from './factory.js';
3
+ export * from './errors.js';