116 lines
3.5 KiB
JavaScript
116 lines
3.5 KiB
JavaScript
/*
|
|
MIT License http://www.opensource.org/licenses/mit-license.php
|
|
Author Tobias Koppers @sokra
|
|
*/
|
|
|
|
"use strict";
|
|
|
|
const { STAGE_ADVANCED } = require("../OptimizationStages");
|
|
const { runtimeEqual } = require("../util/runtime");
|
|
|
|
/** @typedef {import("../Compiler")} Compiler */
|
|
|
|
class MergeDuplicateChunksPlugin {
|
|
/**
|
|
* @param {Compiler} compiler the compiler
|
|
* @returns {void}
|
|
*/
|
|
apply(compiler) {
|
|
compiler.hooks.compilation.tap(
|
|
"MergeDuplicateChunksPlugin",
|
|
compilation => {
|
|
compilation.hooks.optimizeChunks.tap(
|
|
{
|
|
name: "MergeDuplicateChunksPlugin",
|
|
stage: STAGE_ADVANCED
|
|
},
|
|
chunks => {
|
|
const { chunkGraph, moduleGraph } = compilation;
|
|
|
|
// remember already tested chunks for performance
|
|
const notDuplicates = new Set();
|
|
|
|
// for each chunk
|
|
for (const chunk of chunks) {
|
|
// track a Set of all chunk that could be duplicates
|
|
let possibleDuplicates;
|
|
for (const module of chunkGraph.getChunkModulesIterable(chunk)) {
|
|
if (possibleDuplicates === undefined) {
|
|
// when possibleDuplicates is not yet set,
|
|
// create a new Set from chunks of the current module
|
|
// including only chunks with the same number of modules
|
|
for (const dup of chunkGraph.getModuleChunksIterable(
|
|
module
|
|
)) {
|
|
if (
|
|
dup !== chunk &&
|
|
chunkGraph.getNumberOfChunkModules(chunk) ===
|
|
chunkGraph.getNumberOfChunkModules(dup) &&
|
|
!notDuplicates.has(dup)
|
|
) {
|
|
// delay allocating the new Set until here, reduce memory pressure
|
|
if (possibleDuplicates === undefined) {
|
|
possibleDuplicates = new Set();
|
|
}
|
|
possibleDuplicates.add(dup);
|
|
}
|
|
}
|
|
// when no chunk is possible we can break here
|
|
if (possibleDuplicates === undefined) break;
|
|
} else {
|
|
// validate existing possible duplicates
|
|
for (const dup of possibleDuplicates) {
|
|
// remove possible duplicate when module is not contained
|
|
if (!chunkGraph.isModuleInChunk(module, dup)) {
|
|
possibleDuplicates.delete(dup);
|
|
}
|
|
}
|
|
// when all chunks has been removed we can break here
|
|
if (possibleDuplicates.size === 0) break;
|
|
}
|
|
}
|
|
|
|
// when we found duplicates
|
|
if (
|
|
possibleDuplicates !== undefined &&
|
|
possibleDuplicates.size > 0
|
|
) {
|
|
outer: for (const otherChunk of possibleDuplicates) {
|
|
if (otherChunk.hasRuntime() !== chunk.hasRuntime()) continue;
|
|
if (chunkGraph.getNumberOfEntryModules(chunk) > 0) continue;
|
|
if (chunkGraph.getNumberOfEntryModules(otherChunk) > 0)
|
|
continue;
|
|
if (!runtimeEqual(chunk.runtime, otherChunk.runtime)) {
|
|
for (const module of chunkGraph.getChunkModulesIterable(
|
|
chunk
|
|
)) {
|
|
const exportsInfo = moduleGraph.getExportsInfo(module);
|
|
if (
|
|
!exportsInfo.isEquallyUsed(
|
|
chunk.runtime,
|
|
otherChunk.runtime
|
|
)
|
|
) {
|
|
continue outer;
|
|
}
|
|
}
|
|
}
|
|
// merge them
|
|
if (chunkGraph.canChunksBeIntegrated(chunk, otherChunk)) {
|
|
chunkGraph.integrateChunks(chunk, otherChunk);
|
|
compilation.chunks.delete(otherChunk);
|
|
}
|
|
}
|
|
}
|
|
|
|
// don't check already processed chunks twice
|
|
notDuplicates.add(chunk);
|
|
}
|
|
}
|
|
);
|
|
}
|
|
);
|
|
}
|
|
}
|
|
module.exports = MergeDuplicateChunksPlugin;
|