初始化

康凯

2022-05-20 aa4c235a8ca67ea8b731f90c951a465e92c0a865

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
'use strict';
 
var isNative = /\.node$/;
 
function forEach(obj, callback) {
    for ( var key in obj ) {
        if (!Object.prototype.hasOwnProperty.call(obj, key)) {
            continue;
        }
        callback(key);
    }
}
 
function assign(target, source) {
    forEach(source, function (key) {
        target[key] = source[key];
    });
    return target;
}
 
function clearCache(requireCache) {
    forEach(requireCache, function (resolvedPath) {
        if (!isNative.test(resolvedPath)) {
            delete requireCache[resolvedPath];
        }
    });
}
 
module.exports = function (requireCache, callback, callbackForModulesToKeep, module) {
 
    var originalCache = assign({}, requireCache);
    clearCache(requireCache);
 
    if (callbackForModulesToKeep) {
 
        var originalModuleChildren = module.children ? module.children.slice() : false; // Creates a shallow copy of module.children
 
        callbackForModulesToKeep();
 
        // Lists the cache entries made by callbackForModulesToKeep()
        var modulesToKeep = [];
        forEach(requireCache, function (key) {
            modulesToKeep.push(key);
        });
 
        // Discards the modules required in callbackForModulesToKeep()
        clearCache(requireCache);
 
        if (module.children) { // Only true for node.js
            module.children = originalModuleChildren; // Removes last references to modules required in callbackForModulesToKeep() -> No memory leak
        }
 
        // Takes the cache entries of the original cache in case the modules where required before
        for ( var i = 0; i < modulesToKeep.length; i+=1 ) {
            if (originalCache[modulesToKeep[i]]) {
                requireCache[modulesToKeep[i]] = originalCache[modulesToKeep[i]];
            }
        }
 
    }
 
    var freshModule = callback();
 
    var stealthCache = callbackForModulesToKeep ? assign({}, requireCache) : false;
 
    clearCache(requireCache);
 
    if (callbackForModulesToKeep) {
        // In case modules to keep were required inside the stealthy require for the first time, copy them to the restored cache
        for ( var k = 0; k < modulesToKeep.length; k+=1 ) {
            if (stealthCache[modulesToKeep[k]]) {
                requireCache[modulesToKeep[k]] = stealthCache[modulesToKeep[k]];
            }
        }
    }
 
    assign(requireCache, originalCache);
 
    return freshModule;
 
};