2019-04-05 19:57:59 +00:00
|
|
|
// Benchmark measures time it takes to send a message to a group of workers one
|
|
|
|
// at a time and wait for a response from all of them. Just a general
|
|
|
|
// throughput and consistency benchmark.
|
|
|
|
const data = "HTTP/1.1 200 OK\r\nContent-Length: 12\r\n\r\nHello World\n";
|
|
|
|
const workerCount = 4;
|
|
|
|
const cmdsPerWorker = 400;
|
|
|
|
|
|
|
|
function handleAsyncMsgFromWorker(
|
2023-11-22 11:11:20 +00:00
|
|
|
promiseTable: Map<number, ReturnType<typeof Promise.withResolvers<string>>>,
|
2020-07-14 19:24:17 +00:00
|
|
|
msg: { cmdId: number; data: string },
|
2021-08-05 11:08:58 +00:00
|
|
|
) {
|
2019-04-05 19:57:59 +00:00
|
|
|
const promise = promiseTable.get(msg.cmdId);
|
|
|
|
if (promise === null) {
|
|
|
|
throw new Error(`Failed to find promise: cmdId: ${msg.cmdId}, msg: ${msg}`);
|
|
|
|
}
|
2020-05-21 11:08:43 +00:00
|
|
|
promise?.resolve(data);
|
2019-04-05 19:57:59 +00:00
|
|
|
}
|
|
|
|
|
2021-08-05 11:08:58 +00:00
|
|
|
async function main() {
|
2023-11-22 11:11:20 +00:00
|
|
|
const workers: Array<
|
|
|
|
[Map<number, ReturnType<typeof Promise.withResolvers<string>>>, Worker]
|
|
|
|
> = [];
|
2019-09-07 16:27:18 +00:00
|
|
|
for (let i = 1; i <= workerCount; ++i) {
|
2020-06-09 12:33:52 +00:00
|
|
|
const worker = new Worker(
|
2022-07-20 14:56:53 +00:00
|
|
|
import.meta.resolve("./bench_worker.ts"),
|
2020-07-14 19:24:17 +00:00
|
|
|
{ type: "module" },
|
2020-06-09 12:33:52 +00:00
|
|
|
);
|
2023-11-22 11:11:20 +00:00
|
|
|
const { promise, resolve } = Promise.withResolvers<void>();
|
2021-08-05 11:08:58 +00:00
|
|
|
worker.onmessage = (e) => {
|
2023-11-22 11:11:20 +00:00
|
|
|
if (e.data.cmdId === 0) resolve();
|
2020-01-21 08:49:47 +00:00
|
|
|
};
|
2019-04-05 19:57:59 +00:00
|
|
|
worker.postMessage({ cmdId: 0, action: 2 });
|
|
|
|
await promise;
|
|
|
|
workers.push([new Map(), worker]);
|
|
|
|
}
|
|
|
|
// assign callback function
|
|
|
|
for (const [promiseTable, worker] of workers) {
|
2021-08-05 11:08:58 +00:00
|
|
|
worker.onmessage = (e) => {
|
2019-04-05 19:57:59 +00:00
|
|
|
handleAsyncMsgFromWorker(promiseTable, e.data);
|
|
|
|
};
|
|
|
|
}
|
|
|
|
for (const cmdId of Array(cmdsPerWorker).keys()) {
|
|
|
|
const promises: Array<Promise<string>> = [];
|
|
|
|
for (const [promiseTable, worker] of workers) {
|
2023-11-22 11:11:20 +00:00
|
|
|
const deferred = Promise.withResolvers<string>();
|
|
|
|
promiseTable.set(cmdId, deferred);
|
2019-04-05 19:57:59 +00:00
|
|
|
worker.postMessage({ cmdId: cmdId, action: 1, data });
|
2023-11-22 11:11:20 +00:00
|
|
|
promises.push(deferred.promise);
|
2019-04-05 19:57:59 +00:00
|
|
|
}
|
|
|
|
for (const promise of promises) {
|
|
|
|
await promise;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
for (const [, worker] of workers) {
|
2023-11-22 11:11:20 +00:00
|
|
|
const { promise, resolve } = Promise.withResolvers<void>();
|
2021-08-05 11:08:58 +00:00
|
|
|
worker.onmessage = (e) => {
|
2023-11-22 11:11:20 +00:00
|
|
|
if (e.data.cmdId === 3) resolve();
|
2020-01-21 08:49:47 +00:00
|
|
|
};
|
2019-04-05 19:57:59 +00:00
|
|
|
worker.postMessage({ action: 3 });
|
2020-01-21 08:49:47 +00:00
|
|
|
await promise;
|
2019-04-05 19:57:59 +00:00
|
|
|
}
|
|
|
|
console.log("Finished!");
|
|
|
|
}
|
|
|
|
|
|
|
|
main();
|