node/benchmark/fs/promises-copyfile.js at df1d5878c83673c73aa4b8b819e0e6d88db18c0f · jasnell/node · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
'use strict';

const common = require('../common');

const {
  async,
  unlinkSync,
  openSync,
  closeSync,
  copyFile
} = require('fs');
const path = require('path');
const { promisify } = require('util');
const Countdown = require('../../test/common/countdown');
const src = path.resolve(__dirname, '.removeme-benchmark-garbage');
const dest = path.resolve(__dirname, '.removeme-benchmark-garbage2');

const bench = common.createBenchmark(main, {
  n: [20000],
  method: ['legacy', 'promisify', 'promise']
});

function useLegacy(n) {
  const countdown = new Countdown(n, () => {
    bench.end(n);
    unlinkSync(src);
    unlinkSync(dest);
  });
  bench.start();
  for (let i = 0; i < n; i++) {
    copyFile(src, dest, (err) => countdown.dec());
  }
}

function usePromisify(n) {
  const _copyFile = promisify(copyFile);
  const countdown = new Countdown(n, () => {
    bench.end(n);
    unlinkSync(src);
    unlinkSync(dest);
  });
  bench.start();
  for (let i = 0; i < n; i++) {
    _copyFile(src, dest)
      .then(() => countdown.dec());
  }
}

function usePromise(n) {
  const _copyFile = async.copyFile;
  const countdown = new Countdown(n, () => {
    bench.end(n);
    unlinkSync(src);
    unlinkSync(dest);
  });
  bench.start();
  for (let i = 0; i < n; i++) {
    _copyFile(src, dest)
      .then(() => countdown.dec());
  }
}

function main(conf) {
  const n = conf.n >>> 0;
  const method = conf.method;

  closeSync(openSync(src, 'w'));

  switch (method) {
    case 'legacy': return useLegacy(n);
    case 'promisify': return usePromisify(n);
    case 'promise': return usePromise(n);
    default:
      throw new Error('invalid method');
  }
}