dd-trace 5.53.0 → 5.55.0

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, disableHeaderInjection }) {
+  bindStart (ctx) {
+    const { topic, messages, bootstrapServers, clusterId, disableHeaderInjection } = ctx
     const span = this.startSpan({
       resource: topic,
       meta: {
@@ -79,7 +82,7 @@ class KafkajsProducerPlugin extends ProducerPlugin {
       metrics: {
         'kafka.batch_size': messages.length
       }
-    })
+    }, ctx)
     if (bootstrapServers) {
       span.setTag(BOOTSTRAP_SERVERS_KEY, bootstrapServers)
     }
@@ -105,6 +108,8 @@ class KafkajsProducerPlugin extends ProducerPlugin {
         }
       }
     }
+
+    return ctx.currentStore
   }
 }
 

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

@@ -4,7 +4,7 @@ function convertToTextMap (bufferMap) {
   if (!bufferMap) return null
 
   // rdKafka returns an array of header maps
-  if (bufferMap instanceof Array) {
+  if (Array.isArray(bufferMap)) {
     const headers = {}
     for (const headerMap of bufferMap) {
       for (const key of Object.keys(headerMap)) {

package/packages/datadog-plugin-langchain/src/handlers/chain.js

@@ -3,35 +3,35 @@
 const LangChainHandler = require('./default')
 
 class LangChainChainHandler extends LangChainHandler {
-  getSpanStartTags (ctx) {
+  getSpanStartTags (ctx, provider, span) {
     const tags = {}
 
-    if (!this.isPromptCompletionSampled()) return tags
+    if (!this.isPromptCompletionSampled(span)) return tags
 
     let inputs = ctx.args?.[0]
     inputs = Array.isArray(inputs) ? inputs : [inputs]
 
     for (const idx in inputs) {
       const input = inputs[idx]
-      if (typeof input !== 'object') {
-        tags[`langchain.request.inputs.${idx}`] = this.normalize(input)
-      } else {
+      if (typeof input === 'object') {
         for (const [key, value] of Object.entries(input)) {
         // these are mappings to the python client names, ie lc_kwargs
         // only present on BaseMessage types
           if (key.includes('lc_')) continue
           tags[`langchain.request.inputs.${idx}.${key}`] = this.normalize(value)
         }
+      } else {
+        tags[`langchain.request.inputs.${idx}`] = this.normalize(input)
       }
     }
 
     return tags
   }
 
-  getSpanEndTags (ctx) {
+  getSpanEndTags (ctx, span) {
     const tags = {}
 
-    if (!this.isPromptCompletionSampled()) return tags
+    if (!this.isPromptCompletionSampled(span)) return tags
 
     let outputs = ctx.result
     outputs = Array.isArray(outputs) ? outputs : [outputs]

package/packages/datadog-plugin-langchain/src/handlers/embedding.js

@@ -3,12 +3,12 @@
 const LangChainHandler = require('./default')
 
 class LangChainEmbeddingHandler extends LangChainHandler {
-  getSpanStartTags (ctx) {
+  getSpanStartTags (ctx, provider, span) {
     const tags = {}
 
     const inputTexts = ctx.args?.[0]
 
-    const sampled = this.isPromptCompletionSampled()
+    const sampled = this.isPromptCompletionSampled(span)
     if (typeof inputTexts === 'string') {
       // embed query
       if (sampled) {

package/packages/datadog-plugin-langchain/src/handlers/language_models/chat_model.js

@@ -5,7 +5,7 @@ const LangChainLanguageModelHandler = require('.')
 const COMPLETIONS = 'langchain.response.completions'
 
 class LangChainChatModelHandler extends LangChainLanguageModelHandler {
-  getSpanStartTags (ctx, provider) {
+  getSpanStartTags (ctx, provider, span) {
     const tags = {}
 
     const inputs = ctx.args?.[0]
@@ -15,7 +15,7 @@ class LangChainChatModelHandler extends LangChainLanguageModelHandler {
 
       for (const messageIndex in messageSet) {
         const message = messageSet[messageIndex]
-        if (this.isPromptCompletionSampled()) {
+        if (this.isPromptCompletionSampled(span)) {
           tags[`langchain.request.messages.${messageSetIndex}.${messageIndex}.content`] =
             this.normalize(message.content) || ''
         }
@@ -39,11 +39,13 @@ class LangChainChatModelHandler extends LangChainLanguageModelHandler {
     return tags
   }
 
-  getSpanEndTags (ctx) {
+  getSpanEndTags (ctx, span) {
     const { result } = ctx
 
     const tags = {}
 
+    const sampled = this.isPromptCompletionSampled(span)
+
     this.extractTokenMetrics(ctx.currentStore?.span, result)
 
     for (const messageSetIdx in result?.generations) {
@@ -56,7 +58,7 @@ class LangChainChatModelHandler extends LangChainLanguageModelHandler {
         const message = chatCompletion.message
         let toolCalls = message.tool_calls
 
-        if (text && this.isPromptCompletionSampled()) {
+        if (text && sampled) {
           tags[
           `${COMPLETIONS}.${messageSetIdx}.${chatCompletionIdx}.content`
           ] = this.normalize(text)

package/packages/datadog-plugin-langchain/src/handlers/language_models/llm.js

@@ -3,12 +3,12 @@
 const LangChainLanguageModelHandler = require('.')
 
 class LangChainLLMHandler extends LangChainLanguageModelHandler {
-  getSpanStartTags (ctx, provider) {
+  getSpanStartTags (ctx, provider, span) {
     const tags = {}
 
     const prompts = ctx.args?.[0]
     for (const promptIdx in prompts) {
-      if (!this.isPromptCompletionSampled()) continue
+      if (!this.isPromptCompletionSampled(span)) continue
 
       const prompt = prompts[promptIdx]
       tags[`langchain.request.prompts.${promptIdx}.content`] = this.normalize(prompt) || ''
@@ -30,16 +30,17 @@ class LangChainLLMHandler extends LangChainLanguageModelHandler {
     return tags
   }
 
-  getSpanEndTags (ctx) {
+  getSpanEndTags (ctx, span) {
     const { result } = ctx
 
     const tags = {}
+    const sampled = this.isPromptCompletionSampled(span)
 
     this.extractTokenMetrics(ctx.currentStore?.span, result)
 
     for (const completionIdx in result?.generations) {
       const completion = result.generations[completionIdx]
-      if (this.isPromptCompletionSampled()) {
+      if (sampled) {
         tags[`langchain.response.completions.${completionIdx}.text`] = this.normalize(completion[0].text) || ''
       }
 

package/packages/datadog-plugin-langchain/src/tracing.js

@@ -49,23 +49,24 @@ class BaseLangChainTracingPlugin extends TracingPlugin {
     const provider = handler.extractProvider(instance)
     const model = handler.extractModel(instance)
 
-    const tags = handler.getSpanStartTags(ctx, provider) || []
-
-    if (apiKey) tags[API_KEY] = apiKey
-    if (provider) tags[PROVIDER] = provider
-    if (model) tags[MODEL] = model
-    if (type) tags[TYPE] = type
-
     const span = this.startSpan('langchain.request', {
       service: this.config.service,
       resource,
       kind: 'client',
       meta: {
-        [MEASURED]: 1,
-        ...tags
+        [MEASURED]: 1
       }
     }, false)
 
+    const tags = handler.getSpanStartTags(ctx, provider, span) || []
+
+    if (apiKey) tags[API_KEY] = apiKey
+    if (provider) tags[PROVIDER] = provider
+    if (model) tags[MODEL] = model
+    if (type) tags[TYPE] = type
+
+    span.addTags(tags)
+
     const store = storage('legacy').getStore() || {}
     ctx.currentStore = { ...store, span }
 
@@ -78,7 +79,7 @@ class BaseLangChainTracingPlugin extends TracingPlugin {
     const { type } = ctx
 
     const handler = this.handlers[type]
-    const tags = handler.getSpanEndTags(ctx) || {}
+    const tags = handler.getSpanEndTags(ctx, span) || {}
 
     span.addTags(tags)
 

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

@@ -3,8 +3,6 @@
 const { storage } = require('../../datadog-core')
 const MySQLPlugin = require('../../datadog-plugin-mysql/src')
 
-let skippedStore
-
 class MariadbPlugin extends MySQLPlugin {
   static get id () { return 'mariadb' }
   static get system () { return 'mariadb' }
@@ -12,14 +10,10 @@ class MariadbPlugin extends MySQLPlugin {
   constructor (...args) {
     super(...args)
 
-    this.addSub(`apm:${this.component}:pool:skip`, () => {
-      skippedStore = storage('legacy').getStore()
-      storage('legacy').enterWith({ noop: true })
-    })
+    this.addBind(`apm:${this.component}:pool:skip`, () => ({ noop: true }))
 
-    this.addSub(`apm:${this.component}:pool:unskip`, () => {
-      storage('legacy').enterWith(skippedStore)
-      skippedStore = undefined
+    this.addSub(`apm:${this.component}:command:add`, ctx => {
+      ctx.parentStore = storage('legacy').getStore()
     })
   }
 }

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

@@ -2,6 +2,7 @@
 
 const CiPlugin = require('../../dd-trace/src/plugins/ci_plugin')
 const { storage } = require('../../datadog-core')
+const { getEnvironmentVariable } = require('../../dd-trace/src/config-helper')
 
 const {
   TEST_STATUS,
@@ -37,7 +38,9 @@ const {
   TEST_MANAGEMENT_IS_ATTEMPT_TO_FIX,
   TEST_HAS_FAILED_ALL_RETRIES,
   TEST_MANAGEMENT_ATTEMPT_TO_FIX_PASSED,
-  TEST_RETRY_REASON_TYPES
+  TEST_RETRY_REASON_TYPES,
+  TEST_IS_MODIFIED,
+  isModifiedTest
 } = require('../../dd-trace/src/plugins/util/test')
 const { COMPONENT } = require('../../dd-trace/src/constants')
 const {
@@ -131,11 +134,9 @@ class MochaPlugin extends CiPlugin {
         testSuiteMetadata[TEST_ITR_FORCED_RUN] = 'true'
         this.telemetry.count(TELEMETRY_ITR_FORCED_TO_RUN, { testLevel: 'suite' })
       }
-      if (this.repositoryRoot !== this.sourceRoot && !!this.repositoryRoot) {
-        testSuiteMetadata[TEST_SOURCE_FILE] = getTestSuitePath(testSuiteAbsolutePath, this.repositoryRoot)
-      } else {
-        testSuiteMetadata[TEST_SOURCE_FILE] = testSuite
-      }
+      testSuiteMetadata[TEST_SOURCE_FILE] = this.repositoryRoot !== this.sourceRoot && !!this.repositoryRoot
+        ? getTestSuitePath(testSuiteAbsolutePath, this.repositoryRoot)
+        : testSuite
       if (testSuiteMetadata[TEST_SOURCE_FILE]) {
         testSuiteMetadata[TEST_SOURCE_START] = 1
       }
@@ -192,6 +193,19 @@ class MochaPlugin extends CiPlugin {
       return ctx.currentStore
     })
 
+    this.addSub('ci:mocha:test:is-modified', ({ modifiedTests, file, onDone }) => {
+      const testPath = getTestSuitePath(file, this.repositoryRoot)
+      const isModified = isModifiedTest(
+        testPath,
+        null,
+        null,
+        modifiedTests,
+        this.constructor.id
+      )
+
+      onDone(isModified)
+    })
+
     this.addBind('ci:mocha:test:fn', (ctx) => {
       return ctx.currentStore
     })
@@ -416,7 +430,7 @@ class MochaPlugin extends CiPlugin {
         finishAllTraceSpans(this.testSessionSpan)
         this.telemetry.count(TELEMETRY_TEST_SESSION, {
           provider: this.ciProviderName,
-          autoInjected: !!process.env.DD_CIVISIBILITY_AUTO_INSTRUMENTATION_PROVIDER
+          autoInjected: !!getEnvironmentVariable('DD_CIVISIBILITY_AUTO_INSTRUMENTATION_PROVIDER')
         })
       }
       this.libraryConfig = null
@@ -453,6 +467,10 @@ class MochaPlugin extends CiPlugin {
         this.tracer._exporter.export(trace)
       })
     })
+
+    this.addBind('ci:mocha:global:run', (ctx) => {
+      return ctx.currentStore
+    })
   }
 
   startTestSpan (testInfo) {
@@ -466,7 +484,8 @@ class MochaPlugin extends CiPlugin {
       isParallel,
       isAttemptToFix,
       isDisabled,
-      isQuarantined
+      isQuarantined,
+      isModified
     } = testInfo
 
     const extraTags = {}
@@ -495,20 +514,26 @@ class MochaPlugin extends CiPlugin {
       extraTags[TEST_MANAGEMENT_IS_QUARANTINED] = 'true'
     }
 
+    if (isModified) {
+      extraTags[TEST_IS_MODIFIED] = 'true'
+      if (isEfdRetry) {
+        extraTags[TEST_IS_RETRY] = 'true'
+        extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.efd
+      }
+    }
+
     const testSuite = getTestSuitePath(testSuiteAbsolutePath, this.sourceRoot)
     const testSuiteSpan = this._testSuites.get(testSuite)
 
-    if (this.repositoryRoot !== this.sourceRoot && !!this.repositoryRoot) {
-      extraTags[TEST_SOURCE_FILE] = getTestSuitePath(testSuiteAbsolutePath, this.repositoryRoot)
-    } else {
-      extraTags[TEST_SOURCE_FILE] = testSuite
-    }
+    extraTags[TEST_SOURCE_FILE] = this.repositoryRoot !== this.sourceRoot && !!this.repositoryRoot
+      ? getTestSuitePath(testSuiteAbsolutePath, this.repositoryRoot)
+      : testSuite
 
     if (isNew) {
       extraTags[TEST_IS_NEW] = 'true'
       if (isEfdRetry) {
         extraTags[TEST_IS_RETRY] = 'true'
-        extraTags[TEST_RETRY_REASON] = 'early_flake_detection'
+        extraTags[TEST_RETRY_REASON] = TEST_RETRY_REASON_TYPES.efd
       }
     }
 

package/packages/datadog-plugin-mongodb-core/src/index.js

@@ -3,6 +3,7 @@
 const { isTrue } = require('../../dd-trace/src/util')
 const DatabasePlugin = require('../../dd-trace/src/plugins/database')
 const coalesce = require('koalas')
+const { getEnvironmentVariable } = require('../../dd-trace/src/config-helper')
 
 class MongodbCorePlugin extends DatabasePlugin {
   static get id () { return 'mongodb-core' }
@@ -14,7 +15,7 @@ class MongodbCorePlugin extends DatabasePlugin {
   configure (config) {
     super.configure(config)
 
-    const heartbeatFromEnv = process.env.DD_TRACE_MONGODB_HEARTBEAT_ENABLED
+    const heartbeatFromEnv = getEnvironmentVariable('DD_TRACE_MONGODB_HEARTBEAT_ENABLED')
 
     this.config.heartbeatEnabled = coalesce(
       config.heartbeatEnabled,
@@ -103,7 +104,7 @@ function getResource (plugin, ns, query, operationName) {
 }
 
 function truncate (input) {
-  return input.slice(0, Math.min(input.length, 10000))
+  return input.slice(0, Math.min(input.length, 10_000))
 }
 
 function shouldSimplify (input) {

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

@@ -1,5 +1,6 @@
 'use strict'
 
+const { storage } = require('../../datadog-core')
 const CLIENT_PORT_KEY = require('../../dd-trace/src/constants')
 const DatabasePlugin = require('../../dd-trace/src/plugins/database')
 
@@ -7,22 +8,34 @@ class MySQLPlugin extends DatabasePlugin {
   static get id () { return 'mysql' }
   static get system () { return 'mysql' }
 
-  start (payload) {
-    const service = this.serviceName({ pluginConfig: this.config, dbConfig: payload.conf, system: this.system })
+  constructor () {
+    super(...arguments)
+
+    this.addSub(`apm:${this.component}:connection:start`, ctx => {
+      ctx.parentStore = storage('legacy').getStore()
+    })
+
+    this.addBind(`apm:${this.component}:connection:finish`, ctx => ctx.parentStore)
+  }
+
+  bindStart (ctx) {
+    const service = this.serviceName({ pluginConfig: this.config, dbConfig: ctx.conf, system: this.system })
     const span = this.startSpan(this.operationName(), {
       service,
-      resource: payload.sql,
+      resource: ctx.sql,
       type: 'sql',
       kind: 'client',
       meta: {
         'db.type': this.system,
-        'db.user': payload.conf.user,
-        'db.name': payload.conf.database,
-        'out.host': payload.conf.host,
-        [CLIENT_PORT_KEY]: payload.conf.port
+        'db.user': ctx.conf.user,
+        'db.name': ctx.conf.database,
+        'out.host': ctx.conf.host,
+        [CLIENT_PORT_KEY]: ctx.conf.port
       }
-    })
-    payload.sql = this.injectDbmQuery(span, payload.sql, service)
+    }, ctx)
+    ctx.sql = this.injectDbmQuery(span, ctx.sql, service)
+
+    return ctx.currentStore
   }
 }
 

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

@@ -1,9 +1,25 @@
 'use strict'
 
+const { storage } = require('../../datadog-core')
 const MySQLPlugin = require('../../datadog-plugin-mysql/src')
 
 class MySQL2Plugin extends MySQLPlugin {
   static get id () { return 'mysql2' }
+
+  constructor () {
+    super(...arguments)
+
+    this.addSub(`apm:${this.component}:command:add`, ctx => {
+      ctx.parentStore = storage('legacy').getStore()
+    })
+
+    this.addBind(`apm:${this.component}:command:start`, ctx => ctx.parentStore)
+    this.addBind(`apm:${this.component}:command:finish`, ctx => ctx.parentStore)
+  }
+
+  bindStart (ctx) {
+    return storage('legacy').run(ctx.parentStore, () => super.bindStart(ctx))
+  }
 }
 
 module.exports = MySQL2Plugin

package/packages/datadog-plugin-net/src/tcp.js

@@ -31,7 +31,7 @@ class NetTCPPlugin extends ClientPlugin {
 
     this.startSpan('tcp.connect', {
       service: this.config.service,
-      resource: [host, port].filter(val => val).join(':'),
+      resource: [host, port].filter(Boolean).join(':'),
       kind: 'client',
       meta: {
         'tcp.remote.host': host,

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

@@ -6,7 +6,7 @@ const analyticsSampler = require('../../dd-trace/src/analytics_sampler')
 const { COMPONENT } = require('../../dd-trace/src/constants')
 const web = require('../../dd-trace/src/plugins/util/web')
 
-const errorPages = ['/404', '/500', '/_error', '/_not-found', '/_not-found/page']
+const errorPages = new Set(['/404', '/500', '/_error', '/_not-found', '/_not-found/page'])
 
 class NextPlugin extends ServerPlugin {
   static get id () {
@@ -97,7 +97,7 @@ class NextPlugin extends ServerPlugin {
 
     // Only use error page names if there's not already a name
     const current = span.context()._tags['next.page']
-    const isErrorPage = errorPages.includes(page)
+    const isErrorPage = errorPages.has(page)
 
     if (current && isErrorPage) {
       return
@@ -106,7 +106,7 @@ class NextPlugin extends ServerPlugin {
     // remove ending /route or /page for appDir projects
     // need to check if not an error page too, as those are marked as app directory
     // in newer versions
-    if (isAppPath && !isErrorPage) page = page.substring(0, page.lastIndexOf('/'))
+    if (isAppPath && !isErrorPage) page = page.slice(0, Math.max(0, page.lastIndexOf('/')))
 
     // handle static resource
     if (isStatic) {
@@ -134,12 +134,13 @@ function normalizeConfig (config) {
     ? config.validateStatus
     : code => code < 500
 
-  return Object.assign({}, config, { hooks, validateStatus })
+  return { ...config, hooks, validateStatus }
 }
 
+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-openai/src/services.js

@@ -11,8 +11,8 @@ let logger = null
 let interval = null
 
 module.exports.init = function (tracerConfig) {
-  if (tracerConfig && tracerConfig.dogstatsd) {
-    metrics = new DogStatsDClient({
+  metrics = tracerConfig && tracerConfig.dogstatsd
+    ? new DogStatsDClient({
       host: tracerConfig.dogstatsd.hostname,
       port: tracerConfig.dogstatsd.port,
       tags: [
@@ -21,21 +21,17 @@ module.exports.init = function (tracerConfig) {
         `version:${tracerConfig.tags.version}`
       ]
     })
-  } else {
-    metrics = new NoopDogStatsDClient()
-  }
+    : new NoopDogStatsDClient()
 
-  if (tracerConfig && tracerConfig.apiKey) {
-    logger = new ExternalLogger({
+  logger = tracerConfig && tracerConfig.apiKey
+    ? new ExternalLogger({
       ddsource: 'openai',
       hostname: tracerConfig.hostname,
       service: tracerConfig.service,
       apiKey: tracerConfig.apiKey,
       interval: FLUSH_INTERVAL
     })
-  } else {
-    logger = new NoopExternalLogger()
-  }
+    : new NoopExternalLogger()
 
   interval = setInterval(() => {
     metrics.flush()