securityos/node_modules/next/dist/build/webpack/plugins/profiling-plugin.js

273 lines
13 KiB
JavaScript

"use strict";
Object.defineProperty(exports, "__esModule", {
value: true
});
0 && (module.exports = {
spans: null,
webpackInvalidSpans: null,
ProfilingPlugin: null
});
function _export(target, all) {
for(var name in all)Object.defineProperty(target, name, {
enumerable: true,
get: all[name]
});
}
_export(exports, {
spans: function() {
return spans;
},
webpackInvalidSpans: function() {
return webpackInvalidSpans;
},
ProfilingPlugin: function() {
return ProfilingPlugin;
}
});
const _webpack = require("next/dist/compiled/webpack/webpack");
const pluginName = "ProfilingPlugin";
const spans = new WeakMap();
const moduleSpansByCompilation = new WeakMap();
const makeSpanByCompilation = new WeakMap();
const sealSpanByCompilation = new WeakMap();
const webpackInvalidSpans = new WeakMap();
const TRACE_LABELS_SEAL = [
"module assets",
"create chunk assets",
"asset render",
"asset emit",
"store asset"
];
function inTraceLabelsSeal(label) {
return TRACE_LABELS_SEAL.some((l)=>label.startsWith(l));
}
class ProfilingPlugin {
constructor({ runWebpackSpan }){
this.runWebpackSpan = runWebpackSpan;
}
apply(compiler) {
this.traceTopLevelHooks(compiler);
this.traceCompilationHooks(compiler);
this.compiler = compiler;
}
traceHookPair(spanName, startHook, stopHook, { parentSpan, attrs, onStart, onStop } = {}) {
let span;
startHook.tap({
name: pluginName,
stage: -Infinity
}, (...params)=>{
const name = typeof spanName === "function" ? spanName() : spanName;
const attributes = attrs ? attrs(...params) : attrs;
span = parentSpan ? parentSpan(...params).traceChild(name, attributes) : this.runWebpackSpan.traceChild(name, attributes);
if (onStart) onStart(span, ...params);
});
stopHook.tap({
name: pluginName,
stage: Infinity
}, (...params)=>{
// `stopHook` may be triggered when `startHook` has not in cases
// where `stopHook` is used as the terminating event for more
// than one pair of hooks.
if (!span) {
return;
}
if (onStop) onStop(span, ...params);
span.stop();
});
}
traceTopLevelHooks(compiler) {
this.traceHookPair("webpack-compilation", compiler.hooks.compilation, compiler.hooks.afterCompile, {
parentSpan: ()=>webpackInvalidSpans.get(compiler) || this.runWebpackSpan,
attrs: ()=>({
name: compiler.name
}),
onStart: (span, compilation)=>{
spans.set(compilation, span);
spans.set(compiler, span);
moduleSpansByCompilation.set(compilation, new WeakMap());
}
});
if (compiler.options.mode === "development") {
this.traceHookPair(()=>`webpack-invalidated-${compiler.name}`, compiler.hooks.invalid, compiler.hooks.done, {
onStart: (span)=>webpackInvalidSpans.set(compiler, span),
onStop: ()=>webpackInvalidSpans.delete(compiler),
attrs: (fileName)=>({
trigger: fileName || "manual"
})
});
}
}
traceCompilationHooks(compiler) {
this.traceHookPair("emit", compiler.hooks.emit, compiler.hooks.afterEmit, {
parentSpan: ()=>webpackInvalidSpans.get(compiler) || this.runWebpackSpan
});
this.traceHookPair("make", compiler.hooks.make, compiler.hooks.finishMake, {
parentSpan: (compilation)=>{
const compilationSpan = spans.get(compilation);
if (!compilationSpan) {
return webpackInvalidSpans.get(compiler) || this.runWebpackSpan;
}
return compilationSpan;
},
onStart: (span, compilation)=>{
makeSpanByCompilation.set(compilation, span);
},
onStop: (_span, compilation)=>{
makeSpanByCompilation.delete(compilation);
}
});
compiler.hooks.compilation.tap({
name: pluginName,
stage: -Infinity
}, (compilation)=>{
compilation.hooks.buildModule.tap(pluginName, (module1)=>{
var _compilation_moduleGraph;
const moduleType = (()=>{
if (!module1.userRequest) {
return "";
}
return module1.userRequest.split(".").pop();
})();
const issuerModule = compilation == null ? void 0 : (_compilation_moduleGraph = compilation.moduleGraph) == null ? void 0 : _compilation_moduleGraph.getIssuer(module1);
let span;
const moduleSpans = moduleSpansByCompilation.get(compilation);
const spanName = `build-module${moduleType ? `-${moduleType}` : ""}`;
const issuerSpan = issuerModule && (moduleSpans == null ? void 0 : moduleSpans.get(issuerModule));
if (issuerSpan) {
span = issuerSpan.traceChild(spanName);
} else {
let parentSpan;
for (const incomingConnection of compilation.moduleGraph.getIncomingConnections(module1)){
const entrySpan = spans.get(incomingConnection.dependency);
if (entrySpan) {
parentSpan = entrySpan;
break;
}
}
if (!parentSpan) {
const compilationSpan = spans.get(compilation);
if (!compilationSpan) {
return;
}
parentSpan = compilationSpan;
}
span = parentSpan.traceChild(spanName);
}
span.setAttribute("name", module1.userRequest);
span.setAttribute("layer", module1.layer);
moduleSpans.set(module1, span);
});
const moduleHooks = _webpack.NormalModule.getCompilationHooks(compilation);
moduleHooks.readResource.for(undefined).intercept({
register (tapInfo) {
const fn = tapInfo.fn;
tapInfo.fn = (loaderContext, callback)=>{
const moduleSpan = loaderContext.currentTraceSpan.traceChild(`read-resource`);
fn(loaderContext, (err, result)=>{
moduleSpan.stop();
callback(err, result);
});
};
return tapInfo;
}
});
moduleHooks.loader.tap(pluginName, (loaderContext, module1)=>{
var _moduleSpansByCompilation_get;
const moduleSpan = (_moduleSpansByCompilation_get = moduleSpansByCompilation.get(compilation)) == null ? void 0 : _moduleSpansByCompilation_get.get(module1);
loaderContext.currentTraceSpan = moduleSpan;
});
compilation.hooks.succeedModule.tap(pluginName, (module1)=>{
var _moduleSpansByCompilation_get_get, _moduleSpansByCompilation_get;
moduleSpansByCompilation == null ? void 0 : (_moduleSpansByCompilation_get = moduleSpansByCompilation.get(compilation)) == null ? void 0 : (_moduleSpansByCompilation_get_get = _moduleSpansByCompilation_get.get(module1)) == null ? void 0 : _moduleSpansByCompilation_get_get.stop();
});
compilation.hooks.failedModule.tap(pluginName, (module1)=>{
var _moduleSpansByCompilation_get_get, _moduleSpansByCompilation_get;
moduleSpansByCompilation == null ? void 0 : (_moduleSpansByCompilation_get = moduleSpansByCompilation.get(compilation)) == null ? void 0 : (_moduleSpansByCompilation_get_get = _moduleSpansByCompilation_get.get(module1)) == null ? void 0 : _moduleSpansByCompilation_get_get.stop();
});
this.traceHookPair("seal", compilation.hooks.seal, compilation.hooks.afterSeal, {
parentSpan: ()=>spans.get(compilation),
onStart (span) {
sealSpanByCompilation.set(compilation, span);
},
onStop () {
sealSpanByCompilation.delete(compilation);
}
});
compilation.hooks.addEntry.tap(pluginName, (entry)=>{
const parentSpan = makeSpanByCompilation.get(compilation) || spans.get(compilation);
if (!parentSpan) {
return;
}
const addEntrySpan = parentSpan.traceChild("add-entry");
addEntrySpan.setAttribute("request", entry.request);
spans.set(entry, addEntrySpan);
});
compilation.hooks.succeedEntry.tap(pluginName, (entry)=>{
var _spans_get;
(_spans_get = spans.get(entry)) == null ? void 0 : _spans_get.stop();
spans.delete(entry);
});
compilation.hooks.failedEntry.tap(pluginName, (entry)=>{
var _spans_get;
(_spans_get = spans.get(entry)) == null ? void 0 : _spans_get.stop();
spans.delete(entry);
});
this.traceHookPair("chunk-graph", compilation.hooks.beforeChunks, compilation.hooks.afterChunks, {
parentSpan: ()=>sealSpanByCompilation.get(compilation) || spans.get(compilation)
});
this.traceHookPair("optimize", compilation.hooks.optimize, compilation.hooks.reviveModules, {
parentSpan: ()=>sealSpanByCompilation.get(compilation) || spans.get(compilation)
});
this.traceHookPair("optimize-modules", compilation.hooks.optimizeModules, compilation.hooks.afterOptimizeModules, {
parentSpan: ()=>sealSpanByCompilation.get(compilation) || spans.get(compilation)
});
this.traceHookPair("optimize-chunks", compilation.hooks.optimizeChunks, compilation.hooks.afterOptimizeChunks, {
parentSpan: ()=>sealSpanByCompilation.get(compilation) || spans.get(compilation)
});
this.traceHookPair("optimize-tree", compilation.hooks.optimizeTree, compilation.hooks.afterOptimizeTree, {
parentSpan: ()=>sealSpanByCompilation.get(compilation) || spans.get(compilation)
});
this.traceHookPair("optimize-chunk-modules", compilation.hooks.optimizeChunkModules, compilation.hooks.afterOptimizeChunkModules, {
parentSpan: ()=>sealSpanByCompilation.get(compilation) || spans.get(compilation)
});
this.traceHookPair("module-hash", compilation.hooks.beforeModuleHash, compilation.hooks.afterModuleHash, {
parentSpan: ()=>sealSpanByCompilation.get(compilation) || spans.get(compilation)
});
this.traceHookPair("code-generation", compilation.hooks.beforeCodeGeneration, compilation.hooks.afterCodeGeneration, {
parentSpan: ()=>sealSpanByCompilation.get(compilation) || spans.get(compilation)
});
this.traceHookPair("hash", compilation.hooks.beforeHash, compilation.hooks.afterHash, {
parentSpan: ()=>sealSpanByCompilation.get(compilation) || spans.get(compilation)
});
this.traceHookPair("code-generation-jobs", compilation.hooks.afterHash, compilation.hooks.beforeModuleAssets, {
parentSpan: ()=>sealSpanByCompilation.get(compilation) || spans.get(compilation)
});
const logs = new Map();
const originalTime = compilation.logger.time;
const originalTimeEnd = compilation.logger.timeEnd;
compilation.logger.time = (label)=>{
if (!inTraceLabelsSeal(label)) {
return originalTime.call(compilation.logger, label);
}
const span = sealSpanByCompilation.get(compilation);
if (span) {
logs.set(label, span.traceChild(label.replace(/ /g, "-")));
}
return originalTime.call(compilation.logger, label);
};
compilation.logger.timeEnd = (label)=>{
if (!inTraceLabelsSeal(label)) {
return originalTimeEnd.call(compilation.logger, label);
}
const span = logs.get(label);
if (span) {
span.stop();
logs.delete(label);
}
return originalTimeEnd.call(compilation.logger, label);
};
});
}
}
//# sourceMappingURL=profiling-plugin.js.map