dd-trace 5.52.0 → 5.54.0

package/packages/datadog-plugin-google-cloud-pubsub/src/consumer.js

@@ -7,7 +7,8 @@ class GoogleCloudPubsubConsumerPlugin extends ConsumerPlugin {
   static get id () { return 'google-cloud-pubsub' }
   static get operation () { return 'receive' }
 
-  start ({ message }) {
+  bindStart (ctx) {
+    const { message } = ctx
     const subscription = message._subscriber._subscription
     const topic = subscription.metadata && subscription.metadata.topic
     const childOf = this.tracer.extract('text_map', message.attributes) || null
@@ -23,25 +24,29 @@ class GoogleCloudPubsubConsumerPlugin extends ConsumerPlugin {
       metrics: {
         'pubsub.ack': 0
       }
-    })
+    }, ctx)
+
     if (this.config.dsmEnabled && message?.attributes) {
       const payloadSize = getMessageSize(message)
       this.tracer.decodeDataStreamsContext(message.attributes)
       this.tracer
         .setCheckpoint(['direction:in', `topic:${topic}`, 'type:google-pubsub'], span, payloadSize)
     }
-  }
 
-  finish (message) {
-    const span = this.activeSpan
+    return ctx.currentStore
+  }
 
-    if (!span) return
+  bindFinish (ctx) {
+    const { message } = ctx
+    const span = ctx.currentStore.span
 
-    if (message.message._handled) {
+    if (message?._handled) {
       span.setTag('pubsub.ack', 1)
     }
 
     super.finish()
+
+    return ctx.parentStore
   }
 }
 

package/packages/datadog-plugin-google-cloud-pubsub/src/producer.js

@@ -7,7 +7,9 @@ class GoogleCloudPubsubProducerPlugin extends ProducerPlugin {
   static get id () { return 'google-cloud-pubsub' }
   static get operation () { return 'request' }
 
-  start ({ request, api, projectId }) {
+  bindStart (ctx) {
+    const { request, api, projectId } = ctx
+
     if (api !== 'publish') return
 
     const messages = request.messages || []
@@ -19,7 +21,7 @@ class GoogleCloudPubsubProducerPlugin extends ProducerPlugin {
         'pubsub.method': api, // TODO: remove
         'pubsub.topic': topic
       }
-    })
+    }, ctx)
 
     for (const msg of messages) {
       if (!msg.attributes) {
@@ -33,6 +35,8 @@ class GoogleCloudPubsubProducerPlugin extends ProducerPlugin {
         DsmPathwayCodec.encode(dataStreamsContext, msg.attributes)
       }
     }
+
+    return ctx.currentStore
   }
 }
 

package/packages/datadog-plugin-google-cloud-vertexai/src/tracing.js

