TaskRunner.js 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108
  1. 'use strict';
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. var _os = require('os');
  6. var _os2 = _interopRequireDefault(_os);
  7. var _cacache = require('cacache');
  8. var _cacache2 = _interopRequireDefault(_cacache);
  9. var _findCacheDir = require('find-cache-dir');
  10. var _findCacheDir2 = _interopRequireDefault(_findCacheDir);
  11. var _workerFarm = require('worker-farm');
  12. var _workerFarm2 = _interopRequireDefault(_workerFarm);
  13. var _serializeJavascript = require('serialize-javascript');
  14. var _serializeJavascript2 = _interopRequireDefault(_serializeJavascript);
  15. var _minify = require('./minify');
  16. var _minify2 = _interopRequireDefault(_minify);
  17. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  18. const worker = require.resolve('./worker');
  19. class TaskRunner {
  20. constructor(options = {}) {
  21. const { cache, parallel } = options;
  22. this.cacheDir = cache === true ? (0, _findCacheDir2.default)({ name: 'terser-webpack-plugin' }) : cache;
  23. // In some cases cpus() returns undefined
  24. // https://github.com/nodejs/node/issues/19022
  25. const cpus = _os2.default.cpus() || { length: 1 };
  26. this.maxConcurrentWorkers = parallel === true ? cpus.length - 1 : Math.min(Number(parallel) || 0, cpus.length - 1);
  27. }
  28. run(tasks, callback) {
  29. /* istanbul ignore if */
  30. if (!tasks.length) {
  31. callback(null, []);
  32. return;
  33. }
  34. if (this.maxConcurrentWorkers > 1) {
  35. const workerOptions = process.platform === 'win32' ? {
  36. maxConcurrentWorkers: this.maxConcurrentWorkers,
  37. maxConcurrentCallsPerWorker: 1
  38. } : { maxConcurrentWorkers: this.maxConcurrentWorkers };
  39. this.workers = (0, _workerFarm2.default)(workerOptions, worker);
  40. this.boundWorkers = (options, cb) => this.workers((0, _serializeJavascript2.default)(options), cb);
  41. } else {
  42. this.boundWorkers = (options, cb) => {
  43. try {
  44. cb(null, (0, _minify2.default)(options));
  45. } catch (error) {
  46. cb(error);
  47. }
  48. };
  49. }
  50. let toRun = tasks.length;
  51. const results = [];
  52. const step = (index, data) => {
  53. toRun -= 1;
  54. results[index] = data;
  55. if (!toRun) {
  56. callback(null, results);
  57. }
  58. };
  59. tasks.forEach((task, index) => {
  60. const enqueue = () => {
  61. this.boundWorkers(task, (error, data) => {
  62. const result = error ? { error } : data;
  63. const done = () => step(index, result);
  64. if (this.cacheDir && !result.error) {
  65. _cacache2.default.put(this.cacheDir, (0, _serializeJavascript2.default)(task.cacheKeys), JSON.stringify(data)).then(done, done);
  66. } else {
  67. done();
  68. }
  69. });
  70. };
  71. if (this.cacheDir) {
  72. _cacache2.default.get(this.cacheDir, (0, _serializeJavascript2.default)(task.cacheKeys)).then(({ data }) => step(index, JSON.parse(data)), enqueue);
  73. } else {
  74. enqueue();
  75. }
  76. });
  77. }
  78. exit() {
  79. if (this.workers) {
  80. _workerFarm2.default.end(this.workers);
  81. }
  82. }
  83. }
  84. exports.default = TaskRunner;