45 lines
1.3 KiB
TypeScript
45 lines
1.3 KiB
TypeScript
import { tracer } from '../../tracer'
|
|
|
|
const pluginName = 'ProfilingPlugin'
|
|
|
|
export const spans = new WeakMap()
|
|
|
|
export class ProfilingPlugin {
|
|
apply(compiler: any) {
|
|
// Only enabled when instrumentation is loaded
|
|
if (!tracer.getCurrentSpan()) {
|
|
return
|
|
}
|
|
|
|
compiler.hooks.compile.tap(pluginName, () => {
|
|
const span = tracer.startSpan('webpack-compile', {
|
|
attributes: { name: compiler.name },
|
|
})
|
|
spans.set(compiler, span)
|
|
})
|
|
compiler.hooks.done.tap(pluginName, () => {
|
|
spans.get(compiler).end()
|
|
})
|
|
compiler.hooks.compilation.tap(pluginName, (compilation: any) => {
|
|
compilation.hooks.buildModule.tap(pluginName, (module: any) => {
|
|
tracer.withSpan(spans.get(compiler), () => {
|
|
const span = tracer.startSpan('build-module')
|
|
span.setAttribute('name', module.userRequest)
|
|
spans.set(module, span)
|
|
})
|
|
})
|
|
|
|
compilation.hooks.normalModuleLoader.tap(
|
|
pluginName,
|
|
(loaderContext: any, module: any) => {
|
|
const parentSpan = spans.get(module)
|
|
loaderContext.currentTraceSpan = parentSpan
|
|
}
|
|
)
|
|
|
|
compilation.hooks.succeedModule.tap(pluginName, (module: any) => {
|
|
spans.get(module).end()
|
|
})
|
|
})
|
|
}
|
|
}
|