-import Benchmark from 'benchmark'
+import { writeFileSync } from 'node:fs'
+import { env } from 'node:process'
+// eslint-disable-next-line n/no-unsupported-features/node-builtins
+import { parseArgs } from 'node:util'
+
import {
- Measurements,
+ availableParallelism,
PoolTypes,
- WorkerChoiceStrategies,
WorkerTypes,
- availableParallelism
} from '../../lib/index.mjs'
-import { TaskFunctions } from '../benchmarks-types.mjs'
+import { TaskFunctions } from '../benchmarks-types.cjs'
import {
- LIST_FORMATTER,
- buildPoolifierPool,
- runPoolifierTest
+ convertTatamiNgToBmf,
+ runPoolifierBenchmarkTatamiNg,
} from '../benchmarks-utils.mjs'
-const poolifierSuite = new Benchmark.Suite('Poolifier', {
- onCycle: event => {
- console.info(event.target.toString())
- },
- onComplete: function () {
- console.info(
- 'Fastest is ' + LIST_FORMATTER.format(this.filter('fastest').map('name'))
- )
- }
-})
-
const poolSize = availableParallelism()
-const benchmarkSettings = []
-for (const poolType of Object.values(PoolTypes)) {
- for (const workerType of Object.values(WorkerTypes)) {
- if (workerType === WorkerTypes.cluster) {
- continue
- }
- for (const workerChoiceStrategy of Object.values(WorkerChoiceStrategies)) {
- for (const enableTasksQueue of [false, true]) {
- if (workerChoiceStrategy === WorkerChoiceStrategies.FAIR_SHARE) {
- for (const measurement of [Measurements.runTime, Measurements.elu]) {
- benchmarkSettings.push([
- `${poolType}|${workerType}|${workerChoiceStrategy}|tasks queue:${enableTasksQueue}|measurement:${measurement}`,
- workerType,
- poolType,
- poolSize,
- {
- workerChoiceStrategy,
- workerChoiceStrategyOptions: {
- measurement
- },
- enableTasksQueue
- }
- ])
- }
- } else {
- benchmarkSettings.push([
- `${poolType}|${workerType}|${workerChoiceStrategy}|tasks queue:${enableTasksQueue}`,
- workerType,
- poolType,
- poolSize,
- {
- workerChoiceStrategy,
- enableTasksQueue
- }
- ])
- }
- }
- }
- }
-}
-
const taskExecutions = 1
const workerData = {
- function: TaskFunctions.jsonIntegerSerialization,
- taskSize: 100
+ function: TaskFunctions.factorial,
+ taskSize: 1000,
}
+const benchmarkReportFile = 'benchmark-report.json'
+let benchmarkReport
-for (const [
- name,
- workerType,
- poolType,
- poolSize,
- poolOptions
-] of benchmarkSettings) {
- poolifierSuite.add(name, async () => {
- const pool = buildPoolifierPool(workerType, poolType, poolSize, poolOptions)
- await runPoolifierTest(pool, {
- taskExecutions,
- workerData
- })
- await pool.destroy()
- })
+switch (
+ parseArgs({
+ args: process.argv,
+ options: {
+ type: {
+ type: 'string',
+ short: 't',
+ },
+ },
+ strict: true,
+ allowPositionals: true,
+ }).values.type
+) {
+ case 'tatami-ng':
+ default:
+ benchmarkReport = convertTatamiNgToBmf(
+ await runPoolifierBenchmarkTatamiNg(
+ 'FixedThreadPool',
+ WorkerTypes.thread,
+ PoolTypes.fixed,
+ poolSize,
+ {
+ taskExecutions,
+ workerData,
+ }
+ )
+ )
+ benchmarkReport = {
+ ...benchmarkReport,
+ ...convertTatamiNgToBmf(
+ await runPoolifierBenchmarkTatamiNg(
+ 'DynamicThreadPool',
+ WorkerTypes.thread,
+ PoolTypes.dynamic,
+ poolSize,
+ {
+ taskExecutions,
+ workerData,
+ }
+ )
+ ),
+ }
+ benchmarkReport = {
+ ...benchmarkReport,
+ ...convertTatamiNgToBmf(
+ await runPoolifierBenchmarkTatamiNg(
+ 'FixedClusterPool',
+ WorkerTypes.cluster,
+ PoolTypes.fixed,
+ poolSize,
+ {
+ taskExecutions,
+ workerData,
+ }
+ )
+ ),
+ }
+ benchmarkReport = {
+ ...benchmarkReport,
+ ...convertTatamiNgToBmf(
+ await runPoolifierBenchmarkTatamiNg(
+ 'DynamicClusterPool',
+ WorkerTypes.cluster,
+ PoolTypes.dynamic,
+ poolSize,
+ {
+ taskExecutions,
+ workerData,
+ }
+ )
+ ),
+ }
+ // eslint-disable-next-line @typescript-eslint/no-unused-expressions
+ env.CI != null &&
+ writeFileSync(benchmarkReportFile, JSON.stringify(benchmarkReport))
+ break
}
-
-poolifierSuite.run({ async: true })