123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204 |
- "use strict";
- const { STAGE_BASIC } = require("../OptimizationStages");
- function intersectMasks(masks) {
- let result = masks[0];
- for (let i = masks.length - 1; i >= 1; i--) {
- result &= masks[i];
- }
- return result;
- }
- const ZERO_BIGINT = BigInt(0);
- const ONE_BIGINT = BigInt(1);
- const THIRTY_TWO_BIGINT = BigInt(32);
- function* getModulesFromMask(mask, ordinalModules) {
- let offset = 31;
- while (mask !== ZERO_BIGINT) {
-
- let last32 = Number(BigInt.asUintN(32, mask));
- while (last32 > 0) {
- let last = Math.clz32(last32);
-
-
- const moduleIndex = offset - last;
-
- const module = ordinalModules[moduleIndex];
- yield module;
-
-
- last32 &= ~(1 << (31 - last));
- }
-
- mask >>= THIRTY_TWO_BIGINT;
- offset += 32;
- }
- }
- class RemoveParentModulesPlugin {
-
- apply(compiler) {
- compiler.hooks.compilation.tap("RemoveParentModulesPlugin", compilation => {
-
- const handler = (chunks, chunkGroups) => {
- const chunkGraph = compilation.chunkGraph;
- const queue = new Set();
- const availableModulesMap = new WeakMap();
- let nextModuleMask = ONE_BIGINT;
- const maskByModule = new WeakMap();
-
- const ordinalModules = [];
-
- const getOrCreateModuleMask = mod => {
- let id = maskByModule.get(mod);
- if (id === undefined) {
- id = nextModuleMask;
- ordinalModules.push(mod);
- maskByModule.set(mod, id);
- nextModuleMask <<= ONE_BIGINT;
- }
- return id;
- };
-
- const chunkMasks = new WeakMap();
- for (const chunk of chunks) {
- let mask = ZERO_BIGINT;
- for (const m of chunkGraph.getChunkModulesIterable(chunk)) {
- const id = getOrCreateModuleMask(m);
- mask |= id;
- }
- chunkMasks.set(chunk, mask);
- }
- const chunkGroupMasks = new WeakMap();
- for (const chunkGroup of chunkGroups) {
- let mask = ZERO_BIGINT;
- for (const chunk of chunkGroup.chunks) {
- const chunkMask = chunkMasks.get(chunk);
- if (chunkMask !== undefined) {
- mask |= chunkMask;
- }
- }
- chunkGroupMasks.set(chunkGroup, mask);
- }
- for (const chunkGroup of compilation.entrypoints.values()) {
-
- availableModulesMap.set(chunkGroup, ZERO_BIGINT);
- for (const child of chunkGroup.childrenIterable) {
- queue.add(child);
- }
- }
- for (const chunkGroup of compilation.asyncEntrypoints) {
-
- availableModulesMap.set(chunkGroup, ZERO_BIGINT);
- for (const child of chunkGroup.childrenIterable) {
- queue.add(child);
- }
- }
- for (const chunkGroup of queue) {
- let availableModulesMask = availableModulesMap.get(chunkGroup);
- let changed = false;
- for (const parent of chunkGroup.parentsIterable) {
- const availableModulesInParent = availableModulesMap.get(parent);
- if (availableModulesInParent !== undefined) {
- const parentMask =
- availableModulesInParent | chunkGroupMasks.get(parent);
-
- if (availableModulesMask === undefined) {
-
- availableModulesMask = parentMask;
- changed = true;
- } else {
- let newMask = availableModulesMask & parentMask;
- if (newMask !== availableModulesMask) {
- changed = true;
- availableModulesMask = newMask;
- }
- }
- }
- }
- if (changed) {
- availableModulesMap.set(chunkGroup, availableModulesMask);
-
- for (const child of chunkGroup.childrenIterable) {
-
- queue.delete(child);
- queue.add(child);
- }
- }
- }
-
- for (const chunk of chunks) {
- const chunkMask = chunkMasks.get(chunk);
- if (chunkMask === undefined) continue;
- const availableModulesSets = Array.from(
- chunk.groupsIterable,
- chunkGroup => availableModulesMap.get(chunkGroup)
- );
- if (availableModulesSets.some(s => s === undefined)) continue;
- const availableModulesMask = intersectMasks(availableModulesSets);
- const toRemoveMask = chunkMask & availableModulesMask;
- if (toRemoveMask !== ZERO_BIGINT) {
- for (const module of getModulesFromMask(
- toRemoveMask,
- ordinalModules
- )) {
- chunkGraph.disconnectChunkAndModule(chunk, module);
- }
- }
- }
- };
- compilation.hooks.optimizeChunks.tap(
- {
- name: "RemoveParentModulesPlugin",
- stage: STAGE_BASIC
- },
- handler
- );
- });
- }
- }
- module.exports = RemoveParentModulesPlugin;
|