Press n or j to go to the next uncovered block, b, p or k for the previous block.
| 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 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 | /* global harden */
import { E } from '@endo/eventual-send';
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 coordinator 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 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 },
baggage: Baggage,
) {
const name = parameters?.name ?? 'Coordinator';
const logger = unwrapTestLogger(vatPowers, name);
const tlog = (message: string): void => logger(`${name}: ${message}`);
let workCount: number;
let workers: { worker1?: unknown; worker2?: unknown };
if (baggage.has('workCount')) {
workCount = baggage.get('workCount') as number;
workers = baggage.get('workers') as {
worker1?: unknown;
worker2?: unknown;
};
tlog(`resumed with work count: ${workCount}`);
} else {
workCount = 0;
workers = {};
baggage.init('workCount', workCount);
tlog(`initialized`);
}
return makeDefaultExo('coordinator', {
async bootstrap(vats: { worker1: unknown; worker2: unknown }) {
tlog(`bootstrap called`);
if (!baggage.has('workers')) {
// Store worker references for persistence
workers = {
worker1: vats.worker1,
worker2: vats.worker2,
};
baggage.init('workers', harden(workers));
tlog(`stored ${Object.keys(workers).length} workers`);
}
return `Coordinator initialized with ${Object.keys(workers).length} workers`;
},
async resume() {
tlog(`resume called`);
if (!workers?.worker1 || !workers.worker2) {
tlog(`no workers available`);
return `No workers available`;
}
const [result1, result2] = await Promise.all([
E(workers.worker1).resume(),
E(workers.worker2).resume(),
]);
workCount += 1;
baggage.set('workCount', workCount);
const message = `Work completed: ${result1}, ${result2}`;
tlog(message);
return message;
},
});
}
|