fix: handle empty data set in median computation

[poolifier.git] / src / utils.ts

diff --git a/src/utils.ts b/src/utils.ts
index ed118da6b5e10165d23c2f6820af73e9ed7f02bb..403fed951bb21f7af0474b65cc8a02fbec6d5a78 100644 (file)
--- a/src/utils.ts
+++ b/src/utils.ts
@@ -22,15 +22,14 @@ export const DEFAULT_WORKER_CHOICE_STRATEGY_OPTIONS: WorkerChoiceStrategyOptions
  * @returns The median of the given data set.
  */
 export const median = (dataSet: number[]): number => {
+  if (Array.isArray(dataSet) && dataSet.length === 0) {
+    return 0
+  }
   if (Array.isArray(dataSet) && dataSet.length === 1) {
     return dataSet[0]
   }
-  const sortedDataSet = dataSet.slice().sort((a, b) => a - b)
-  const middleIndex = Math.floor(sortedDataSet.length / 2)
-  if (sortedDataSet.length % 2 === 0) {
-    return sortedDataSet[middleIndex]
-  }
-  return (sortedDataSet[middleIndex - 1] + sortedDataSet[middleIndex]) / 2
+  dataSet = dataSet.slice().sort((a, b) => a - b)
+  return (dataSet[(dataSet.length - 1) >> 1] + dataSet[dataSet.length >> 1]) / 2
 }
 
 export const isPlainObject = (obj: unknown): boolean =>