can1357
3ab675a83b
- Added `WorkerInbox` and `installWorkerInbox(port)` to queue worker messages before bind. - Added `consumeWorkerInbox()` to replay buffered messages and clear one active inbox. - Added buffered inbox consumption in JS and tab worker transports before direct message handlers. - Normalized worker selector arguments to the `__omp_worker_*` naming across workers and tests.
30 lines
1.1 KiB
TypeScript
30 lines
1.1 KiB
TypeScript
import { parentPort } from "node:worker_threads";
|
|
import { consumeWorkerInbox } from "@oh-my-pi/pi-utils/worker-host";
|
|
import type { Transport, WorkerInbound, WorkerOutbound } from "./tab-protocol";
|
|
import { WorkerCore } from "./tab-worker";
|
|
|
|
if (!parentPort) throw new Error("tab-worker-entry: missing parentPort");
|
|
|
|
const port = parentPort;
|
|
// When the CLI host pre-buffered messages (it imports this module dynamically),
|
|
// bind that inbox so the parent's already-delivered `init` is replayed. Loaded
|
|
// directly (test/SDK fallback), this module's top-level runs synchronously at
|
|
// worker start, so the direct `parentPort.on` below wins the flush on its own.
|
|
const inbox = consumeWorkerInbox();
|
|
const transport: Transport = {
|
|
send(msg, transferList) {
|
|
port.postMessage(msg, transferList ?? []);
|
|
},
|
|
onMessage(handler) {
|
|
if (inbox) return inbox.bind(data => handler(data as WorkerOutbound | WorkerInbound));
|
|
const wrap = (message: unknown): void => handler(message as WorkerOutbound | WorkerInbound);
|
|
port.on("message", wrap);
|
|
return () => port.off("message", wrap);
|
|
},
|
|
close() {
|
|
port.close();
|
|
},
|
|
};
|
|
|
|
new WorkerCore(transport);
|