DEBUG

Author: szegedi

integration-tests/profiler/codehotspots.js

@@ -2,6 +2,7 @@
 
 const DDTrace = require('dd-trace')
 const tracer = DDTrace.init()
+const NativeWallProfiler = require('dd-trace/packages/dd-trace/src/profiling/profilers/wall')
 
 // Busy cycle duration is communicated in nanoseconds through the environment
 // variable by the test. On first execution, it'll be 10 * the sampling period
@@ -29,12 +30,18 @@ function busyLoop () {
 let counter = 0
 
 function runBusySpans () {
-  tracer.trace('x' + counter, { type: 'web', resource: `endpoint-${counter}` }, (_, done) => {
+  const id1 = `x-${counter}`
+  tracer.trace(id1, { type: 'web', resource: `endpoint-${counter}` }, (_, done) => {
+    logData(id1)
     setImmediate(() => {
+      logData(`${id1} timeout`)
       for (let i = 0; i < 3; ++i) {
         const z = i
-        tracer.trace('y' + i, (_, done2) => {
+        const id2 = `y-${counter}-${i}`
+        tracer.trace(id2, (_, done2) => {
+          logData(id2)
           const busyWork = () => {
+            logData(`${id2}-timeout`)
             busyLoop()
             done2()
             if (z === 2) {
@@ -60,4 +67,12 @@ function runBusySpans () {
   })
 }
 
+function logData (codeContext) {
+  const active = NativeWallProfiler.prototype.getActiveSpan()
+  const sampleContext = NativeWallProfiler.prototype.getSampleContext()
+  const indicator = (active.spanId === sampleContext.spanId) ? '✅' : '❌'
+  const CPEDContextCount = NativeWallProfiler.prototype.getCPEDContextCount()
+  console.log(indicator, codeContext, 'activeSpan:', active.spanId, ', sampleContext:', sampleContext.spanId, ', CPEDContextCount:', CPEDContextCount)
+}
+
 tracer.profilerStarted().then(runBusySpans)

integration-tests/profiler/profiler.spec.js

@@ -340,15 +340,16 @@ describe('profiler', () => {
         DD_TRACE_AGENT_PORT: agent.port
       }
       // With Node 23 or later, test the profiler with async context frame use.
-      let execArgv = []
+      const execArgv = []
       if (satisfies(process.versions.node, '>=23.0.0')) {
         env.DD_PROFILING_USE_ASYNC_CONTEXT_FRAME = 1
         if (!satisfies(process.versions.node, '>=24.0.0')) {
           // For Node 23, use the experimental command line flag for Node to enable
           // async context frame. Node 24 has it enabled by default.
-          execArgv = ['--experimental-async-context-frame']
+          execArgv.push('--experimental-async-context-frame')
         }
       }
+      console.log({path: path.join(cwd, 'profiler/codehotspots.js'), env, execArgv })
       const proc = fork(path.join(cwd, 'profiler/codehotspots.js'), { cwd, env, execArgv })
 
       await processExitPromise(proc, timeout)

packages/dd-trace/src/profiling/profilers/wall.js

@@ -17,13 +17,14 @@ const {
 const { isWebServerSpan, endpointNameFromTags, getStartedSpans } = require('../webspan-utils')
 
 let beforeCh
+let lastInstance
 const enterCh = dc.channel('dd-trace:storage:enter')
 const spanFinishCh = dc.channel('dd-trace:span:finish')
 const profilerTelemetryMetrics = telemetryMetrics.manager.namespace('profilers')
 
 const ProfilingContext = Symbol('NativeWallProfiler.ProfilingContext')
 
-let kSampleCount
+let kSampleCount, kCPEDContextCount
 
 function getActiveSpan () {
   const store = storage('legacy').getStore()
@@ -154,6 +155,7 @@ class NativeWallProfiler {
 
     this._logger = options.logger
     this._started = false
+    lastInstance = this
   }
 
   codeHotspotsEnabled () {
@@ -170,6 +172,7 @@ class NativeWallProfiler {
     this._mapper = mapper
     this._pprof = require('@datadog/pprof')
     kSampleCount = this._pprof.time.constants.kSampleCount
+    kCPEDContextCount = this._pprof.time.constants.kCPEDContextCount
 
     // pprof otherwise crashes in worker threads
     if (!process._startProfilerIdleNotifier) {
@@ -285,6 +288,15 @@ class NativeWallProfiler {
     return profilingContext
   }
 
+  _getSampleContext () {
+    const context = this._pprof.time.getContext()
+    return this._useAsyncContextFrame ? context : context.ref
+  }
+
+  _getCPEDContextCount () {
+    return this._profilerState[kCPEDContextCount]
+  }
+
   _setNewContext () {
     this._pprof.time.setContext(
       this._currentContext = {
@@ -421,4 +433,24 @@ class NativeWallProfiler {
   }
 }
 
+NativeWallProfiler.prototype.getActiveSpan = function () {
+  const span = getActiveSpan()
+  if (span === undefined) {
+    return {}
+  }
+  const spanData = lastInstance._getProfilingContext(span)
+  updateContext(spanData)
+  return { spanId: spanData.spanId, rootSpanId: spanData.rootSpanId }
+}
+NativeWallProfiler.prototype.getSampleContext = function () {
+  const ctx = lastInstance._getSampleContext()
+  if (ctx === undefined) {
+    return {}
+  }
+  updateContext(ctx)
+  return { spanId: ctx.spanId, rootSpanId: ctx.rootSpanId }
+}
+NativeWallProfiler.prototype.getCPEDContextCount = function () {
+  return lastInstance._getCPEDContextCount()
+}
 module.exports = NativeWallProfiler