TaskRunner.js 3.1 KB

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