OccurrenceModuleOrderPlugin.js 2.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103
  1. /*
  2. MIT License http://www.opensource.org/licenses/mit-license.php
  3. Author Tobias Koppers @sokra
  4. */
  5. "use strict";
  6. const validateOptions = require("schema-utils");
  7. const schema = require("../../schemas/plugins/optimize/OccurrenceOrderModuleIdsPlugin.json");
  8. class OccurrenceOrderModuleIdsPlugin {
  9. constructor(options = {}) {
  10. validateOptions(schema, options, "Occurrence Order Module Ids Plugin");
  11. this.options = options;
  12. }
  13. apply(compiler) {
  14. const prioritiseInitial = this.options.prioritiseInitial;
  15. compiler.hooks.compilation.tap(
  16. "OccurrenceOrderModuleIdsPlugin",
  17. compilation => {
  18. compilation.hooks.optimizeModuleOrder.tap(
  19. "OccurrenceOrderModuleIdsPlugin",
  20. modules => {
  21. const occursInInitialChunksMap = new Map();
  22. const occursInAllChunksMap = new Map();
  23. const initialChunkChunkMap = new Map();
  24. const entryCountMap = new Map();
  25. for (const m of modules) {
  26. let initial = 0;
  27. let entry = 0;
  28. for (const c of m.chunksIterable) {
  29. if (c.canBeInitial()) initial++;
  30. if (c.entryModule === m) entry++;
  31. }
  32. initialChunkChunkMap.set(m, initial);
  33. entryCountMap.set(m, entry);
  34. }
  35. const countOccursInEntry = (sum, r) => {
  36. if (!r.module) {
  37. return sum;
  38. }
  39. return sum + initialChunkChunkMap.get(r.module);
  40. };
  41. const countOccurs = (sum, r) => {
  42. if (!r.module) {
  43. return sum;
  44. }
  45. let factor = 1;
  46. if (typeof r.dependency.getNumberOfIdOccurrences === "function") {
  47. factor = r.dependency.getNumberOfIdOccurrences();
  48. }
  49. if (factor === 0) {
  50. return sum;
  51. }
  52. return sum + factor * r.module.getNumberOfChunks();
  53. };
  54. if (prioritiseInitial) {
  55. for (const m of modules) {
  56. const result =
  57. m.reasons.reduce(countOccursInEntry, 0) +
  58. initialChunkChunkMap.get(m) +
  59. entryCountMap.get(m);
  60. occursInInitialChunksMap.set(m, result);
  61. }
  62. }
  63. const originalOrder = new Map();
  64. let i = 0;
  65. for (const m of modules) {
  66. const result =
  67. m.reasons.reduce(countOccurs, 0) +
  68. m.getNumberOfChunks() +
  69. entryCountMap.get(m);
  70. occursInAllChunksMap.set(m, result);
  71. originalOrder.set(m, i++);
  72. }
  73. modules.sort((a, b) => {
  74. if (prioritiseInitial) {
  75. const aEntryOccurs = occursInInitialChunksMap.get(a);
  76. const bEntryOccurs = occursInInitialChunksMap.get(b);
  77. if (aEntryOccurs > bEntryOccurs) return -1;
  78. if (aEntryOccurs < bEntryOccurs) return 1;
  79. }
  80. const aOccurs = occursInAllChunksMap.get(a);
  81. const bOccurs = occursInAllChunksMap.get(b);
  82. if (aOccurs > bOccurs) return -1;
  83. if (aOccurs < bOccurs) return 1;
  84. const orgA = originalOrder.get(a);
  85. const orgB = originalOrder.get(b);
  86. return orgA - orgB;
  87. });
  88. }
  89. );
  90. }
  91. );
  92. }
  93. }
  94. module.exports = OccurrenceOrderModuleIdsPlugin;