dd-trace 5.83.0 → 5.85.0

package/packages/datadog-plugin-protobufjs/src/schema_iterator.js

@@ -7,11 +7,11 @@ const {
   SCHEMA_NAME,
   SCHEMA_OPERATION,
   SCHEMA_WEIGHT,
-  SCHEMA_TYPE
+  SCHEMA_TYPE,
 } = require('../../dd-trace/src/constants')
 const log = require('../../dd-trace/src/log')
 const {
-  SchemaBuilder
+  SchemaBuilder,
 } = require('../../dd-trace/src/datastreams')
 
 class SchemaExtractor {
@@ -39,7 +39,7 @@ class SchemaExtractor {
       Enum: ['enum', null],
       Type: ['type', null],
       map: ['map', null],
-      repeated: ['array', null]
+      repeated: ['array', null],
     }
 
     return typeFormatMapping[type] || ['string', null]

package/packages/datadog-plugin-redis/src/index.js

@@ -31,8 +31,8 @@ class RedisPlugin extends CachePlugin {
         'db.name': db || '0',
         [`${this._spanType}.raw_command`]: formatCommand(normalizedCommand, args),
         'out.host': connectionOptions.host,
-        [CLIENT_PORT_KEY]: connectionOptions.port
-      }
+        [CLIENT_PORT_KEY]: connectionOptions.port,
+      },
     }, ctx)
 
     return ctx.currentStore

package/packages/datadog-plugin-restify/src/index.js

@@ -21,7 +21,7 @@ class RestifyPlugin extends RouterPlugin {
   configure (config) {
     return super.configure({
       ...config,
-      middleware: false // not supported
+      middleware: false, // not supported
     })
   }
 }

package/packages/datadog-plugin-rhea/src/consumer.js

@@ -27,8 +27,8 @@ class RheaConsumerPlugin extends ConsumerPlugin {
       meta: {
         component: 'rhea',
         'amqp.link.source.address': name,
-        'amqp.link.role': 'receiver'
-      }
+        'amqp.link.role': 'receiver',
+      },
     }, ctx)
 
     if (

package/packages/datadog-plugin-rhea/src/index.js

@@ -9,7 +9,7 @@ class RheaPlugin extends CompositePlugin {
   static get plugins () {
     return {
       producer: ProducerPlugin,
-      consumer: ConsumerPlugin
+      consumer: ConsumerPlugin,
     }
   }
 }

package/packages/datadog-plugin-rhea/src/producer.js

@@ -23,8 +23,8 @@ class RheaProducerPlugin extends ProducerPlugin {
         'amqp.link.target.address': name,
         'amqp.link.role': 'sender',
         'out.host': host,
-        [CLIENT_PORT_KEY]: port
-      }
+        [CLIENT_PORT_KEY]: port,
+      },
     }, ctx)
 
     return ctx.currentStore

package/packages/datadog-plugin-router/src/index.js

@@ -107,8 +107,8 @@ class RouterPlugin extends WebPlugin {
       integrationName: this.constructor.id,
       tags: {
         [COMPONENT]: this.constructor.id,
-        'resource.name': name || '<anonymous>'
-      }
+        'resource.name': name || '<anonymous>',
+      },
     })
 
     analyticsSampler.sample(span, this.config.measured)
@@ -137,7 +137,7 @@ class RouterPlugin extends WebPlugin {
         span,
         stack: [route],
         route,
-        middleware: []
+        middleware: [],
       }
 
       this._contexts.set(req, context)

package/packages/datadog-plugin-selenium/src/index.js

@@ -9,7 +9,7 @@ const {
   TEST_BROWSER_DRIVER_VERSION,
   TEST_BROWSER_NAME,
   TEST_BROWSER_VERSION,
-  TEST_TYPE
+  TEST_TYPE,
 } = require('../../dd-trace/src/plugins/util/test')
 const { SPAN_TYPE } = require('../../../ext/tags')
 
