dd-trace 5.84.0 → 5.85.0

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

@@ -1,22 +1,141 @@
 'use strict'
 
-const CompositePlugin = require('../../dd-trace/src/plugins/composite')
+const DatabasePlugin = require('../../dd-trace/src/plugins/database')
+const { CLIENT_PORT_KEY } = require('../../dd-trace/src/constants')
+const DatadogTracingHelper = require('./datadog-tracing-helper')
 
-const PrismaClientPlugin = require('./client')
-const PrismaEnginePlugin = require('./engine')
+const databaseDriverMapper = {
+  postgresql: {
+    type: 'sql',
+    'db.type': 'postgres',
+  },
+  mysql: {
+    type: 'sql',
+    'db.type': 'mysql',
+  },
+  mongodb: {
+    type: 'mongodb',
+    'db.type': 'mongodb',
+  },
+  sqlite: {
+    type: 'sql',
+    'db.type': 'sqlite',
+  },
+}
 
-class PrismaPlugin extends CompositePlugin {
+class PrismaPlugin extends DatabasePlugin {
   static id = 'prisma'
-  static get plugins () {
-    return {
-      client: PrismaClientPlugin,
-      engine: PrismaEnginePlugin
+  static system = 'prisma'
+  static prefix = 'tracing:apm:prisma'
+
+  constructor (...args) {
+    super(...args)
+
+    // Subscribe to helper initialization to inject callbacks
+    this.addSub('apm:prisma:helper:init', (prismaHelperCtx) => {
+      prismaHelperCtx.helper = new DatadogTracingHelper(prismaHelperCtx.dbConfig, this)
+    })
+  }
+
+  startEngineSpan (ctx) {
+    const { engineSpan, allEngineSpans, childOf, dbConfig } = ctx
+    const service = this.serviceName({ pluginConfig: this.config, system: this.system })
+    const spanName = engineSpan.name.slice(14) // remove 'prisma:engine:' prefix
+    const options = {
+      childOf,
+      resource: spanName,
+      service,
+      kind: engineSpan.kind,
+      meta: {
+        prisma: {
+          name: spanName,
+          type: 'engine',
+        },
+      },
+    }
+
+    if (spanName === 'db_query') {
+      const query = engineSpan.attributes['db.query.text']
+      const originalStatement = this.maybeTruncate(query)
+      const type = databaseDriverMapper[engineSpan.attributes['db.system']]?.type
+      const dbType = databaseDriverMapper[engineSpan.attributes['db.system']]?.['db.type']
+
+      options.resource = originalStatement
+      options.type = type || engineSpan.attributes['db.system']
+      options.meta['db.type'] = dbType || engineSpan.attributes['db.system']
+      options.meta['db.name'] = dbConfig?.database
+      options.meta['db.user'] = dbConfig?.user
+      options.meta['out.host'] = dbConfig?.host
+      options.meta[CLIENT_PORT_KEY] = dbConfig?.port
     }
+
+    const activeSpan = this.startSpan(this.operationName({ operation: 'engine' }), options)
+    activeSpan._startTime = hrTimeToUnixTimeMs(engineSpan.startTime)
+    for (const span of allEngineSpans) {
+      if (span.parentId === engineSpan.id) {
+        const startCtx = { engineSpan: span, allEngineSpans, childOf: activeSpan, dbConfig }
+        this.startEngineSpan(startCtx)
+      }
+    }
+    const unixEndTime = hrTimeToUnixTimeMs(engineSpan.endTime)
+    activeSpan.finish(unixEndTime)
   }
 
-  configure (config) {
-    return super.configure(config)
+  bindStart (ctx) {
+    const service = this.serviceName({ pluginConfig: this.config })
+    const resource = formatResourceName(ctx.resourceName, ctx.attributes)
+
+    const options = { service, resource }
+
+    if (ctx.resourceName === 'operation') {
+      options.meta = {
+        prisma: {
+          method: ctx.attributes.method,
+          model: ctx.attributes.model,
+          type: 'client',
+        },
+      }
+    }
+    const operationName = this.operationName({ operation: 'client' })
+    this.startSpan(operationName, options, ctx)
+
+    return ctx.currentStore
   }
+
+  end (ctx) {
+    // Only synchronous operations would have `result` on `end`.
+    if (Object.hasOwn(ctx, 'result')) {
+      this.finish(ctx)
+    }
+  }
+
+  bindAsyncStart (ctx) {
+    return this.bindFinish(ctx)
+  }
+
+  asyncStart (ctx) {
+    this.finish(ctx)
+  }
+
+  error (error) {
+    this.addError(error)
+  }
+}
+
+function formatResourceName (resource, attributes) {
+  if (attributes?.name) {
+    return `${attributes.name}`.trim()
+  }
+  if (attributes?.model && attributes.method) {
+    return `${attributes.model}.${attributes.method}`.trim()
+  }
+  return resource
+}
+
+// Opentelemetry time format is defined here
+// https://github.com/open-telemetry/opentelemetry-js/blob/cbc912d/api/src/common/Time.ts#L19-L30.
+function hrTimeToUnixTimeMs ([seconds, nanoseconds]) {
+  return seconds * 1000 + nanoseconds / 1e6
 }
 
 module.exports = PrismaPlugin

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,