init

马宇豪

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
/*
    MIT License http://www.opensource.org/licenses/mit-license.php
    Author Tobias Koppers @sokra
*/
 
"use strict";
 
const Cache = require("../Cache");
 
/** @typedef {import("webpack-sources").Source} Source */
/** @typedef {import("../Cache").Etag} Etag */
/** @typedef {import("../Compiler")} Compiler */
/** @typedef {import("../Module")} Module */
 
class MemoryCachePlugin {
    /**
     * Apply the plugin
     * @param {Compiler} compiler the compiler instance
     * @returns {void}
     */
    apply(compiler) {
        /** @type {Map<string, { etag: Etag | null, data: any }>} */
        const cache = new Map();
        compiler.cache.hooks.store.tap(
            { name: "MemoryCachePlugin", stage: Cache.STAGE_MEMORY },
            (identifier, etag, data) => {
                cache.set(identifier, { etag, data });
            }
        );
        compiler.cache.hooks.get.tap(
            { name: "MemoryCachePlugin", stage: Cache.STAGE_MEMORY },
            (identifier, etag, gotHandlers) => {
                const cacheEntry = cache.get(identifier);
                if (cacheEntry === null) {
                    return null;
                } else if (cacheEntry !== undefined) {
                    return cacheEntry.etag === etag ? cacheEntry.data : null;
                }
                gotHandlers.push((result, callback) => {
                    if (result === undefined) {
                        cache.set(identifier, null);
                    } else {
                        cache.set(identifier, { etag, data: result });
                    }
                    return callback();
                });
            }
        );
        compiler.cache.hooks.shutdown.tap(
            { name: "MemoryCachePlugin", stage: Cache.STAGE_MEMORY },
            () => {
                cache.clear();
            }
        );
    }
}
module.exports = MemoryCachePlugin;