index.js 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132
  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 _class = function () {
  26. function _class() {
  27. var options = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : {};
  28. _classCallCheck(this, _class);
  29. var cache = options.cache,
  30. parallel = options.parallel;
  31. this.cacheDir = cache === true ? (0, _findCacheDir2.default)({ name: 'uglifyjs-webpack-plugin' }) : cache;
  32. this.maxConcurrentWorkers = parallel === true ? _os2.default.cpus().length - 1 : Math.min(Number(parallel) || 0, _os2.default.cpus().length - 1);
  33. }
  34. _createClass(_class, [{
  35. key: 'runTasks',
  36. value: function runTasks(tasks, callback) {
  37. var _this = this;
  38. if (!tasks.length) {
  39. callback(null, []);
  40. return;
  41. }
  42. if (this.maxConcurrentWorkers > 0) {
  43. var workerOptions = process.platform === 'win32' ? { maxConcurrentWorkers: this.maxConcurrentWorkers, maxConcurrentCallsPerWorker: 1 } : { maxConcurrentWorkers: this.maxConcurrentWorkers };
  44. this.workers = (0, _workerFarm2.default)(workerOptions, workerFile);
  45. this.boundWorkers = function (options, cb) {
  46. return _this.workers((0, _serializeJavascript2.default)(options), cb);
  47. };
  48. } else {
  49. this.boundWorkers = function (options, cb) {
  50. try {
  51. cb(null, (0, _minify2.default)(options));
  52. } catch (errors) {
  53. cb(errors);
  54. }
  55. };
  56. }
  57. var toRun = tasks.length;
  58. var results = [];
  59. var step = function step(index, data) {
  60. toRun -= 1;
  61. results[index] = data;
  62. if (!toRun) {
  63. callback(null, results);
  64. }
  65. };
  66. tasks.forEach(function (task, index) {
  67. var enqueue = function enqueue() {
  68. _this.boundWorkers(task, function (error, data) {
  69. var result = error ? { error } : data;
  70. var done = function done() {
  71. return step(index, result);
  72. };
  73. if (_this.cacheDir && !result.error) {
  74. _cacache2.default.put(_this.cacheDir, task.cacheKey, JSON.stringify(data)).then(done, done);
  75. } else {
  76. done();
  77. }
  78. });
  79. };
  80. if (_this.cacheDir) {
  81. _cacache2.default.get(_this.cacheDir, task.cacheKey).then(function (_ref) {
  82. var data = _ref.data;
  83. return step(index, JSON.parse(data));
  84. }, enqueue);
  85. } else {
  86. enqueue();
  87. }
  88. });
  89. }
  90. }, {
  91. key: 'exit',
  92. value: function exit() {
  93. if (this.workers) {
  94. _workerFarm2.default.end(this.workers);
  95. }
  96. }
  97. }]);
  98. return _class;
  99. }();
  100. exports.default = _class;