@@ -37,7 +37,7 @@ class SeleniumPlugin extends CiPlugin {
       seleniumVersion,
       browserName,
       browserVersion,
-      isRumActive
+      isRumActive,
     }) => {
       const store = storage('legacy').getStore()
       const span = store?.span

package/packages/datadog-plugin-sharedb/src/index.js

@@ -13,8 +13,8 @@ class SharedbPlugin extends ServerPlugin {
       resource: getReadableResourceName(actionName, request.c, request.q),
       kind: 'server',
       meta: {
-        'sharedb.action': actionName
-      }
+        'sharedb.action': actionName,
+      },
     }, ctx)
 
     if (this.config.hooks && this.config.hooks.receive) {

package/packages/datadog-plugin-tedious/src/index.js

@@ -22,8 +22,8 @@ class TediousPlugin extends DatabasePlugin {
         [CLIENT_PORT_KEY]: ctx.connectionConfig.options.port,
         'db.user': ctx.connectionConfig.userName || ctx.connectionConfig.authentication.options.userName,
         'db.name': ctx.connectionConfig.options.database,
-        'db.instance': ctx.connectionConfig.options.instanceName
-      }
+        'db.instance': ctx.connectionConfig.options.instanceName,
+      },
     }, ctx)
 
     // SQL Server includes comments when caching queries

package/packages/datadog-plugin-undici/src/index.js

@@ -71,14 +71,14 @@ class UndiciPlugin extends HttpClientPlugin {
         'span.kind': 'client',
         'http.method': method,
         'http.url': uri,
-        'out.host': hostname
+        'out.host': hostname,
       },
       metrics: {
-        [CLIENT_PORT_KEY]: port ? Number.parseInt(port, 10) : undefined
+        [CLIENT_PORT_KEY]: port ? Number.parseInt(port, 10) : undefined,
       },
       service: this.serviceName({ pluginConfig: this.config, sessionDetails: { host: hostname, port } }),
       resource: method,
-      type: 'http'
+      type: 'http',
     }, false)
 
     // Disable recording if not allowed
