transfer from monorepo

worker/tests/simulateTask.ts

@@ -0,0 +1,84 @@
+import { taskRunner } from "@_koii/task-manager";
+
+import "../src/index.js";
+import { namespaceWrapper } from "@_koii/task-manager/namespace-wrapper";
+import { Keypair } from "@_koii/web3.js";
+
+const numRounds = parseInt(process.argv[2]) || 1;
+const roundDelay = parseInt(process.argv[3]) || 5000;
+const functionDelay = parseInt(process.argv[4]) || 1000;
+
+let TASK_TIMES: number[] = [];
+let SUBMISSION_TIMES: number[] = [];
+let AUDIT_TIMES: number[] = [];
+
+function sleep(ms: number) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
+
+async function executeTasks() {
+  const keypair = Keypair.generate();
+  await namespaceWrapper.stakeOnChain(keypair.publicKey, keypair, keypair.publicKey, 10000);
+  for (let round = 0; round < numRounds; round++) {
+    const taskStartTime = Date.now();
+    await taskRunner.task(round);
+    const taskEndTime = Date.now();
+    TASK_TIMES.push(taskEndTime - taskStartTime);
+    await sleep(functionDelay);
+
+    const taskSubmissionStartTime = Date.now();
+    await taskRunner.submitTask(round);
+    const taskSubmissionEndTime = Date.now();
+    SUBMISSION_TIMES.push(taskSubmissionEndTime - taskSubmissionStartTime);
+    await sleep(functionDelay);
+
+    const auditStartTime = Date.now();
+    await taskRunner.auditTask(round);
+    const auditEndTime = Date.now();
+    AUDIT_TIMES.push(auditEndTime - auditStartTime);
+    await sleep(functionDelay);
+
+    await taskRunner.selectAndGenerateDistributionList(round);
+    await sleep(functionDelay);
+
+    await taskRunner.auditDistribution(round);
+
+    if (round < numRounds - 1) {
+      await sleep(roundDelay);
+    }
+  }
+  console.log("TIME METRICS BELOW");
+  function metrics(name: string, times: number[]) {
+    const average = (arr: number[]) => arr.reduce((a, b) => a + b, 0) / arr.length;
+    const formatTime = (ms: number) => (ms / 1000).toFixed(4);
+    const formatSlot = (ms: number) => Math.ceil(ms / 408);
+    const min = Math.min(...times);
+    const max = Math.max(...times);
+    const avg = average(times);
+    const timeMin = formatTime(min);
+    const timeMax = formatTime(max);
+    const timeAvg = formatTime(avg);
+    const slotMin = formatSlot(min);
+    const slotMax = formatSlot(max);
+    const slotAvg = formatSlot(avg);
+
+    return {
+      Metric: `SIMULATED ${name} WINDOW`,
+      "Avg Time (s)": timeAvg,
+      "Avg Slots": slotAvg,
+      "Min Time (s)": timeMin,
+      "Min Slots": slotMin,
+      "Max Time (s)": timeMax,
+      "Max Slots": slotMax,
+    };
+  }
+  const timeMetrics = metrics("TASK", TASK_TIMES);
+  const submissionMetrics = metrics("SUBMISSION", SUBMISSION_TIMES);
+  const auditMetrics = metrics("AUDIT", AUDIT_TIMES);
+
+  console.table([timeMetrics, submissionMetrics, auditMetrics]);
+
+  console.log("All tasks executed. Test completed.");
+  process.exit(0);
+}
+setTimeout(executeTasks, 1500);