From: Jérôme Benoit <jerome.benoit@piment-noir.org>
Date: Thu, 21 Dec 2023 10:45:57 +0000 (+0100)
Subject: fix: make worker weight default computation more robust
X-Git-Tag: v3.1.8~2
X-Git-Url: https://git.piment-noir.org/?a=commitdiff_plain;h=19910d842d15c3937508060925bec675609099d7;p=poolifier.git

fix: make worker weight default computation more robust

Signed-off-by: Jérôme Benoit <jerome.benoit@piment-noir.org>
---

diff --git a/src/utils.ts b/src/utils.ts
index 3226e7eb..f7aab637 100644
--- a/src/utils.ts
+++ b/src/utils.ts
@@ -1,5 +1,5 @@
 import * as os from 'node:os'
-import { getRandomValues } from 'node:crypto'
+import { getRandomValues, randomInt } from 'node:crypto'
 import { Worker as ClusterWorker } from 'node:cluster'
 import { Worker as ThreadWorker } from 'node:worker_threads'
 import { cpus } from 'node:os'
@@ -315,8 +315,14 @@ const getDefaultWeights = (
 }
 
 const getDefaultWorkerWeight = (): number => {
+  const cpuSpeed = randomInt(500, 2500)
   let cpusCycleTimeWeight = 0
   for (const cpu of cpus()) {
+    if (cpu.speed == null || cpu.speed === 0) {
+      cpu.speed =
+        cpus().find(cpu => cpu.speed != null && cpu.speed !== 0)?.speed ??
+        cpuSpeed
+    }
     // CPU estimated cycle time
     const numberOfDigits = cpu.speed.toString().length - 1
     const cpuCycleTime = 1 / (cpu.speed / Math.pow(10, numberOfDigits))