githubmirror/node

benchmark/run.js
122 строки4 KB
Rafael Gonzaga
benchmark: add --track to benchmark
26 июл 2025, 02:32
26 июл 2025, 02:3228538f2
'use strict';

const path = require('path');
const { spawn, fork } = require('node:child_process');
const CLI = require('./_cli.js');
const { styleText } = require('node:util');

const cli = new CLI(`usage: ./node run.js [options] [--] <category> ...
  Run each benchmark in the <category> directory a single time, more than one
  <category> directory can be specified.

  --filter   pattern        includes only benchmark scripts matching <pattern>
                            (can be repeated)
  --exclude  pattern        excludes scripts matching <pattern> (can be
                            repeated)
  --runs   variable=value   set the amount of benchmark suite execution.
                            Default: 1
  --set    variable=value   set benchmark variable (can be repeated)
  --format [simple|csv]     optional value that specifies the output format
  --track                   Display the time elapsed to run each benchmark file.
  test                      only run a single configuration from the options
                            matrix
  all                       each benchmark category is run one after the other

  Examples:
    --set CPUSET=0            Runs benchmarks on CPU core 0.
    --set CPUSET=0-2          Specifies that benchmarks should run on CPU cores 0 to 2.

  Note: The CPUSET format should match the specifications of the 'taskset' command on your system.
`, { arrayArgs: ['set', 'filter', 'exclude'], boolArgs: ['track'] });

const benchmarks = cli.benchmarks();

if (benchmarks.length === 0) {
  console.error('No benchmarks found');
  process.exitCode = 1;
  return;
}

const validFormats = ['csv', 'simple'];
const format = cli.optional.format || 'simple';
if (!validFormats.includes(format)) {
  console.error('Invalid format detected');
  process.exitCode = 1;
  return;
}

if (format === 'csv') {
  console.log('"filename", "configuration", "rate", "time"');
}

function runBenchmark(filename) {
  const scriptPath = path.resolve(__dirname, filename);

  const args = cli.test ? ['--test'] : cli.optional.set;
  const cpuCore = cli.getCpuCoreSetting();
  let child;
  if (cpuCore !== null) {
    child = spawn('taskset', ['-c', cpuCore, 'node', scriptPath, ...args], {
      stdio: ['inherit', 'inherit', 'inherit', 'ipc'],
    });
  } else {
    child = fork(
      scriptPath,
      args,
    );
  }

  return new Promise((resolve, reject) => {
    child.on('message', (data) => {
      if (data.type !== 'report') {
        return;
      }
      // Construct configuration string, " A=a, B=b, ..."
      let conf = '';
      for (const key of Object.keys(data.conf)) {
        if (conf !== '')
          conf += ' ';
        conf += `${key}=${JSON.stringify(data.conf[key])}`;
      }
      if (format === 'csv') {
        // Escape quotes (") for correct csv formatting
        conf = conf.replace(/"/g, '""');
        console.log(`"${data.name}", "${conf}", ${data.rate}, ${data.time}`);
      } else {
        let rate = data.rate.toString().split('.');
        rate[0] = rate[0].replace(/(\d)(?=(?:\d\d\d)+(?!\d))/g, '$1,');
        rate = (rate[1] ? rate.join('.') : rate[0]);
        console.log(`${data.name} ${conf}: ${rate}`);
      }
    });
    child.once('close', (code) => {
      if (code) {
        reject(code);
      } else {
        resolve(code);
      }
    });
  });
}

async function run() {
  for (let i = 0; i < benchmarks.length; ++i) {
    let runs = cli.optional.runs ?? 1;
    const filename = benchmarks[i];
    if (format !== 'csv') {
      console.log();
      console.log(filename);
    }

    while (runs-- > 0) {
      const start = performance.now();
      await runBenchmark(filename);
      if (format !== 'csv' && cli.optional.track) {
        const ms = styleText(['bold', 'yellow'], `${Math.round(performance.now() - start)}ms`);
        console.log(`[${ms}] ${filename}`);
      }
    }
  }
}

run();