@@ -108,7 +108,7 @@ class UndiciPlugin extends HttpClientPlugin {
     requestContexts.set(request, {
       span,
       store,
-      uri
+      uri,
     })
 
     // Enter the span context
@@ -285,7 +285,7 @@ function normalizeConfig (config) {
   return {
     ...config,
     validateStatus,
-    hooks
+    hooks,
   }
 }
 

package/packages/datadog-plugin-vitest/src/index.js

@@ -32,13 +32,13 @@ const {
   getLibraryCapabilitiesTags,
   TEST_RETRY_REASON_TYPES,
   isModifiedTest,
-  TEST_IS_MODIFIED
+  TEST_IS_MODIFIED,
 } = require('../../dd-trace/src/plugins/util/test')
 const { COMPONENT } = require('../../dd-trace/src/constants')
 const {
   TELEMETRY_EVENT_CREATED,
   TELEMETRY_EVENT_FINISHED,
-  TELEMETRY_TEST_SESSION
+  TELEMETRY_TEST_SESSION,
 } = require('../../dd-trace/src/ci-visibility/telemetry')
 const { DD_MAJOR } = require('../../../version')
 
@@ -69,7 +69,7 @@ class VitestPlugin extends CiPlugin {
       testManagementTests,
       testSuiteAbsolutePath,
       testName,
-      onDone
+      onDone,
     }) => {
       const testSuite = getTestSuitePath(testSuiteAbsolutePath, this.repositoryRoot)
       const { isAttemptToFix } = this.getTestProperties(testManagementTests, testSuite, testName)
@@ -101,7 +101,7 @@ class VitestPlugin extends CiPlugin {
     this.addSub('ci:vitest:is-early-flake-detection-faulty', ({
       knownTests,
       testFilepaths,
-      onDone
+      onDone,
     }) => {
       const isFaulty = getIsFaultyEarlyFlakeDetection(
         testFilepaths.map(testFilepath => getTestSuitePath(testFilepath, this.repositoryRoot)),
@@ -124,14 +124,14 @@ class VitestPlugin extends CiPlugin {
         isRetryReasonEfd,
         isRetryReasonAttemptToFix,
         isRetryReasonAtr,
-        isModified
+        isModified,
       } = ctx
 
       const testSuite = getTestSuitePath(testSuiteAbsolutePath, this.repositoryRoot)
       const store = storage('legacy').getStore()
 
       const extraTags = {
-        [TEST_SOURCE_FILE]: testSuite
+        [TEST_SOURCE_FILE]: testSuite,
       }
       if (isRetry) {
         extraTags[TEST_IS_RETRY] = 'true'
@@ -207,7 +207,7 @@ class VitestPlugin extends CiPlugin {
     this.addSub('ci:vitest:test:pass', ({ span, task }) => {
       if (span) {
         this.telemetry.ciVisEvent(TELEMETRY_EVENT_FINISHED, 'test', {
-          hasCodeowners: !!span.context()._tags[TEST_CODE_OWNERS]
+          hasCodeowners: !!span.context()._tags[TEST_CODE_OWNERS],
         })
         span.setTag(TEST_STATUS, 'pass')
         span.finish(this.taskToFinishTime.get(task))
@@ -222,7 +222,7 @@ class VitestPlugin extends CiPlugin {
       shouldSetProbe,
       promises,
       hasFailedAllRetries,
-      attemptToFixFailed
+      attemptToFixFailed,
     }) => {
       if (!span) {
         return
@@ -237,7 +237,7 @@ class VitestPlugin extends CiPlugin {
         }
       }
       this.telemetry.ciVisEvent(TELEMETRY_EVENT_FINISHED, 'test', {
-        hasCodeowners: !!span.context()._tags[TEST_CODE_OWNERS]
+        hasCodeowners: !!span.context()._tags[TEST_CODE_OWNERS],
       })
       span.setTag(TEST_STATUS, 'fail')
 
@@ -269,11 +269,11 @@ class VitestPlugin extends CiPlugin {
           [TEST_SOURCE_START]: 1, // we can't get the proper start line in vitest
           [TEST_STATUS]: 'skip',
           ...(isDisabled ? { [TEST_MANAGEMENT_IS_DISABLED]: 'true' } : {}),
-          ...(isNew ? { [TEST_IS_NEW]: 'true' } : {})
+          ...(isNew ? { [TEST_IS_NEW]: 'true' } : {}),
         }
       )
       this.telemetry.ciVisEvent(TELEMETRY_EVENT_FINISHED, 'test', {
-        hasCodeowners: !!testSpan.context()._tags[TEST_CODE_OWNERS]
+        hasCodeowners: !!testSpan.context()._tags[TEST_CODE_OWNERS],
       })
       testSpan.finish()
     })
@@ -285,7 +285,7 @@ class VitestPlugin extends CiPlugin {
       this.frameworkVersion = frameworkVersion
       const testSessionSpanContext = this.tracer.extract('text_map', {
         'x-datadog-trace-id': getValueFromEnvSources('DD_CIVISIBILITY_TEST_SESSION_ID'),
-        'x-datadog-parent-id': getValueFromEnvSources('DD_CIVISIBILITY_TEST_MODULE_ID')
+        'x-datadog-parent-id': getValueFromEnvSources('DD_CIVISIBILITY_TEST_MODULE_ID'),
       })
 
       const trimmedCommand = DD_MAJOR < 6 ? this.command : 'vitest run'
@@ -294,14 +294,14 @@ class VitestPlugin extends CiPlugin {
       const metadataTags = {}
       for (const testLevel of TEST_LEVEL_EVENT_TYPES) {
         metadataTags[testLevel] = {
-          [TEST_SESSION_NAME]: testSessionName
+          [TEST_SESSION_NAME]: testSessionName,
         }
       }
       if (this.tracer._exporter.addMetadataTags) {
         const libraryCapabilitiesTags = getLibraryCapabilitiesTags(this.constructor.id)
         metadataTags.test = {
           ...metadataTags.test,
-          ...libraryCapabilitiesTags
+          ...libraryCapabilitiesTags,
         }
         this.tracer._exporter.addMetadataTags(metadataTags)
       }
@@ -326,8 +326,8 @@ class VitestPlugin extends CiPlugin {
         tags: {
           [COMPONENT]: this.constructor.id,
           ...this.testEnvironmentMetadata,
-          ...testSuiteMetadata
-        }
+          ...testSuiteMetadata,
+        },
       })
       this.telemetry.ciVisEvent(TELEMETRY_EVENT_CREATED, 'suite')
       const store = storage('legacy').getStore()
@@ -374,7 +374,7 @@ class VitestPlugin extends CiPlugin {
       isEarlyFlakeDetectionFaulty,
       isTestManagementTestsEnabled,
       vitestPool,
-      onFinish
+      onFinish,
     }) => {
       this.testSessionSpan.setTag(TEST_STATUS, status)
       this.testModuleSpan.setTag(TEST_STATUS, status)
@@ -405,10 +405,31 @@ class VitestPlugin extends CiPlugin {
       finishAllTraceSpans(this.testSessionSpan)
       this.telemetry.count(TELEMETRY_TEST_SESSION, {
         provider: this.ciProviderName,
-        autoInjected: !!getValueFromEnvSources('DD_CIVISIBILITY_AUTO_INSTRUMENTATION_PROVIDER')
+        autoInjected: !!getValueFromEnvSources('DD_CIVISIBILITY_AUTO_INSTRUMENTATION_PROVIDER'),
       })
       this.tracer._exporter.flush(onFinish)
     })
+
+    this.addSub('ci:vitest:coverage-report', ({ rootDir, onDone }) => {
+      this.handleCoverageReport(rootDir, onDone)
+    })
+  }
+
+  /**
+   * Handles the coverage report by discovering and uploading it if enabled.
+   * @param {string} rootDir - The root directory where coverage reports are located.
+   * @param {Function} [onDone] - Callback to signal completion.
+   */
+  handleCoverageReport (rootDir, onDone) {
+    if (!this.libraryConfig?.isCoverageReportUploadEnabled) {
+      onDone()
+      return
+    }
+    this.uploadCoverageReports({
+      rootDir,
+      testEnvironmentMetadata: this.testEnvironmentMetadata,
+      onDone,
+    })
   }
 
   getTestProperties (testManagementTests, testSuite, testName) {

package/packages/datadog-plugin-ws/src/close.js

@@ -4,12 +4,12 @@ const TracingPlugin = require('../../dd-trace/src/plugins/tracing.js')
 const {
   WEBSOCKET_PTR_KIND,
   SPAN_POINTER_DIRECTION,
-  SPAN_POINTER_DIRECTION_NAME
+  SPAN_POINTER_DIRECTION_NAME,
 } = require('../../dd-trace/src/constants')
 const {
   incrementWebSocketCounter,
   buildWebSocketSpanPointerHash,
-  hasDistributedTracingContext
+  hasDistributedTracingContext,
 } = require('./util')
 
 class WSClosePlugin extends TracingPlugin {
@@ -21,14 +21,14 @@ class WSClosePlugin extends TracingPlugin {
   bindStart (ctx) {
     const {
       traceWebsocketMessagesInheritSampling,
-      traceWebsocketMessagesSeparateTraces
+      traceWebsocketMessagesSeparateTraces,
     } = this.config
 
     const { code, data, socket, isPeerClose } = ctx
     if (!socket?.spanContext) return
 
     const spanKind = isPeerClose ? 'consumer' : 'producer'
-    const spanTags = socket.spanContext.spanTags
+    const spanTags = socket.spanTags
     const path = spanTags['resource.name'].split(' ')[1]
     const service = this.serviceName({ pluginConfig: this.config })
     const span = this.startSpan(this.operationName(), {
@@ -37,9 +37,9 @@ class WSClosePlugin extends TracingPlugin {
         'resource.name': `websocket ${path}`,
         'span.type': 'websocket',
         'span.kind': spanKind,
-        'websocket.close.code': code
+        'websocket.close.code': code,
 
-      }
+      },
     }, ctx)
 
     if (data?.toString().length > 0) {
@@ -76,14 +76,13 @@ class WSClosePlugin extends TracingPlugin {
       linkAttributes['dd.kind'] = isIncoming ? 'executed_by' : 'resuming'
 
       // Add span pointer for context propagation
-      if (this.config.traceWebsocketMessagesEnabled && ctx.socket.handshakeSpan) {
-        const handshakeSpan = ctx.socket.handshakeSpan
+      if (this.config.traceWebsocketMessagesEnabled && ctx.socket.spanContext) {
+        const handshakeContext = ctx.socket.spanContext
 
         // Only add span pointers if distributed tracing is enabled and handshake has distributed context
-        if (hasDistributedTracingContext(handshakeSpan, ctx.socket)) {
+        if (hasDistributedTracingContext(handshakeContext, ctx.socket)) {
           const counterType = isIncoming ? 'receiveCounter' : 'sendCounter'
           const counter = incrementWebSocketCounter(ctx.socket, counterType)
-          const handshakeContext = handshakeSpan.context()
 
           const ptrHash = buildWebSocketSpanPointerHash(
             handshakeContext._traceId,
@@ -111,7 +110,7 @@ class WSClosePlugin extends TracingPlugin {
 
       ctx.span.addLink({
         context: ctx.socket.spanContext,
-        attributes: linkAttributes
+        attributes: linkAttributes,
       })
     }
 

package/packages/datadog-plugin-ws/src/index.js

@@ -14,7 +14,7 @@ class WSPlugin extends CompositePlugin {
       server: WSServerPlugin,
       producer: WSProducerPlugin,
       receiver: WSReceiverPlugin,
-      close: WSClosePlugin
+      close: WSClosePlugin,
     }
   }
 

package/packages/datadog-plugin-ws/src/producer.js

@@ -4,12 +4,12 @@ const TracingPlugin = require('../../dd-trace/src/plugins/tracing.js')
 const {
   WEBSOCKET_PTR_KIND,
   SPAN_POINTER_DIRECTION,
-  SPAN_POINTER_DIRECTION_NAME
+  SPAN_POINTER_DIRECTION_NAME,
 } = require('../../dd-trace/src/constants')
 const {
   incrementWebSocketCounter,
   buildWebSocketSpanPointerHash,
-  hasDistributedTracingContext
+  hasDistributedTracingContext,
 } = require('./util')
 
 class WSProducerPlugin extends TracingPlugin {
@@ -22,7 +22,7 @@ class WSProducerPlugin extends TracingPlugin {
     const { byteLength, socket, binary } = ctx
     if (!socket.spanContext) return
 
-    const spanTags = socket.spanContext.spanTags
+    const spanTags = socket.spanTags
     const path = spanTags['resource.name'].split(' ')[1]
     const opCode = binary ? 'binary' : 'text'
     const service = this.serviceName({ pluginConfig: this.config })
@@ -36,8 +36,8 @@ class WSProducerPlugin extends TracingPlugin {
 
       },
       metrics: {
-        'websocket.message.length': byteLength
-      }
+        'websocket.message.length': byteLength,
+      },
 
     }, ctx)
 
@@ -61,13 +61,12 @@ class WSProducerPlugin extends TracingPlugin {
       const linkAttributes = { 'dd.kind': 'resuming' }
 
       // Add span pointer for context propagation
-      if (this.config.traceWebsocketMessagesEnabled && ctx.socket.handshakeSpan) {
-        const handshakeSpan = ctx.socket.handshakeSpan
+      if (this.config.traceWebsocketMessagesEnabled && ctx.socket.spanContext) {
+        const handshakeContext = ctx.socket.spanContext
 
         // Only add span pointers if distributed tracing is enabled and handshake has distributed context
-        if (hasDistributedTracingContext(handshakeSpan, ctx.socket)) {
+        if (hasDistributedTracingContext(handshakeContext, ctx.socket)) {
           const counter = incrementWebSocketCounter(ctx.socket, 'sendCounter')
-          const handshakeContext = handshakeSpan.context()
 
           const ptrHash = buildWebSocketSpanPointerHash(
             handshakeContext._traceId,

package/packages/datadog-plugin-ws/src/receiver.js

@@ -4,12 +4,12 @@ const TracingPlugin = require('../../dd-trace/src/plugins/tracing.js')
 const {
   WEBSOCKET_PTR_KIND,
   SPAN_POINTER_DIRECTION,
-  SPAN_POINTER_DIRECTION_NAME
+  SPAN_POINTER_DIRECTION_NAME,
 } = require('../../dd-trace/src/constants')
 const {
   incrementWebSocketCounter,
   buildWebSocketSpanPointerHash,
-  hasDistributedTracingContext
+  hasDistributedTracingContext,
 } = require('./util')
 
 class WSReceiverPlugin extends TracingPlugin {
@@ -21,13 +21,13 @@ class WSReceiverPlugin extends TracingPlugin {
   bindStart (ctx) {
     const {
       traceWebsocketMessagesInheritSampling,
-      traceWebsocketMessagesSeparateTraces
+      traceWebsocketMessagesSeparateTraces,
     } = this.config
 
     const { byteLength, socket, binary } = ctx
     if (!socket.spanContext) return
 
-    const spanTags = socket.spanContext.spanTags
+    const spanTags = socket.spanTags
     const path = spanTags['resource.name'].split(' ')[1]
     const opCode = binary ? 'binary' : 'text'
 
@@ -43,7 +43,7 @@ class WSReceiverPlugin extends TracingPlugin {
       },
       metrics: {
         'websocket.message.length': byteLength,
-      }
+      },
 
     }, ctx)
 
@@ -72,13 +72,12 @@ class WSReceiverPlugin extends TracingPlugin {
       const linkAttributes = { 'dd.kind': 'executed_by' }
 
       // Add span pointer for context propagation
-      if (this.config.traceWebsocketMessagesEnabled && ctx.socket.handshakeSpan) {
-        const handshakeSpan = ctx.socket.handshakeSpan
+      if (this.config.traceWebsocketMessagesEnabled && ctx.socket.spanContext) {
+        const handshakeContext = ctx.socket.spanContext
 
         // Only add span pointers if distributed tracing is enabled and handshake has distributed context
-        if (hasDistributedTracingContext(handshakeSpan, ctx.socket)) {
+        if (hasDistributedTracingContext(handshakeContext, ctx.socket)) {
           const counter = incrementWebSocketCounter(ctx.socket, 'receiveCounter')
-          const handshakeContext = handshakeSpan.context()
 
           const ptrHash = buildWebSocketSpanPointerHash(
             handshakeContext._traceId,

package/packages/datadog-plugin-ws/src/server.js

@@ -2,8 +2,12 @@
 
 const TracingPlugin = require('../../dd-trace/src/plugins/tracing.js')
 const tags = require('../../../ext/tags.js')
-const { FORMAT_HTTP_HEADERS } = require('../../../ext/formats')
-const { initWebSocketMessageCounters } = require('./util')
+const { HTTP_HEADERS } = require('../../../ext/formats')
+const {
+  createWebSocketSpanContext,
+  hasTraceHeaders,
+  initWebSocketMessageCounters,
+} = require('./util')
 
 const HTTP_STATUS_CODE = tags.HTTP_STATUS_CODE
 
@@ -27,11 +31,12 @@ class WSServerPlugin extends TracingPlugin {
     const indexOfParam = url.indexOf('?')
     const route = indexOfParam === -1 ? url : url.slice(0, indexOfParam)
     const uri = `${protocol}//${host}${route}`
+    const resourceName = `${options.method} ${route}`
 
     ctx.args = { options }
 
     // Extract distributed tracing context from request headers
-    const childOf = this.tracer.extract(FORMAT_HTTP_HEADERS, req.headers)
+    const childOf = this.tracer.extract(HTTP_HEADERS, req.headers)
 
     const service = this.serviceName({ pluginConfig: this.config })
     const span = this.startSpan(this.operationName(), {
@@ -42,19 +47,19 @@ class WSServerPlugin extends TracingPlugin {
         'http.upgraded': 'websocket',
         'http.method': options.method,
         'http.url': uri,
-        'resource.name': `${options.method} ${route}`,
-        'span.kind': 'server'
-      }
+        'resource.name': resourceName,
+        'span.kind': 'server',
+      },
 
     }, ctx)
     ctx.span = span
 
-    ctx.socket.spanContext = ctx.span._spanContext
-    ctx.socket.spanContext.spanTags = ctx.span._spanContext._tags
-    // Store the handshake span for use in message span pointers
-    ctx.socket.handshakeSpan = ctx.span
-    // Store the request headers for distributed tracing check
-    ctx.socket.requestHeaders = req.headers
+    ctx.socket.spanTags = {
+      'resource.name': resourceName,
+      'service.name': service,
+    }
+    ctx.socket.spanContext = createWebSocketSpanContext(ctx.span._spanContext)
+    ctx.socket.hasTraceHeaders = hasTraceHeaders(req.headers)
 
     // Initialize message counters for span pointers
     initWebSocketMessageCounters(ctx.socket)