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
|
// 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;
import {
Deferred,
deferred,
} from "../../../../test_util/std/async/deferred.ts";
function handleAsyncMsgFromWorker(
promiseTable: Map<number, Deferred<string>>,
msg: { cmdId: number; data: string },
) {
const promise = promiseTable.get(msg.cmdId);
if (promise === null) {
throw new Error(`Failed to find promise: cmdId: ${msg.cmdId}, msg: ${msg}`);
}
promise?.resolve(data);
}
async function main() {
const workers: Array<[Map<number, Deferred<string>>, Worker]> = [];
for (let i = 1; i <= workerCount; ++i) {
const worker = new Worker(
new URL("bench_worker.ts", import.meta.url).href,
{ type: "module" },
);
const promise = deferred();
worker.onmessage = (e) => {
if (e.data.cmdId === 0) promise.resolve();
};
worker.postMessage({ cmdId: 0, action: 2 });
await promise;
workers.push([new Map(), worker]);
}
// assign callback function
for (const [promiseTable, worker] of workers) {
worker.onmessage = (e) => {
handleAsyncMsgFromWorker(promiseTable, e.data);
};
}
for (const cmdId of Array(cmdsPerWorker).keys()) {
const promises: Array<Promise<string>> = [];
for (const [promiseTable, worker] of workers) {
const promise = deferred<string>();
promiseTable.set(cmdId, promise);
worker.postMessage({ cmdId: cmdId, action: 1, data });
promises.push(promise);
}
for (const promise of promises) {
await promise;
}
}
for (const [, worker] of workers) {
const promise = deferred();
worker.onmessage = (e) => {
if (e.data.cmdId === 3) promise.resolve();
};
worker.postMessage({ action: 3 });
await promise;
}
console.log("Finished!");
}
main();
|
