马宇豪
2024-07-16 f591c27b57e2418c9495bc02ae8cfff84d35bc18
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
const { STAGE_ADVANCED } = require("../OptimizationStages");
const createSchemaValidation = require("../util/create-schema-validation");
 
/** @typedef {import("../../declarations/plugins/optimize/MinChunkSizePlugin").MinChunkSizePluginOptions} MinChunkSizePluginOptions */
/** @typedef {import("../Chunk")} Chunk */
/** @typedef {import("../Compiler")} Compiler */
 
const validate = createSchemaValidation(
    require("../../schemas/plugins/optimize/MinChunkSizePlugin.check.js"),
    () => require("../../schemas/plugins/optimize/MinChunkSizePlugin.json"),
    {
        name: "Min Chunk Size Plugin",
        baseDataPath: "options"
    }
);
 
class MinChunkSizePlugin {
    /**
     * @param {MinChunkSizePluginOptions} options options object
     */
    constructor(options) {
        validate(options);
        this.options = options;
    }
 
    /**
     * Apply the plugin
     * @param {Compiler} compiler the compiler instance
     * @returns {void}
     */
    apply(compiler) {
        const options = this.options;
        const minChunkSize = options.minChunkSize;
        compiler.hooks.compilation.tap("MinChunkSizePlugin", compilation => {
            compilation.hooks.optimizeChunks.tap(
                {
                    name: "MinChunkSizePlugin",
                    stage: STAGE_ADVANCED
                },
                chunks => {
                    const chunkGraph = compilation.chunkGraph;
                    const equalOptions = {
                        chunkOverhead: 1,
                        entryChunkMultiplicator: 1
                    };
 
                    const chunkSizesMap = new Map();
                    /** @type {[Chunk, Chunk][]} */
                    const combinations = [];
                    /** @type {Chunk[]} */
                    const smallChunks = [];
                    const visitedChunks = [];
                    for (const a of chunks) {
                        // check if one of the chunks sizes is smaller than the minChunkSize
                        // and filter pairs that can NOT be integrated!
                        if (chunkGraph.getChunkSize(a, equalOptions) < minChunkSize) {
                            smallChunks.push(a);
                            for (const b of visitedChunks) {
                                if (chunkGraph.canChunksBeIntegrated(b, a))
                                    combinations.push([b, a]);
                            }
                        } else {
                            for (const b of smallChunks) {
                                if (chunkGraph.canChunksBeIntegrated(b, a))
                                    combinations.push([b, a]);
                            }
                        }
                        chunkSizesMap.set(a, chunkGraph.getChunkSize(a, options));
                        visitedChunks.push(a);
                    }
 
                    const sortedSizeFilteredExtendedPairCombinations = combinations
                        .map(pair => {
                            // extend combination pairs with size and integrated size
                            const a = chunkSizesMap.get(pair[0]);
                            const b = chunkSizesMap.get(pair[1]);
                            const ab = chunkGraph.getIntegratedChunksSize(
                                pair[0],
                                pair[1],
                                options
                            );
                            /** @type {[number, number, Chunk, Chunk]} */
                            const extendedPair = [a + b - ab, ab, pair[0], pair[1]];
                            return extendedPair;
                        })
                        .sort((a, b) => {
                            // sadly javascript does an in place sort here
                            // sort by size
                            const diff = b[0] - a[0];
                            if (diff !== 0) return diff;
                            return a[1] - b[1];
                        });
 
                    if (sortedSizeFilteredExtendedPairCombinations.length === 0) return;
 
                    const pair = sortedSizeFilteredExtendedPairCombinations[0];
 
                    chunkGraph.integrateChunks(pair[2], pair[3]);
                    compilation.chunks.delete(pair[3]);
                    return true;
                }
            );
        });
    }
}
module.exports = MinChunkSizePlugin;