1 files changed, 79 insertions, 0 deletions
diff --git a/cli/tests/workers_round_robin_bench.ts b/cli/tests/workers_round_robin_bench.ts
new file mode 100644
index 000000000..7c34e75e5
--- /dev/null
+++ b/cli/tests/workers_round_robin_bench.ts
@@ -0,0 +1,79 @@
+// 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;
+
+export interface ResolvableMethods<T> {
+  resolve: (value?: T | PromiseLike<T>) => void;
+  // eslint-disable-next-line @typescript-eslint/no-explicit-any
+  reject: (reason?: any) => void;
+}
+
+export type Resolvable<T> = Promise<T> & ResolvableMethods<T>;
+
+export function createResolvable<T>(): Resolvable<T> {
+  let methods: ResolvableMethods<T>;
+  const promise = new Promise<T>(
+    (resolve, reject): void => {
+      methods = { resolve, reject };
+    }
+  );
+  // TypeScript doesn't know that the Promise callback occurs synchronously
+  // therefore use of not null assertion (`!`)
+  return Object.assign(promise, methods!) as Resolvable<T>;
+}
+
+function handleAsyncMsgFromWorker(
+  promiseTable: Map<number, Resolvable<string>>,
+  msg: { cmdId: number; data: string }
+): void {
+  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(): Promise<void> {
+  const workers: Array<[Map<number, Resolvable<string>>, Worker]> = [];
+  for (let i = 1; i <= workerCount; ++i) {
+    const worker = new Worker("./subdir/bench_worker.ts");
+    const promise = new Promise(
+      (resolve): void => {
+        worker.onmessage = (e): void => {
+          if (e.data.cmdId === 0) 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): void => {
+      handleAsyncMsgFromWorker(promiseTable, e.data);
+    };
+  }
+  for (const cmdId of Array(cmdsPerWorker).keys()) {
+    const promises: Array<Promise<string>> = [];
+    for (const [promiseTable, worker] of workers) {
+      const promise = createResolvable<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) {
+    worker.postMessage({ action: 3 });
+    await worker.closed; // Required to avoid a cmdId not in table error.
+  }
+  console.log("Finished!");
+}
+
+main();