Runner.js 4.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136
  1. 'use strict';
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. var _createClass = function () { function defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } } return function (Constructor, protoProps, staticProps) { if (protoProps) defineProperties(Constructor.prototype, protoProps); if (staticProps) defineProperties(Constructor, staticProps); return Constructor; }; }();
  6. var _os = require('os');
  7. var _os2 = _interopRequireDefault(_os);
  8. var _cacache = require('cacache');
  9. var _cacache2 = _interopRequireDefault(_cacache);
  10. var _findCacheDir = require('find-cache-dir');
  11. var _findCacheDir2 = _interopRequireDefault(_findCacheDir);
  12. var _workerFarm = require('worker-farm');
  13. var _workerFarm2 = _interopRequireDefault(_workerFarm);
  14. var _serializeJavascript = require('serialize-javascript');
  15. var _serializeJavascript2 = _interopRequireDefault(_serializeJavascript);
  16. var _minify = require('./minify');
  17. var _minify2 = _interopRequireDefault(_minify);
  18. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  19. function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
  20. var workerFile = require.resolve('./worker');
  21. try {
  22. // run test
  23. workerFile = require.resolve('../../dist/uglify/worker');
  24. } catch (e) {} // eslint-disable-line no-empty
  25. var Runner = function () {
  26. function Runner() {
  27. var options = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
  28. _classCallCheck(this, Runner);
  29. var cache = options.cache,
  30. parallel = options.parallel;
  31. this.cacheDir = cache === true ? (0, _findCacheDir2.default)({ name: 'uglifyjs-webpack-plugin' }) : cache;
  32. // In some cases cpus() returns undefined
  33. // https://github.com/nodejs/node/issues/19022
  34. var cpus = _os2.default.cpus() || { length: 1 };
  35. this.maxConcurrentWorkers = parallel === true ? cpus.length - 1 : Math.min(Number(parallel) || 0, cpus.length - 1);
  36. }
  37. _createClass(Runner, [{
  38. key: 'runTasks',
  39. value: function runTasks(tasks, callback) {
  40. var _this = this;
  41. /* istanbul ignore if */
  42. if (!tasks.length) {
  43. callback(null, []);
  44. return;
  45. }
  46. if (this.maxConcurrentWorkers > 1) {
  47. var workerOptions = process.platform === 'win32' ? { maxConcurrentWorkers: this.maxConcurrentWorkers, maxConcurrentCallsPerWorker: 1 } : { maxConcurrentWorkers: this.maxConcurrentWorkers };
  48. this.workers = (0, _workerFarm2.default)(workerOptions, workerFile);
  49. this.boundWorkers = function (options, cb) {
  50. return _this.workers((0, _serializeJavascript2.default)(options), cb);
  51. };
  52. } else {
  53. this.boundWorkers = function (options, cb) {
  54. try {
  55. cb(null, (0, _minify2.default)(options));
  56. } catch (error) {
  57. cb(error);
  58. }
  59. };
  60. }
  61. var toRun = tasks.length;
  62. var results = [];
  63. var step = function step(index, data) {
  64. toRun -= 1;
  65. results[index] = data;
  66. if (!toRun) {
  67. callback(null, results);
  68. }
  69. };
  70. tasks.forEach(function (task, index) {
  71. var enqueue = function enqueue() {
  72. _this.boundWorkers(task, function (error, data) {
  73. var result = error ? { error } : data;
  74. var done = function done() {
  75. return step(index, result);
  76. };
  77. if (_this.cacheDir && !result.error) {
  78. _cacache2.default.put(_this.cacheDir, (0, _serializeJavascript2.default)(task.cacheKeys), JSON.stringify(data)).then(done, done);
  79. } else {
  80. done();
  81. }
  82. });
  83. };
  84. if (_this.cacheDir) {
  85. _cacache2.default.get(_this.cacheDir, (0, _serializeJavascript2.default)(task.cacheKeys)).then(function (_ref) {
  86. var data = _ref.data;
  87. return step(index, JSON.parse(data));
  88. }, enqueue);
  89. } else {
  90. enqueue();
  91. }
  92. });
  93. }
  94. }, {
  95. key: 'exit',
  96. value: function exit() {
  97. if (this.workers) {
  98. _workerFarm2.default.end(this.workers);
  99. }
  100. }
  101. }]);
  102. return Runner;
  103. }();
  104. exports.default = Runner;