@aztec/world-state 0.0.1-commit.96dac018d → 0.0.1-commit.993d240

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 (45) hide show
  1. package/dest/native/fork_checkpoint.d.ts +7 -1
  2. package/dest/native/fork_checkpoint.d.ts.map +1 -1
  3. package/dest/native/fork_checkpoint.js +15 -3
  4. package/dest/native/ipc_world_state_instance.d.ts +50 -0
  5. package/dest/native/ipc_world_state_instance.d.ts.map +1 -0
  6. package/dest/native/ipc_world_state_instance.js +628 -0
  7. package/dest/native/merkle_trees_facade.d.ts +8 -5
  8. package/dest/native/merkle_trees_facade.d.ts.map +1 -1
  9. package/dest/native/merkle_trees_facade.js +32 -12
  10. package/dest/native/message.d.ts +13 -6
  11. package/dest/native/message.d.ts.map +1 -1
  12. package/dest/native/native_world_state.d.ts +31 -5
  13. package/dest/native/native_world_state.d.ts.map +1 -1
  14. package/dest/native/native_world_state.js +80 -27
  15. package/dest/native/native_world_state_instance.d.ts +5 -4
  16. package/dest/native/native_world_state_instance.d.ts.map +1 -1
  17. package/dest/native/native_world_state_instance.js +32 -25
  18. package/dest/native/world_state_ops_queue.js +5 -5
  19. package/dest/synchronizer/config.d.ts +1 -1
  20. package/dest/synchronizer/config.d.ts.map +1 -1
  21. package/dest/synchronizer/config.js +9 -10
  22. package/dest/synchronizer/factory.d.ts +5 -5
  23. package/dest/synchronizer/factory.d.ts.map +1 -1
  24. package/dest/synchronizer/factory.js +7 -6
  25. package/dest/synchronizer/server_world_state_synchronizer.d.ts +4 -4
  26. package/dest/synchronizer/server_world_state_synchronizer.d.ts.map +1 -1
  27. package/dest/synchronizer/server_world_state_synchronizer.js +71 -19
  28. package/dest/testing.d.ts +4 -3
  29. package/dest/testing.d.ts.map +1 -1
  30. package/dest/testing.js +10 -6
  31. package/dest/world-state-db/merkle_tree_db.d.ts +1 -10
  32. package/dest/world-state-db/merkle_tree_db.d.ts.map +1 -1
  33. package/package.json +12 -11
  34. package/src/native/fork_checkpoint.ts +19 -3
  35. package/src/native/ipc_world_state_instance.ts +717 -0
  36. package/src/native/merkle_trees_facade.ts +40 -15
  37. package/src/native/message.ts +14 -5
  38. package/src/native/native_world_state.ts +98 -34
  39. package/src/native/native_world_state_instance.ts +38 -28
  40. package/src/native/world_state_ops_queue.ts +5 -5
  41. package/src/synchronizer/config.ts +14 -10
  42. package/src/synchronizer/factory.ts +13 -11
  43. package/src/synchronizer/server_world_state_synchronizer.ts +74 -24
  44. package/src/testing.ts +8 -9
  45. package/src/world-state-db/merkle_tree_db.ts +0 -10
@@ -4,6 +4,7 @@ import { createLogger } from '@aztec/foundation/log';
4
4
  import { serializeToBuffer } from '@aztec/foundation/serialize';
5
5
  import { sleep } from '@aztec/foundation/sleep';
6
6
  import { type IndexedTreeLeafPreimage, SiblingPath } from '@aztec/foundation/trees';