@@ -25,18 +25,18 @@ class GoogleCloudVertexAITracingPlugin extends TracingPlugin {
   bindStart (ctx) {
     const { instance, request, resource, stream } = ctx
 
-    const tags = this.tagRequest(request, instance, stream)
-
     const span = this.startSpan('vertexai.request', {
       service: this.config.service,
       resource,
       kind: 'client',
       meta: {
-        [MEASURED]: 1,
-        ...tags
+        [MEASURED]: 1
       }
     }, false)
 
+    const tags = this.tagRequest(request, instance, stream, span)
+    span.addTags(tags)
+
     const store = storage('legacy').getStore() || {}
     ctx.currentStore = { ...store, span }
 
@@ -51,14 +51,23 @@ class GoogleCloudVertexAITracingPlugin extends TracingPlugin {
 
     const response = result?.response
     if (response) {
-      const tags = this.tagResponse(response)
+      const tags = this.tagResponse(response, span)
       span.addTags(tags)
     }
 
     span.finish()
   }
 
-  tagRequest (request, instance, stream) {
+  /**
+   * Generate the request tags.
+   *
+   * @param {Object} request
+   * @param {Object} instance
+   * @param {boolean} stream
+   * @param {Span} span
+   * @returns {Object}
+   */
+  tagRequest (request, instance, stream, span) {
     const model = extractModel(instance)
     const tags = {
       'vertexai.request.model': model
@@ -81,7 +90,7 @@ class GoogleCloudVertexAITracingPlugin extends TracingPlugin {
       tags['vertexai.request.stream'] = true
     }
 
-    if (!this.isPromptCompletionSampled()) return tags
+    if (!this.isPromptCompletionSampled(span)) return tags
 
     const systemInstructions = extractSystemInstructions(instance)
 
@@ -141,8 +150,16 @@ class GoogleCloudVertexAITracingPlugin extends TracingPlugin {
     }
   }
 
-  tagResponse (response) {
+  /**
+   * Generate the response tags.
+   *
+   * @param {Object} response
+   * @param {Span} span
+   * @returns {Object}
+   */
+  tagResponse (response, span) {
     const tags = {}
+    const isSampled = this.isPromptCompletionSampled(span)
 
     const candidates = response.candidates
     for (const [candidateIdx, candidate] of candidates.entries()) {
@@ -154,7 +171,7 @@ class GoogleCloudVertexAITracingPlugin extends TracingPlugin {
       const role = candidateContent.role
       tags[`vertexai.response.candidates.${candidateIdx}.content.role`] = role
 
-      if (!this.isPromptCompletionSampled()) continue
+      if (!isSampled) continue
 
       const parts = candidateContent.parts
       for (const [partIdx, part] of parts.entries()) {

package/packages/datadog-plugin-graphql/src/execute.js

@@ -70,12 +70,12 @@ function getSignature (document, operationName, operationType, calculate) {
       }
 
       return tools.defaultEngineReportingSignature(document, operationName)
-    } catch (e) {
+    } catch {
       // safety net
     }
   }
 
-  return [operationType, operationName].filter(val => val).join(' ')
+  return [operationType, operationName].filter(Boolean).join(' ')
 }
 
 module.exports = GraphQLExecutePlugin

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

@@ -47,7 +47,7 @@ function getDepth (config) {
 function getVariablesFilter (config) {
   if (typeof config.variables === 'function') {
     return config.variables
-  } else if (config.variables instanceof Array) {
+  } else if (Array.isArray(config.variables)) {
     return variables => pick(variables, config.variables)
   } else if (config.hasOwnProperty('variables')) {
     log.error('Expected `variables` to be an array or function.')
@@ -55,11 +55,12 @@ function getVariablesFilter (config) {
   return null
 }
 
-function getHooks (config) {
-  const noop = () => { }
-  const execute = config.hooks?.execute || noop
-  const parse = config.hooks?.parse || noop
-  const validate = config.hooks?.validate || noop
+const noop = () => {}
+
+function getHooks ({ hooks }) {
+  const execute = hooks?.execute ?? noop
+  const parse = hooks?.parse ?? noop
+  const validate = hooks?.validate ?? noop
 
   return { execute, parse, validate }
 }

package/packages/datadog-plugin-graphql/src/resolve.js

@@ -29,7 +29,7 @@ class GraphQLResolvePlugin extends TracingPlugin {
     const document = context.source
     const fieldNode = info.fieldNodes.find(fieldNode => fieldNode.kind === 'Field')
     const loc = this.config.source && document && fieldNode && fieldNode.loc
-    const source = loc && document.substring(loc.start, loc.end)
+    const source = loc && document.slice(loc.start, loc.end)
 
     const span = this.startSpan('graphql.resolve', {
       service: this.config.service,
@@ -122,7 +122,7 @@ function getResolverInfo (info, args) {
   let resolverInfo = null
   const resolverVars = {}
 
-  if (args && Object.keys(args).length) {
+  if (args) {
     Object.assign(resolverVars, args)
   }
 

package/packages/datadog-plugin-graphql/src/tools/index.js

@@ -1,3 +1,4 @@
+/* eslint-disable unicorn/no-abusive-eslint-disable */
 /* eslint-disable */
 // file mostly untouched from apollo-graphql
 

package/packages/datadog-plugin-graphql/src/tools/signature.js

@@ -1,3 +1,4 @@
+/* eslint-disable unicorn/no-abusive-eslint-disable */
 /* eslint-disable */
 // file mostly untouched from apollo-graphql
 

package/packages/datadog-plugin-graphql/src/tools/transforms.js

@@ -1,3 +1,4 @@
+/* eslint-disable unicorn/no-abusive-eslint-disable */
 /* eslint-disable */
 // file mostly untouched from apollo-graphql
 

package/packages/datadog-plugin-grpc/src/client.js

@@ -87,8 +87,8 @@ class GrpcClientPlugin extends ClientPlugin {
       // more are supported by the library
       // https://github.com/grpc/grpc/blob/v1.60.0/doc/naming.md
       const parts = peer.split(':')
-      if (parts[parts.length - 1].match(/^\d+/)) {
-        const port = parts[parts.length - 1]
+      if (/^\d+/.test(parts.at(-1))) {
+        const port = parts.at(-1)
         const ip = parts.slice(0, -1).join(':')
         span.setTag('network.destination.ip', ip)
         span.setTag('network.destination.port', port)

package/packages/datadog-plugin-grpc/src/util.js

@@ -27,7 +27,7 @@ module.exports = {
       tags.service = service
       tags.package = pkg
     } else {
-      tags.name = methodParts[methodParts.length - 1]
+      tags.name = methodParts.at(-1)
     }
 
     return tags
@@ -49,7 +49,7 @@ module.exports = {
       return config[filter]
     }
 
-    if (config[filter] instanceof Array) {
+    if (Array.isArray(config[filter])) {
       return element => pick(element, config[filter])
     }
 

package/packages/datadog-plugin-http/src/client.js

@@ -49,7 +49,7 @@ class HttpClientPlugin extends ClientPlugin {
         'out.host': hostname
       },
       metrics: {
-        [CLIENT_PORT_KEY]: parseInt(options.port)
+        [CLIENT_PORT_KEY]: Number.parseInt(options.port)
       }
     }, false)
 
@@ -202,15 +202,26 @@ function getFilter (config) {
 function getHeaders (config) {
   if (!Array.isArray(config.headers)) return []
 
-  return config.headers
-    .filter(key => typeof key === 'string')
-    .map(h => h.split(':'))
-    .map(([key, tag]) => [key.toLowerCase(), tag])
+  const result = []
+  for (const header of config.headers) {
+    if (typeof header === 'string') {
+      const separatorIndex = header.indexOf(':')
+      result.push(separatorIndex === -1
+        ? [header, undefined]
+        : [
+            header.slice(0, separatorIndex).toLowerCase(),
+            header.slice(separatorIndex + 1)
+          ]
+      )
+    }
+  }
+  return result
 }
 
+const noop = () => {}
+
 function getHooks (config) {
-  const noop = () => {}
-  const request = (config.hooks && config.hooks.request) || noop
+  const request = config.hooks?.request ?? noop
 
   return { request }
 }

package/packages/datadog-plugin-http2/src/client.js

@@ -51,7 +51,7 @@ class Http2ClientPlugin extends ClientPlugin {
         'out.host': sessionDetails.host
       },
       metrics: {
-        [CLIENT_PORT_KEY]: parseInt(sessionDetails.port)
+        [CLIENT_PORT_KEY]: Number.parseInt(sessionDetails.port)
       }
     }, false)
 
@@ -124,9 +124,9 @@ function extractSessionDetails (authority, options) {
   }
 
   const protocol = authority.protocol || options.protocol || 'https:'
-  let port = '' + (authority.port !== ''
-    ? authority.port
-    : (authority.protocol === 'http:' ? 80 : 443))
+  let port = '' + (authority.port === ''
+    ? (authority.protocol === 'http:' ? 80 : 443)
+    : authority.port)
   let host = authority.hostname || authority.host || 'localhost'
 
   if (protocol === 'https:' && options) {
@@ -138,26 +138,26 @@ function extractSessionDetails (authority, options) {
 }
 
 function hasAmazonSignature (headers, path) {
-  if (headers) {
-    headers = Object.keys(headers)
-      .reduce((prev, next) => Object.assign(prev, {
-        [next.toLowerCase()]: headers[next]
-      }), {})
-
-    if (headers['x-amz-signature']) {
-      return true
-    }
+  if (path?.toLowerCase().includes('x-amz-signature=')) {
+    return true
+  }
 
-    if ([].concat(headers.authorization).some(startsWith('AWS4-HMAC-SHA256'))) {
-      return true
+  if (headers) {
+    for (const [key, value] of Object.entries(headers)) {
+      const lowerCaseKey = key.toLowerCase()
+      if (lowerCaseKey === 'x-amz-signature' && value) {
+        return true
+      }
+      if (lowerCaseKey === 'authorization' && value) {
+        const authorization = Array.isArray(value) ? value : [value]
+        if (authorization.some((val) => val.startsWith('AWS4-HMAC-SHA256'))) {
+          return true
+        }
+      }
     }
   }
 
-  return path && path.toLowerCase().indexOf('x-amz-signature=') !== -1
-}
-
-function startsWith (searchString) {
-  return value => String(value).startsWith(searchString)
+  return false
 }
 
 function getStatusValidator (config) {

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

@@ -0,0 +1,18 @@
+'use strict'
+
+const RedisPlugin = require('../../datadog-plugin-redis/src')
+
+class IOValkeyPlugin extends RedisPlugin {
+  static get id () {
+    return 'iovalkey'
+  }
+
+  static get system () { return 'valkey' }
+
+  constructor (...args) {
+    super(...args)
+    this._spanType = 'valkey'
+  }
+}
+
+module.exports = IOValkeyPlugin

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

@@ -31,7 +31,8 @@ const {
   TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
   TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED,
   TEST_HAS_FAILED_ALL_RETRIES,
-  TEST_RETRY_REASON_TYPES
+  TEST_RETRY_REASON_TYPES,
+  TEST_IS_MODIFIED
 } = require('../../dd-trace/src/plugins/util/test')
 const { COMPONENT } = require('../../dd-trace/src/constants')
 const id = require('../../dd-trace/src/id')
@@ -185,6 +186,7 @@ class JestPlugin extends CiPlugin {
         config._ddFlakyTestRetriesCount = this.libraryConfig?.flakyTestRetriesCount
         config._ddIsDiEnabled = this.libraryConfig?.isDiEnabled ?? false
         config._ddIsKnownTestsEnabled = this.libraryConfig?.isKnownTestsEnabled ?? false
+        config._ddIsImpactedTestsEnabled = this.libraryConfig?.isImpactedTestsEnabled ?? false
       })
     })
 
@@ -333,15 +335,24 @@ class JestPlugin extends CiPlugin {
       this.telemetry.distribution(TELEMETRY_CODE_COVERAGE_NUM_FILES, {}, files.length)
     })
 
-    this.addSub('ci:jest:test:start', (test) => {
+    this.addBind('ci:jest:test:start', (ctx) => {
       const store = storage('legacy').getStore()
-      const span = this.startTestSpan(test)
+      const span = this.startTestSpan(ctx)
+
+      ctx.parentStore = store
+      ctx.currentStore = { ...store, span }
 
-      this.enter(span, store)
       this.activeTestSpan = span
+
+      return ctx.currentStore
+    })
+
+    this.addBind('ci:jest:test:fn', (ctx) => {
+      return ctx.currentStore
     })
 
     this.addSub('ci:jest:test:finish', ({
+      span,
       status,
       testStartLine,
       attemptToFixPassed,
@@ -349,7 +360,6 @@ class JestPlugin extends CiPlugin {
       attemptToFixFailed,
       isAtrRetry
     }) => {
-      const span = storage('legacy').getStore().span
       span.setTag(TEST_STATUS, status)
       if (testStartLine) {
         span.setTag(TEST_SOURCE_START, testStartLine)
@@ -384,19 +394,15 @@ class JestPlugin extends CiPlugin {
       this.activeTestSpan = null
     })
 
-    this.addSub('ci:jest:test:err', ({ error, shouldSetProbe, promises }) => {
-      if (error) {
-        const store = storage('legacy').getStore()
-        if (store && store.span) {
-          const span = store.span
-          span.setTag(TEST_STATUS, 'fail')
-          span.setTag('error', getFormattedError(error, this.repositoryRoot))
-          if (shouldSetProbe) {
-            const probeInformation = this.addDiProbe(error)
-            if (probeInformation) {
-              const { setProbePromise } = probeInformation
-              promises.isProbeReady = withTimeout(setProbePromise, 2000)
-            }
+    this.addSub('ci:jest:test:err', ({ span, error, shouldSetProbe, promises }) => {
+      if (error && span) {
+        span.setTag(TEST_STATUS, 'fail')
+        span.setTag('error', getFormattedError(error, this.repositoryRoot))
+        if (shouldSetProbe) {
+          const probeInformation = this.addDiProbe(error)
+          if (probeInformation) {
+            const { setProbePromise } = probeInformation
+            promises.isProbeReady = withTimeout(setProbePromise, 2000)
           }
         }
       }
@@ -432,7 +438,8 @@ class JestPlugin extends CiPlugin {
       isAttemptToFixRetry,
       isJestRetry,
       isDisabled,
-      isQuarantined
+      isQuarantined,
+      isModified
     } = test
 
     const extraTags = {
@@ -457,6 +464,12 @@ class JestPlugin extends CiPlugin {
     if (isAttemptToFixRetry) {
       extraTags[TEST_IS_RETRY] = 'true'
       extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.atf
+    } else if (isEfdRetry) {
+      extraTags[TEST_IS_RETRY] = 'true'
+      extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.efd
+    } else if (isJestRetry) {
+      extraTags[TEST_IS_RETRY] = 'true'
+      extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.ext
     }
 
     if (isDisabled) {
@@ -467,17 +480,12 @@ class JestPlugin extends CiPlugin {
       extraTags[TEST_MANAGEMENT_IS_QUARANTINED] = 'true'
     }
 
-    if (isNew) {
-      extraTags[TEST_IS_NEW] = 'true'
-      if (isEfdRetry) {
-        extraTags[TEST_IS_RETRY] = 'true'
-        extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.efd
-      }
+    if (isModified) {
+      extraTags[TEST_IS_MODIFIED] = 'true'
     }
 
-    if (isJestRetry) {
-      extraTags[TEST_IS_RETRY] = 'true'
-      extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.ext
+    if (isNew) {
+      extraTags[TEST_IS_NEW] = 'true'
     }
 
     return super.startTestSpan(name, suite, this.testSuiteSpan, extraTags)

package/packages/datadog-plugin-jest/src/util.js

@@ -25,15 +25,15 @@ function getFormattedJestTestParameters (testParameters) {
   // Way 2.
   const parameterKeys = parameterArray[0].split('|').map(key => key.trim())
   const formattedParameters = []
+  let lastFormattedParameter = {}
   for (let index = 0; index < parameterValues.length; index++) {
-    const parameterValue = parameterValues[index]
     const parameterIndex = index % parameterKeys.length
-    if (!parameterIndex) {
-      formattedParameters.push({})
+    if (parameterIndex === 0) {
+      lastFormattedParameter = {}
+      formattedParameters.push(lastFormattedParameter)
     }
-    const parameterKey = parameterKeys[parameterIndex]
-    const lastFormattedParameter = formattedParameters[formattedParameters.length - 1]
-    lastFormattedParameter[parameterKey] = parameterValue
+    const key = parameterKeys[parameterIndex]
+    lastFormattedParameter[key] = parameterValues[index]
   }
 
   return formattedParameters
@@ -57,7 +57,7 @@ function isMarkedAsUnskippable (test) {
   try {
     const testSource = readFileSync(test.path, 'utf8')
     docblocks = parse(extract(testSource))
-  } catch (e) {
+  } catch {
     // If we have issues parsing the file, we'll assume no unskippable was passed
     return false
   }
@@ -69,7 +69,7 @@ function isMarkedAsUnskippable (test) {
 
   try {
     return JSON.parse(docblocks.datadog).unskippable
-  } catch (e) {
+  } catch {
     // If the @datadog block comment is present but malformed, we'll run the suite
     log.warn('@datadog block comment is malformed.')
     return true

package/packages/datadog-plugin-kafkajs/src/batch-consumer.js

@@ -6,7 +6,9 @@ class KafkajsBatchConsumerPlugin extends ConsumerPlugin {
   static get id () { return 'kafkajs' }
   static get operation () { return 'consume-batch' }
 
-  start ({ topic, partition, messages, groupId, clusterId }) {
+  start (ctx) {
+    const { topic, messages, groupId, clusterId } = ctx.extractedArgs || ctx
+
     if (!this.config.dsmEnabled) return
     for (const message of messages) {
       if (!message || !message.headers) continue

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

@@ -64,7 +64,9 @@ class KafkajsConsumerPlugin extends ConsumerPlugin {
     }
   }
 
-  start ({ topic, partition, message, groupId, clusterId }) {
+  bindStart (ctx) {
+    const { topic, partition, message, groupId, clusterId } = ctx.extractedArgs || ctx
+
     let childOf
     const headers = convertToTextMap(message?.headers)
     if (headers) {
@@ -83,7 +85,7 @@ class KafkajsConsumerPlugin extends ConsumerPlugin {
       metrics: {
         'kafka.partition': partition
       }
-    })
+    }, ctx)
     if (message?.offset) span.setTag('kafka.message.offset', message?.offset)
 
     if (this.config.dsmEnabled && headers) {
@@ -97,16 +99,18 @@ class KafkajsConsumerPlugin extends ConsumerPlugin {
     }
 
     if (afterStartCh.hasSubscribers) {
-      afterStartCh.publish({ topic, partition, message, groupId })
+      afterStartCh.publish({ topic, partition, message, groupId, currentStore: ctx.currentStore })
     }
+
+    return ctx.currentStore
   }
 
-  finish () {
+  finish (ctx) {
     if (beforeFinishCh.hasSubscribers) {
       beforeFinishCh.publish()
     }
 
-    super.finish()
+    super.finish(ctx)
   }
 }
 

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

@@ -52,7 +52,9 @@ class KafkajsProducerPlugin extends ProducerPlugin {
    * @param {ProducerResponseItem[]} commitList
    * @returns {void}
    */
-  commit (commitList) {
+  commit (ctx) {
+    const commitList = ctx.result
+
     if (!this.config.dsmEnabled) return
     if (!commitList || !Array.isArray(commitList)) return
     const keys = [
@@ -67,7 +69,8 @@ class KafkajsProducerPlugin extends ProducerPlugin {
     }
   }
 
-  start ({ topic, messages, bootstrapServers, clusterId }) {
+  bindStart (ctx) {
+    const { topic, messages, bootstrapServers, clusterId, disableHeaderInjection } = ctx
     const span = this.startSpan({
       resource: topic,
       meta: {
@@ -79,16 +82,17 @@ class KafkajsProducerPlugin extends ProducerPlugin {
       metrics: {
         'kafka.batch_size': messages.length
       }
-    })
+    }, ctx)
     if (bootstrapServers) {
       span.setTag(BOOTSTRAP_SERVERS_KEY, bootstrapServers)
     }
     for (const message of messages) {
       if (message !== null && typeof message === 'object') {
-        if (!message.headers) {
-          message.headers = {}
+        // message headers are not supported for kafka broker versions <0.11
+        if (!disableHeaderInjection) {
+          message.headers ??= {}
+          this.tracer.inject(span, 'text_map', message.headers)
         }
-        this.tracer.inject(span, 'text_map', message.headers)
         if (this.config.dsmEnabled) {
           const payloadSize = getMessageSize(message)
           const edgeTags = ['direction:out', `topic:${topic}`, 'type:kafka']
@@ -98,10 +102,14 @@ class KafkajsProducerPlugin extends ProducerPlugin {
           }
 
           const dataStreamsContext = this.tracer.setCheckpoint(edgeTags, span, payloadSize)
-          DsmPathwayCodec.encode(dataStreamsContext, message.headers)
+          if (!disableHeaderInjection) {
+            DsmPathwayCodec.encode(dataStreamsContext, message.headers)
+          }
         }
       }
     }
+
+    return ctx.currentStore
   }
 }