All files / kernel-test/src/vats persistence-worker-vat.ts

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
import { makeDefaultExo } from '@metamask/kernel-utils/exo';
import type { Baggage } from '@metamask/ocap-kernel';
 
import { unwrapTestLogger } from '../test-powers.ts';
import type { TestPowers } from '../test-powers.ts';
 
/**
 * Build function for a persistent worker vat.
 *
 * @param vatPowers - Special powers granted to this vat.
 * @param vatPowers.logger - The logger for the vat.
 * @param parameters - Initialization parameters from the vat's config object.
 * @param parameters.name - The name of the vat.
 * @param parameters.id - The worker ID.
 * @param baggage - Root of vat's persistent state.
 * @returns The root object for the new vat.
 */
// eslint-disable-next-line @typescript-eslint/explicit-function-return-type
export function buildRootObject(
  vatPowers: TestPowers,
  parameters: { name?: string; id?: number },
  baggage: Baggage,
) {
  const name = parameters?.name ?? 'Worker';
  const id = parameters?.id ?? 0;
  const logger = unwrapTestLogger(vatPowers, name);
  const tlog = (message: string): void => logger(`${name}: ${message}`);
 
  let workCount: number;
 
  if (baggage.has('workCount')) {
    // Resume from persistent state
    workCount = baggage.get('workCount') as number;
    tlog(`resumed with work count: ${workCount}`);
  } else {
    // Initialize new worker
    workCount = 0;
    baggage.init('workCount', workCount);
    tlog(`initialized with id: ${id}`);
  }
 
  return makeDefaultExo('worker', {
    async bootstrap() {
      tlog(`bootstrap called`);
      return `Worker${id} initialized`;
    },
 
    async resume() {
      tlog(`resume called`);
      // Do work when resume is called
      workCount += 1;
      baggage.set('workCount', workCount);
      tlog(`did work, count: ${workCount}`);
      return `Worker${id}(${workCount})`;
    },
  });
}