feat: add ELU tasks accounting

[poolifier.git] / src / pools / abstract-pool.ts

diff --git a/src/pools/abstract-pool.ts b/src/pools/abstract-pool.ts
index 5088f3d90957bd276756c120d2496e8ffe01bb5a..065b228ada73e6d21f6fb8238bddf403cc04ac83 100644 (file)
--- a/src/pools/abstract-pool.ts
+++ b/src/pools/abstract-pool.ts
@@ -1,4 +1,5 @@
 import crypto from 'node:crypto'
+import { performance } from 'node:perf_hooks'
 import type { MessageValue, PromiseResponseWrapper } from '../utility-types'
 import {
   DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS,
@@ -299,7 +300,8 @@ export abstract class AbstractPool<
         waitTimeHistory: new CircularArray(),
         avgWaitTime: 0,
         medWaitTime: 0,
-        error: 0
+        error: 0,
+        elu: undefined
       })
     }
     this.workerChoiceStrategyContext.setWorkerChoiceStrategy(
@@ -474,6 +476,7 @@ export abstract class AbstractPool<
     }
     this.updateRunTimeTasksUsage(workerTasksUsage, message)
     this.updateWaitTimeTasksUsage(workerTasksUsage, message)
+    this.updateEluTasksUsage(workerTasksUsage, message)
   }
 
   private updateRunTimeTasksUsage (
@@ -522,6 +525,25 @@ export abstract class AbstractPool<
     }
   }
 
+  private updateEluTasksUsage (
+    workerTasksUsage: TasksUsage,
+    message: MessageValue<Response>
+  ): void {
+    if (this.workerChoiceStrategyContext.getRequiredStatistics().elu) {
+      if (workerTasksUsage.elu != null && message.elu != null) {
+        // TODO: cumulative or delta?
+        workerTasksUsage.elu = {
+          idle: workerTasksUsage.elu.idle + message.elu.idle,
+          active: workerTasksUsage.elu.active + message.elu.active,
+          utilization:
+            workerTasksUsage.elu.utilization + message.elu.utilization
+        }
+      } else if (message.elu != null) {
+        workerTasksUsage.elu = message.elu
+      }
+    }
+  }
+
   /**
    * Chooses a worker node for the next task.
    *
@@ -704,7 +726,8 @@ export abstract class AbstractPool<
         waitTimeHistory: new CircularArray(),
         avgWaitTime: 0,
         medWaitTime: 0,
-        error: 0
+        error: 0,
+        elu: undefined
       },
       tasksQueue: new Queue<Task<Data>>()
     })