From 7a3df0a18465ceebe43f3183daa2f9397c4e5ebb Mon Sep 17 00:00:00 2001
From: andy finch <andyfinch7@gmail.com>
Date: Fri, 5 Apr 2019 15:57:59 -0400
Subject: Add worker benchmarks (#2059)

---
 tests/subdir/bench_worker.ts       | 20 ++++++++++
 tests/workers_round_robin_bench.ts | 75 ++++++++++++++++++++++++++++++++++++++
 tests/workers_startup_bench.ts     | 25 +++++++++++++
 3 files changed, 120 insertions(+)
 create mode 100644 tests/subdir/bench_worker.ts
 create mode 100644 tests/workers_round_robin_bench.ts
 create mode 100644 tests/workers_startup_bench.ts

(limited to 'tests')

diff --git a/tests/subdir/bench_worker.ts b/tests/subdir/bench_worker.ts
new file mode 100644
index 000000000..6dd2f9541
--- /dev/null
+++ b/tests/subdir/bench_worker.ts
@@ -0,0 +1,20 @@
+onmessage = function(e) {
+  const { cmdId, action, data } = e.data;
+  switch (action) {
+    case 0: // Static response
+      postMessage({
+        cmdId,
+        data: "HTTP/1.1 200 OK\r\nContent-Length: 12\r\n\r\nHello World\n"
+      });
+      break;
+    case 1: // Respond with request data
+      postMessage({ cmdId, data });
+      break;
+    case 2: // Ping
+      postMessage({ cmdId });
+      break;
+    case 3: // Close
+      workerClose();
+      break;
+  }
+};
diff --git a/tests/workers_round_robin_bench.ts b/tests/workers_round_robin_bench.ts
new file mode 100644
index 000000000..818a1145f
--- /dev/null
+++ b/tests/workers_round_robin_bench.ts
@@ -0,0 +1,75 @@
+// 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) => {
+    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 (var i = 1; i <= workerCount; ++i) {
+    const worker = new Worker("tests/subdir/bench_worker.ts");
+    const promise = new Promise(resolve => {
+      worker.onmessage = e => {
+        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 => {
+      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();
diff --git a/tests/workers_startup_bench.ts b/tests/workers_startup_bench.ts
new file mode 100644
index 000000000..46b2b2801
--- /dev/null
+++ b/tests/workers_startup_bench.ts
@@ -0,0 +1,25 @@
+// Benchmark measures time it takes to start and stop a number of workers.
+const workerCount = 50;
+
+async function bench(): Promise<void> {
+  const workers: Worker[] = [];
+  for (var i = 1; i <= workerCount; ++i) {
+    const worker = new Worker("tests/subdir/bench_worker.ts");
+    const promise = new Promise(resolve => {
+      worker.onmessage = e => {
+        if (e.data.cmdId === 0) resolve();
+      };
+    });
+    worker.postMessage({ cmdId: 0, action: 2 });
+    await promise;
+    workers.push(worker);
+  }
+  console.log("Done creating workers closing workers!");
+  for (const worker of workers) {
+    worker.postMessage({ action: 3 });
+    await worker.closed; // Required to avoid a cmdId not in table error.
+  }
+  console.log("Finished!");
+}
+
+bench();
-- 
cgit v1.2.3