-import crypto from 'crypto'
-import fs from 'fs'
+import crypto from 'node:crypto'
+import fs from 'node:fs'
import {
DynamicClusterPool,
DynamicThreadPool,
FixedClusterPool,
- FixedThreadPool
+ FixedThreadPool,
+ PoolTypes,
+ WorkerTypes
} from '../lib/index.mjs'
-import { PoolTypes, WorkerFunctions, WorkerTypes } from './benchmarks-types.mjs'
+import { TaskFunctions } from './benchmarks-types.mjs'
-async function runTest (pool, { taskExecutions, workerData }) {
+export const buildPoolifierPool = (
+ workerType,
+ poolType,
+ poolSize,
+ poolOptions
+) => {
+ switch (poolType) {
+ case PoolTypes.fixed:
+ switch (workerType) {
+ case WorkerTypes.thread:
+ return new FixedThreadPool(
+ poolSize,
+ './benchmarks/internal/thread-worker.mjs',
+ poolOptions
+ )
+ case WorkerTypes.cluster:
+ return new FixedClusterPool(
+ poolSize,
+ './benchmarks/internal/cluster-worker.mjs',
+ poolOptions
+ )
+ }
+ break
+ case PoolTypes.dynamic:
+ switch (workerType) {
+ case WorkerTypes.thread:
+ return new DynamicThreadPool(
+ Math.floor(poolSize / 2),
+ poolSize,
+ './benchmarks/internal/thread-worker.mjs',
+ poolOptions
+ )
+ case WorkerTypes.cluster:
+ return new DynamicClusterPool(
+ Math.floor(poolSize / 2),
+ poolSize,
+ './benchmarks/internal/cluster-worker.mjs',
+ poolOptions
+ )
+ }
+ break
+ }
+}
+
+export const runPoolifierTest = async (
+ pool,
+ { taskExecutions, workerData }
+) => {
return new Promise((resolve, reject) => {
let executions = 0
for (let i = 1; i <= taskExecutions; i++) {
})
}
-function generateRandomInteger (max = Number.MAX_SAFE_INTEGER, min = 0) {
+export const getPoolImplementationName = pool => {
+ if (pool instanceof FixedThreadPool) {
+ return 'FixedThreadPool'
+ } else if (pool instanceof DynamicThreadPool) {
+ return 'DynamicThreadPool'
+ } else if (pool instanceof FixedClusterPool) {
+ return 'FixedClusterPool'
+ } else if (pool instanceof DynamicClusterPool) {
+ return 'DynamicClusterPool'
+ }
+}
+
+export const LIST_FORMATTER = new Intl.ListFormat('en-US', {
+ style: 'long',
+ type: 'conjunction'
+})
+
+export const executeAsyncFn = async fn => {
+ try {
+ await fn()
+ } catch (e) {
+ console.error(e)
+ // eslint-disable-next-line n/no-process-exit
+ process.exit(1)
+ }
+}
+
+export const generateRandomInteger = (
+ max = Number.MAX_SAFE_INTEGER,
+ min = 0
+) => {
if (max < min || max < 0 || min < 0) {
throw new RangeError('Invalid interval')
}
return Math.floor(Math.random() * (max + 1))
}
-function jsonIntegerSerialization (n) {
+const jsonIntegerSerialization = n => {
for (let i = 0; i < n; i++) {
const o = {
a: i
}
JSON.stringify(o)
}
+ return { ok: 1 }
}
/**
* @param {number} n - The number of fibonacci numbers to generate.
* @returns {number} - The nth fibonacci number.
*/
-function fibonacci (n) {
- if (n <= 1) return 1
+const fibonacci = n => {
+ if (n <= 1) return n
return fibonacci(n - 1) + fibonacci(n - 2)
}
* @param {number} n - The number to calculate the factorial of.
* @returns {number} - The factorial of n.
*/
-function factorial (n) {
+const factorial = n => {
if (n === 0) {
return 1
}
return factorial(n - 1) * n
}
-function readWriteFiles (
+const readWriteFiles = (
n,
baseDirectory = `/tmp/poolifier-benchmarks/${crypto.randomInt(
281474976710655
)}`
-) {
+) => {
if (fs.existsSync(baseDirectory) === true) {
fs.rmSync(baseDirectory, { recursive: true })
}
fs.readFileSync(filePath, 'utf8')
}
fs.rmSync(baseDirectory, { recursive: true })
+ return { ok: 1 }
}
-function executeWorkerFunction (data) {
+export const executeTaskFunction = data => {
switch (data.function) {
- case WorkerFunctions.jsonIntegerSerialization:
+ case TaskFunctions.jsonIntegerSerialization:
return jsonIntegerSerialization(data.taskSize || 1000)
- case WorkerFunctions.fibonacci:
+ case TaskFunctions.fibonacci:
return fibonacci(data.taskSize || 1000)
- case WorkerFunctions.factorial:
+ case TaskFunctions.factorial:
return factorial(data.taskSize || 1000)
- case WorkerFunctions.readWriteFiles:
+ case TaskFunctions.readWriteFiles:
return readWriteFiles(data.taskSize || 1000)
default:
- throw new Error('Unknown worker function')
+ throw new Error('Unknown task function')
}
}
-
-function buildPool (workerType, poolType, poolSize, poolOptions) {
- switch (poolType) {
- case PoolTypes.fixed:
- switch (workerType) {
- case WorkerTypes.thread:
- return new FixedThreadPool(
- poolSize,
- './benchmarks/internal/thread-worker.mjs',
- poolOptions
- )
- case WorkerTypes.cluster:
- return new FixedClusterPool(
- poolSize,
- './benchmarks/internal/cluster-worker.mjs',
- poolOptions
- )
- }
- break
- case PoolTypes.dynamic:
- switch (workerType) {
- case WorkerTypes.thread:
- return new DynamicThreadPool(
- poolSize / 2,
- poolSize * 3,
- './benchmarks/internal/thread-worker.mjs',
- poolOptions
- )
- case WorkerTypes.cluster:
- return new DynamicClusterPool(
- poolSize / 2,
- poolSize * 3,
- './benchmarks/internal/cluster-worker.mjs',
- poolOptions
- )
- }
- break
- }
-}
-
-export {
- WorkerFunctions,
- buildPool,
- executeWorkerFunction,
- generateRandomInteger,
- readWriteFiles,
- runTest
-}