7
+ import { BlockHash } from '@aztec/stdlib/block';
7
8
  import type {
8
9
  BatchInsertionResult,
9
10
  IndexedTreeId,
@@ -118,7 +119,7 @@ export class MerkleTreesFacade implements MerkleTreeReadOperations {
118
119
 
119
120
  const leaf = deserializeLeafValue(resp);
120
121
  if (leaf instanceof Fr) {
121
- return leaf as any;
122
+ return treeId === MerkleTreeId.ARCHIVE ? (new BlockHash(leaf) as any) : (leaf as any);
122
123
  } else {
123
124
  return leaf.toBuffer() as any;
124
125
  }
@@ -207,6 +208,7 @@ export class MerkleTreesFacade implements MerkleTreeReadOperations {
207
208
 
208
209
  export class MerkleTreesForkFacade extends MerkleTreesFacade implements MerkleTreeWriteOperations {
209
210
  private log = createLogger('world-state:merkle-trees-fork-facade');
211
+ private closePromise: Promise<void> | undefined;
210
212
 
211
213
  constructor(
212
214
  instance: NativeWorldStateInstance,
@@ -228,7 +230,7 @@ export class MerkleTreesForkFacade extends MerkleTreesFacade implements MerkleTr
228
230
 
229
231
  async appendLeaves<ID extends MerkleTreeId>(treeId: ID, leaves: MerkleTreeLeafType<ID>[]): Promise<void> {
230
232
  await this.instance.call(WorldStateMessageType.APPEND_LEAVES, {
231
- leaves: leaves.map(leaf => leaf as any),
233
+ leaves: leaves.map(leaf => serializeLeaf(hydrateLeaf(treeId, leaf as any))),
232
234
  forkId: this.revision.forkId,
233
235
  treeId,
234
236
  });
@@ -290,8 +292,17 @@ export class MerkleTreesForkFacade extends MerkleTreesFacade implements MerkleTr
290
292
  };
291
293
  }
292
294
 
293
- public async close(): Promise<void> {
295
+ public close(): Promise<void> {
294
296
  assert.notEqual(this.revision.forkId, 0, 'Fork ID must be set');
297
+ // Share the in-flight close promise across duplicate dispose calls so DELETE_FORK is sent at most once.
298
+ if (this.closePromise) {
299
+ return this.closePromise;
300
+ }
301
+ this.closePromise = this.doClose();
302
+ return this.closePromise;
303
+ }
304
+
305
+ private async doClose(): Promise<void> {
295
306
  try {
296
307
  await this.instance.call(WorldStateMessageType.DELETE_FORK, { forkId: this.revision.forkId });
297
308
  } catch (err: any) {
@@ -300,6 +311,12 @@ export class MerkleTreesForkFacade extends MerkleTreesFacade implements MerkleTr
300
311
  if (err?.message === 'Native instance is closed') {
301
312
  return;
302
313
  }
314
+ // Ignore "Fork not found": the native fork was already destroyed by a pending-chain unwind or a
315
+ // historical prune (both call C++ remove_forks_for_block). Fork IDs are monotonic and never reused,
316
+ // so swallowing this on close cannot mask a deletion of a different fork.
317
+ if (err?.message === 'Fork not found') {
318
+ return;
319
+ }
303
320
  throw err;
304
321
  }
305
322
  }
@@ -309,9 +326,6 @@ export class MerkleTreesForkFacade extends MerkleTreesFacade implements MerkleTr
309
326
  void sleep(this.opts.closeDelayMs)
310
327
  .then(() => this.close())
311
328
  .catch(err => {
312
- if (err && 'message' in err && err.message === 'Native instance is closed') {
313
- return; // Ignore errors due to native instance being closed
314
- }
315
329
  this.log.warn('Error closing MerkleTreesForkFacade after delay', { err });
316
330
  });
317
331
  } else {
@@ -319,9 +333,10 @@ export class MerkleTreesForkFacade extends MerkleTreesFacade implements MerkleTr
319
333
  }
320
334
  }
321
335
 
322
- public async createCheckpoint(): Promise<void> {
336
+ public async createCheckpoint(): Promise<number> {
323
337
  assert.notEqual(this.revision.forkId, 0, 'Fork ID must be set');
324
- await this.instance.call(WorldStateMessageType.CREATE_CHECKPOINT, { forkId: this.revision.forkId });
338
+ const resp = await this.instance.call(WorldStateMessageType.CREATE_CHECKPOINT, { forkId: this.revision.forkId });
339
+ return resp.depth;
325
340
  }
326
341
 
327
342
  public async commitCheckpoint(): Promise<void> {
@@ -334,20 +349,28 @@ export class MerkleTreesForkFacade extends MerkleTreesFacade implements MerkleTr
334
349
  await this.instance.call(WorldStateMessageType.REVERT_CHECKPOINT, { forkId: this.revision.forkId });
335
350
  }
336
351
 
337
- public async commitAllCheckpoints(): Promise<void> {
352
+ public async commitAllCheckpointsTo(depth: number): Promise<void> {
338
353
  assert.notEqual(this.revision.forkId, 0, 'Fork ID must be set');
339
- await this.instance.call(WorldStateMessageType.COMMIT_ALL_CHECKPOINTS, { forkId: this.revision.forkId });
354
+ await this.instance.call(WorldStateMessageType.COMMIT_ALL_CHECKPOINTS, {
355
+ forkId: this.revision.forkId,
356
+ depth,
357
+ });
340
358
  }
341
359
 
342
- public async revertAllCheckpoints(): Promise<void> {
360
+ public async revertAllCheckpointsTo(depth: number): Promise<void> {
343
361
  assert.notEqual(this.revision.forkId, 0, 'Fork ID must be set');
344
- await this.instance.call(WorldStateMessageType.REVERT_ALL_CHECKPOINTS, { forkId: this.revision.forkId });
362
+ await this.instance.call(WorldStateMessageType.REVERT_ALL_CHECKPOINTS, {
363
+ forkId: this.revision.forkId,
364
+ depth,
365
+ });
345
366
  }
346
367
  }
347
368
 
348
- function hydrateLeaf<ID extends MerkleTreeId>(treeId: ID, leaf: Fr | Buffer) {
369
+ function hydrateLeaf(treeId: MerkleTreeId, leaf: Fr | BlockHash | Buffer) {
349
370
  if (leaf instanceof Fr) {
350
371
  return leaf;
372
+ } else if (leaf instanceof BlockHash) {
373
+ return leaf.toFr();
351
374
  } else if (treeId === MerkleTreeId.NULLIFIER_TREE) {
352
375
  return NullifierLeaf.fromBuffer(leaf);
353
376
  } else if (treeId === MerkleTreeId.PUBLIC_DATA_TREE) {
@@ -357,8 +380,10 @@ function hydrateLeaf<ID extends MerkleTreeId>(treeId: ID, leaf: Fr | Buffer) {
357
380
  }
358
381
  }
359
382
 
360
- export function serializeLeaf(leaf: Fr | NullifierLeaf | PublicDataTreeLeaf): SerializedLeafValue {
361
- if (leaf instanceof Fr) {
383
+ export function serializeLeaf(leaf: Fr | BlockHash | NullifierLeaf | PublicDataTreeLeaf): SerializedLeafValue {
384
+ if (leaf instanceof BlockHash) {
385
+ return leaf.toBuffer();
386
+ } else if (leaf instanceof Fr) {
362
387
  return leaf.toBuffer();
363
388
  } else if (leaf instanceof NullifierLeaf) {
364
389
  return { nullifier: leaf.nullifier.toBuffer() };
@@ -1,7 +1,6 @@
1
1
  import { BlockNumber } from '@aztec/foundation/branded-types';
2
2
  import { Fr } from '@aztec/foundation/curves/bn254';
3
3
  import type { Tuple } from '@aztec/foundation/serialize';
4
- import type { BlockHash } from '@aztec/stdlib/block';
5
4
  import { AppendOnlyTreeSnapshot, MerkleTreeId } from '@aztec/stdlib/trees';
6
5
  import type { StateReference } from '@aztec/stdlib/tx';
7
6
  import type { UInt32 } from '@aztec/stdlib/types';
@@ -284,6 +283,16 @@ interface WithForkId {
284
283
  forkId: number;
285
284
  }
286
285
 
286
+ interface CreateCheckpointResponse {
287
+ depth: number;
288
+ }
289
+
290
+ /** Request to commit/revert all checkpoints down to a target depth. The resulting depth after the operation equals the given depth. */
291
+ interface CheckpointDepthRequest extends WithForkId {
292
+ /** The target depth after the operation. All checkpoints above this depth are committed/reverted. */
293
+ depth: number;
294
+ }
295
+
287
296
  interface WithWorldStateRevision {
288
297
  revision: WorldStateRevision;
289
298
  }
@@ -413,7 +422,7 @@ interface UpdateArchiveRequest extends WithForkId {
413
422
  interface SyncBlockRequest extends WithCanonicalForkId {
414
423
  blockNumber: BlockNumber;
415
424
  blockStateRef: BlockStateReference;
416
- blockHeaderHash: BlockHash;
425
+ blockHeaderHash: Buffer;
417
426
  paddedNoteHashes: readonly SerializedLeafValue[];
418
427
  paddedL1ToL2Messages: readonly SerializedLeafValue[];
419
428
  paddedNullifiers: readonly SerializedLeafValue[];
@@ -487,8 +496,8 @@ export type WorldStateRequest = {
487
496
  [WorldStateMessageType.CREATE_CHECKPOINT]: WithForkId;
488
497
  [WorldStateMessageType.COMMIT_CHECKPOINT]: WithForkId;
489
498
  [WorldStateMessageType.REVERT_CHECKPOINT]: WithForkId;
490
- [WorldStateMessageType.COMMIT_ALL_CHECKPOINTS]: WithForkId;
491
- [WorldStateMessageType.REVERT_ALL_CHECKPOINTS]: WithForkId;
499
+ [WorldStateMessageType.COMMIT_ALL_CHECKPOINTS]: CheckpointDepthRequest;
500
+ [WorldStateMessageType.REVERT_ALL_CHECKPOINTS]: CheckpointDepthRequest;
492
501
 
493
502
  [WorldStateMessageType.COPY_STORES]: CopyStoresRequest;
494
503
 
@@ -529,7 +538,7 @@ export type WorldStateResponse = {
529
538
 
530
539
  [WorldStateMessageType.GET_STATUS]: WorldStateStatusSummary;
531
540
 
532
- [WorldStateMessageType.CREATE_CHECKPOINT]: void;
541
+ [WorldStateMessageType.CREATE_CHECKPOINT]: CreateCheckpointResponse;
533
542
  [WorldStateMessageType.COMMIT_CHECKPOINT]: void;
534
543
  [WorldStateMessageType.REVERT_CHECKPOINT]: void;
535
544
  [WorldStateMessageType.COMMIT_ALL_CHECKPOINTS]: void;
@@ -14,8 +14,8 @@ import type {
14
14
  } from '@aztec/stdlib/interfaces/server';
15
15
  import type { SnapshotDataKeys } from '@aztec/stdlib/snapshots';
16
16
  import { MerkleTreeId, NullifierLeaf, type NullifierLeafPreimage, PublicDataTreeLeaf } from '@aztec/stdlib/trees';
17
- import { BlockHeader, PartialStateReference, StateReference } from '@aztec/stdlib/tx';
18
- import { WorldStateRevision } from '@aztec/stdlib/world-state';
17
+ import { BlockHeader, GlobalVariables, PartialStateReference, StateReference } from '@aztec/stdlib/tx';
18
+ import { EMPTY_GENESIS_DATA, type GenesisData, WorldStateRevision } from '@aztec/stdlib/world-state';
19
19
  import { getTelemetryClient } from '@aztec/telemetry-client';
20
20
 
21
21
  import assert from 'assert/strict';
@@ -44,6 +44,29 @@ export const WORLD_STATE_DB_VERSION = 2; // The initial version
44
44
 
45
45
  export const WORLD_STATE_DIR = 'world_state';
46
46
 
47
+ const DEFAULT_TMP_TREE_MAP_SIZE_KB = 10 * 1024 * 1024;
48
+
49
+ /**
50
+ * Sets up a fresh `mkdtemp` directory + default `WorldStateTreeMapSizes` shared by both
51
+ * the `.tmp` (fsync-on) and `.ephemeral` (fsync-off) factories. Returns the raw tmpdir,
52
+ * the tree map sizes, and the package logger.
53
+ */
54
+ async function createTmpWorldStateDir(
55
+ bindings?: LoggerBindings,
56
+ ): Promise<{ dataDir: string; wsTreeMapSizes: WorldStateTreeMapSizes; log: Logger }> {
57
+ const log = createLogger('world-state:database', bindings);
58
+ const dataDir = await mkdtemp(join(tmpdir(), 'aztec-world-state-'));
59
+ const wsTreeMapSizes: WorldStateTreeMapSizes = {
60
+ archiveTreeMapSizeKb: DEFAULT_TMP_TREE_MAP_SIZE_KB,
61
+ nullifierTreeMapSizeKb: DEFAULT_TMP_TREE_MAP_SIZE_KB,
62
+ noteHashTreeMapSizeKb: DEFAULT_TMP_TREE_MAP_SIZE_KB,
63
+ messageTreeMapSizeKb: DEFAULT_TMP_TREE_MAP_SIZE_KB,
64
+ publicDataTreeMapSizeKb: DEFAULT_TMP_TREE_MAP_SIZE_KB,
65
+ };
66
+ log.debug(`Created temporary world state database at: ${dataDir} (map size ${DEFAULT_TMP_TREE_MAP_SIZE_KB} KB)`);
67
+ return { dataDir, wsTreeMapSizes, log };
68
+ }
69
+
47
70
  export class NativeWorldStateService implements MerkleTreeDatabase {
48
71
  protected initialHeader: BlockHeader | undefined;
49
72
  // This is read heavily and only changes when data is persisted, so we cache it
@@ -53,34 +76,46 @@ export class NativeWorldStateService implements MerkleTreeDatabase {
53
76
  protected instance: NativeWorldState,
54
77
  protected readonly worldStateInstrumentation: WorldStateInstrumentation,
55
78
  protected readonly log: Logger,
79
+ private readonly genesis: GenesisData = EMPTY_GENESIS_DATA,
56
80
  private readonly cleanup = () => Promise.resolve(),
57
81
  ) {}
58
82
 
83
+ /**
84
+ * Opens a persistent world state at `dataDir`. Goes through `DatabaseVersionManager` so the
85
+ * caller's rollup address is bound to the on-disk schema and incompatible versions surface
86
+ * loudly. The LMDB envs commit with full fsync.
87
+ */
59
88
  static async new(
60
89
  rollupAddress: EthAddress,
61
90
  dataDir: string,
62
91
  wsTreeMapSizes: WorldStateTreeMapSizes,
63
- prefilledPublicData: PublicDataTreeLeaf[] = [],
92
+ genesis: GenesisData = EMPTY_GENESIS_DATA,
64
93
  instrumentation = new WorldStateInstrumentation(getTelemetryClient()),
65
94
  bindings?: LoggerBindings,
66
95
  cleanup = () => Promise.resolve(),
67
96
  ): Promise<NativeWorldStateService> {
68
97
  const log = createLogger('world-state:database', bindings);
69
98
  const worldStateDirectory = join(dataDir, WORLD_STATE_DIR);
70
- // Create a version manager to handle versioning
71
99
  const versionManager = new DatabaseVersionManager({
72
100
  schemaVersion: WORLD_STATE_DB_VERSION,
73
101
  rollupAddress,
74
102
  dataDirectory: worldStateDirectory,
75
- onOpen: (dir: string) => {
76
- return Promise.resolve(
77
- new NativeWorldState(dir, wsTreeMapSizes, prefilledPublicData, instrumentation, bindings),
78
- );
79
- },
103
+ onOpen: (dir: string) =>
104
+ Promise.resolve(
105
+ new NativeWorldState(
106
+ dir,
107
+ wsTreeMapSizes,
108
+ genesis,
109
+ instrumentation,
110
+ bindings,
111
+ undefined,
112
+ /*ephemeral=*/ false,
113
+ ),
114
+ ),
80
115
  });
81
116
 
82
117
  const [instance] = await versionManager.open();
83
- const worldState = new this(instance, instrumentation, log, cleanup);
118
+ const worldState = new this(instance, instrumentation, log, genesis, cleanup);
84
119
  try {
85
120
  await worldState.init();
86
121
  } catch (e) {
@@ -91,26 +126,23 @@ export class NativeWorldStateService implements MerkleTreeDatabase {
91
126
  return worldState;
92
127
  }
93
128
 
129
+ /**
130
+ * Opens a world state in a fresh tmpdir with full fsync semantics. Use when you need the
131
+ * on-disk file to remain crash-recoverable (e.g. for snapshot/backup tests) but don't
132
+ * want a persistent dataDir. Pass `cleanupTmpDir=false` to keep the directory after
133
+ * close for inspection.
134
+ *
135
+ * If you don't care about crash-recoverability — i.e. you just want a fast scratch
136
+ * database for tests — use {@link ephemeral} instead.
137
+ */
94
138
  static async tmp(
95
139
  rollupAddress = EthAddress.ZERO,
96
140
  cleanupTmpDir = true,
97
- prefilledPublicData: PublicDataTreeLeaf[] = [],
141
+ genesis: GenesisData = EMPTY_GENESIS_DATA,
98
142
  instrumentation = new WorldStateInstrumentation(getTelemetryClient()),
99
143
  bindings?: LoggerBindings,
100
144
  ): Promise<NativeWorldStateService> {
101
- const log = createLogger('world-state:database', bindings);
102
- const dataDir = await mkdtemp(join(tmpdir(), 'aztec-world-state-'));
103
- const dbMapSizeKb = 10 * 1024 * 1024;
104
- const worldStateTreeMapSizes: WorldStateTreeMapSizes = {
105
- archiveTreeMapSizeKb: dbMapSizeKb,
106
- nullifierTreeMapSizeKb: dbMapSizeKb,
107
- noteHashTreeMapSizeKb: dbMapSizeKb,
108
- messageTreeMapSizeKb: dbMapSizeKb,
109
- publicDataTreeMapSizeKb: dbMapSizeKb,
110
- };
111
- log.debug(`Created temporary world state database at: ${dataDir} with tree map size: ${dbMapSizeKb}`);
112
-
113
- // pass a cleanup callback because process.on('beforeExit', cleanup) does not work under Jest
145
+ const { dataDir, wsTreeMapSizes, log } = await createTmpWorldStateDir(bindings);
114
146
  const cleanup = async () => {
115
147
  if (cleanupTmpDir) {
116
148
  await rm(dataDir, { recursive: true, force: true, maxRetries: 3 });
@@ -119,16 +151,45 @@ export class NativeWorldStateService implements MerkleTreeDatabase {
119
151
  log.debug(`Leaving temporary world state database: ${dataDir}`);
120
152
  }
121
153
  };
154
+ return this.new(rollupAddress, dataDir, wsTreeMapSizes, genesis, instrumentation, bindings, cleanup);
155
+ }
122
156
 
123
- return this.new(
124
- rollupAddress,
125
- dataDir,
126
- worldStateTreeMapSizes,
127
- prefilledPublicData,
157
+ /**
158
+ * Opens a fully-ephemeral world state. The directory is created in `os.tmpdir()`, the LMDB
159
+ * envs open with `MDB_NOSYNC | MDB_NOMETASYNC` so commits never block on fsync, and the
160
+ * directory is removed on dispose. A crash mid-write leaves the env unrecoverable.
161
+ *
162
+ * For unit tests and other isolated runs. Use {@link tmp} when you need fsync semantics in a
163
+ * tmp dir, and {@link new} for a persistent store. Skips {@link DatabaseVersionManager} —
164
+ * there is no on-disk schema to bind to and no rollup address is taken.
165
+ */
166
+ static async ephemeral(
167
+ genesis: GenesisData = EMPTY_GENESIS_DATA,
168
+ instrumentation = new WorldStateInstrumentation(getTelemetryClient()),
169
+ bindings?: LoggerBindings,
170
+ ): Promise<NativeWorldStateService> {
171
+ const { dataDir, wsTreeMapSizes, log } = await createTmpWorldStateDir(bindings);
172
+ const cleanup = async () => {
173
+ await rm(dataDir, { recursive: true, force: true, maxRetries: 3 });
174
+ log.debug(`Deleted ephemeral world state database: ${dataDir}`);
175
+ };
176
+ const instance = new NativeWorldState(
177
+ join(dataDir, WORLD_STATE_DIR),
178
+ wsTreeMapSizes,
179
+ genesis,
128
180
  instrumentation,
129
181
  bindings,
130
- cleanup,
182
+ undefined,
183
+ /*ephemeral=*/ true,
131
184
  );
185
+ const worldState = new this(instance, instrumentation, log, genesis, cleanup);
186
+ try {
187
+ await worldState.init();
188
+ } catch (e) {
189
+ log.error(`Error initializing ephemeral world state: ${e}`);
190
+ throw e;
191
+ }
192
+ return worldState;
132
193
  }
133
194
 
134
195
  protected async init() {
@@ -147,7 +208,7 @@ export class NativeWorldStateService implements MerkleTreeDatabase {
147
208
 
148
209
  // the initial header _must_ be the first element in the archive tree
149
210
  // if this assertion fails, check that the hashing done in Header in yarn-project matches the initial header hash done in world_state.cpp
150
- const indices = await committed.findLeafIndices(MerkleTreeId.ARCHIVE, [(await this.initialHeader.hash()).toFr()]);
211
+ const indices = await committed.findLeafIndices(MerkleTreeId.ARCHIVE, [await this.initialHeader.hash()]);
151
212
  const initialHeaderIndex = indices[0];
152
213
  assert.strictEqual(initialHeaderIndex, 0n, 'Invalid initial archive state');
153
214
  }
@@ -185,7 +246,7 @@ export class NativeWorldStateService implements MerkleTreeDatabase {
185
246
  this.initialHeader!,
186
247
  new WorldStateRevision(
187
248
  /*forkId=*/ resp.forkId,
188
- /* blockNumber=*/ BlockNumber.ZERO,
249
+ /* blockNumber=*/ WorldStateRevision.LATEST,
189
250
  /* includeUncommitted=*/ true,
190
251
  ),
191
252
  opts,
@@ -232,7 +293,7 @@ export class NativeWorldStateService implements MerkleTreeDatabase {
232
293
  WorldStateMessageType.SYNC_BLOCK,
233
294
  {
234
295
  blockNumber: l2Block.number,
235
- blockHeaderHash: await l2Block.hash(),
296
+ blockHeaderHash: (await l2Block.hash()).toBuffer(),
236
297
  paddedL1ToL2Messages: paddedL1ToL2Messages.map(serializeLeaf),
237
298
  paddedNoteHashes: paddedNoteHashes.map(serializeLeaf),
238
299
  paddedNullifiers: paddedNullifiers.map(serializeLeaf),
@@ -256,7 +317,10 @@ export class NativeWorldStateService implements MerkleTreeDatabase {
256
317
 
257
318
  private async buildInitialHeader(): Promise<BlockHeader> {
258
319
  const state = await this.getInitialStateReference();
259
- return BlockHeader.empty({ state });
320
+ return BlockHeader.empty({
321
+ state,
322
+ globalVariables: GlobalVariables.empty({ timestamp: this.genesis.genesisTimestamp }),
323
+ });
260
324
  }
261
325
 
262
326
  private sanitizeAndCacheSummaryFromFull(response: WorldStateStatusFull) {
@@ -11,7 +11,7 @@ import {
11
11
  import { type Logger, type LoggerBindings, createLogger } from '@aztec/foundation/log';
12
12
  import { NativeWorldState as BaseNativeWorldState, MsgpackChannel } from '@aztec/native';
13
13
  import { MerkleTreeId } from '@aztec/stdlib/trees';
14
- import type { PublicDataTreeLeaf } from '@aztec/stdlib/trees';
14
+ import { EMPTY_GENESIS_DATA, type GenesisData } from '@aztec/stdlib/world-state';
15
15
 
16
16
  import assert from 'assert';
17
17
  import { cpus } from 'os';
@@ -55,18 +55,22 @@ export class NativeWorldState implements NativeWorldStateInstance {
55
55
  constructor(
56
56
  private readonly dataDir: string,
57
57
  private readonly wsTreeMapSizes: WorldStateTreeMapSizes,
58
- private readonly prefilledPublicData: PublicDataTreeLeaf[] = [],
58
+ private readonly genesis: GenesisData = EMPTY_GENESIS_DATA,
59
59
  private readonly instrumentation: WorldStateInstrumentation,
60
60
  bindings?: LoggerBindings,
61
61
  private readonly log: Logger = createLogger('world-state:database', bindings),
62
+ private readonly ephemeral: boolean = false,
62
63
  ) {
63
64
  const threads = Math.min(cpus().length, MAX_WORLD_STATE_THREADS);
64
65
  log.info(
65
66
  `Creating world state data store at directory ${dataDir} with map sizes ${JSON.stringify(
66
67
  wsTreeMapSizes,
67
- )} and ${threads} threads.`,
68
+ )} and ${threads} threads (ephemeral=${ephemeral}).`,
68
69
  );
69
- const prefilledPublicDataBufferArray = prefilledPublicData.map(d => [d.slot.toBuffer(), d.value.toBuffer()]);
70
+ const prefilledPublicDataBufferArray = genesis.prefilledPublicData.map(d => [
71
+ d.slot.toBuffer(),
72
+ d.value.toBuffer(),
73
+ ]);
70
74
  const ws = new BaseNativeWorldState(
71
75
  dataDir,
72
76
  {
@@ -82,6 +86,7 @@ export class NativeWorldState implements NativeWorldStateInstance {
82
86
  },
83
87
  prefilledPublicDataBufferArray,
84
88
  DomainSeparator.BLOCK_HEADER_HASH,
89
+ Number(genesis.genesisTimestamp),
85
90
  {
86
91
  [MerkleTreeId.NULLIFIER_TREE]: wsTreeMapSizes.nullifierTreeMapSizeKb,
87
92
  [MerkleTreeId.NOTE_HASH_TREE]: wsTreeMapSizes.noteHashTreeMapSizeKb,
@@ -90,6 +95,7 @@ export class NativeWorldState implements NativeWorldStateInstance {
90
95
  [MerkleTreeId.ARCHIVE]: wsTreeMapSizes.archiveTreeMapSizeKb,
91
96
  },
92
97
  threads,
98
+ ephemeral,
93
99
  );
94
100
  this.instance = new MsgpackChannel(ws);
95
101
  // Manually create the queue for the canonical fork
@@ -104,10 +110,11 @@ export class NativeWorldState implements NativeWorldStateInstance {
104
110
  return new NativeWorldState(
105
111
  this.dataDir,
106
112
  this.wsTreeMapSizes,
107
- this.prefilledPublicData,
113
+ this.genesis,
108
114
  this.instrumentation,
109
115
  this.log.getBindings(),
110
116
  this.log,
117
+ this.ephemeral,
111
118
  );
112
119
  }
113
120
 
@@ -180,30 +187,33 @@ export class NativeWorldState implements NativeWorldStateInstance {
180
187
  this.queues.set(forkId, requestQueue);
181
188
  }
182
189
 
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);
190
+ // Enqueue the request and wait for the response. The per-fork queue is cleaned up in `finally` even on
191
+ // error, so the JS-side queues map cannot outlive the native fork (e.g. when the native fork was already
192
+ // destroyed by an unwind/historical-prune and DELETE_FORK rejects with "Fork not found").
193
+ try {
194
+ const response = await requestQueue.execute(
195
+ async () => {
196
+ assert.notEqual(messageType, WorldStateMessageType.CLOSE, 'Use close() to close the native instance');
197
+ assert.equal(this.open, true, 'Native instance is closed');
198
+ let response: WorldStateResponse[T];
199
+ try {
200
+ response = await this._sendMessage(messageType, body);
201
+ } catch (error: any) {
202
+ errorHandler(error.message);
203
+ throw error;
204
+ }
205
+ return responseHandler(response);
206
+ },
207
+ messageType,
208
+ committedOnly,
209
+ );
210
+ return response;
211
+ } finally {
212
+ if (messageType === WorldStateMessageType.DELETE_FORK) {
213
+ await requestQueue.stop();
214
+ this.queues.delete(forkId);
215
+ }
205
216
  }
206
- return response;
207
217
  }
208
218
 
209
219
  /**
@@ -96,7 +96,7 @@ export class WorldStateOpsQueue {
96
96
  // then send the request immediately
97
97
  // If a mutating request is in flight then we must wait
98
98
  // If a mutating request is not in flight but something is queued then it must be a mutating request
99
- if (this.inFlightMutatingCount == 0 && this.requests.length == 0) {
99
+ if (this.inFlightMutatingCount === 0 && this.requests.length === 0) {
100
100
  this.sendEnqueuedRequest(op);
101
101
  } else {
102
102
  this.requests.push(op);
@@ -122,7 +122,7 @@ export class WorldStateOpsQueue {
122
122
  --this.inFlightCount;
123
123
 
124
124
  // If there are still requests in flight then do nothing further
125
- if (this.inFlightCount != 0) {
125
+ if (this.inFlightCount !== 0) {
126
126
  return;
127
127
  }
128
128
 
@@ -134,7 +134,7 @@ export class WorldStateOpsQueue {
134
134
  while (this.requests.length > 0) {
135
135
  const next = this.requests[0];
136
136
  if (next.mutating) {
137
- if (this.inFlightCount == 0) {
137
+ if (this.inFlightCount === 0) {
138
138
  // send the mutating request
139
139
  this.requests.shift();
140
140
  this.sendEnqueuedRequest(next);
@@ -149,7 +149,7 @@ export class WorldStateOpsQueue {
149
149
  }
150
150
 
151
151
  // If the queue is empty, there is nothing in flight and we have been told to stop, then resolve the stop promise
152
- if (this.inFlightCount == 0 && this.stopResolve !== undefined) {
152
+ if (this.inFlightCount === 0 && this.stopResolve !== undefined) {
153
153
  this.stopResolve();
154
154
  }
155
155
  }
@@ -182,7 +182,7 @@ export class WorldStateOpsQueue {
182
182
  });
183
183
 
184
184
  // If no outstanding requests then immediately resolve the promise
185
- if (this.requests.length == 0 && this.inFlightCount == 0 && this.stopResolve !== undefined) {
185
+ if (this.requests.length === 0 && this.inFlightCount === 0 && this.stopResolve !== undefined) {
186
186
  this.stopResolve();
187
187
  }
188
188
  return this.stopPromise;
@@ -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
  },