githubmirror/node

benchmark/zlib/pipe.js
52 строки1 KB
Jan Krems
zlib: add zstd support
08 фев 2025, 15:43
08 фев 2025, 15:43bf12d72
'use strict';
const common = require('../common.js');
const fs = require('fs');
const zlib = require('zlib');

const bench = common.createBenchmark(main, {
  inputLen: [1024],
  duration: [5],
  type: ['string', 'buffer'],
  algorithm: ['gzip', 'brotli', 'zstd'],
}, {
  test: {
    inputLen: 1024,
    duration: 0.2,
  },
});

const algorithms = {
  'gzip': [zlib.createGzip, zlib.createGunzip],
  'brotli': [zlib.createBrotliCompress, zlib.createBrotliDecompress],
  'zstd': [zlib.createZstdCompress, zlib.createZstdDecompress],
};

function main({ inputLen, duration, type, algorithm }) {
  const buffer = Buffer.alloc(inputLen, fs.readFileSync(__filename));
  const chunk = type === 'buffer' ? buffer : buffer.toString('utf8');

  const [createCompress, createUncompress] = algorithms[algorithm];
  const input = createCompress();
  const output = createUncompress();

  let readFromOutput = 0;
  input.pipe(output);
  if (type === 'string')
    output.setEncoding('utf8');
  output.on('data', (chunk) => readFromOutput += chunk.length);

  function write() {
    input.write(chunk, write);
  }

  bench.start();
  write();

  setTimeout(() => {
    // Give result in GBit/s, like the net benchmarks do
    bench.end(readFromOutput * 8 / (1024 ** 3));

    // Cut off writing the easy way.
    input.write = () => {};
  }, duration * 1000);
}