dd-trace 5.56.0 → 5.57.1

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

@@ -27,13 +27,14 @@ class HttpServerPlugin extends ServerPlugin {
       this.tracer,
       {
         ...this.config,
-        service: this.config.service || this.serviceName()
+        service: this.config.service || this.serviceName(),
       },
       req,
       res,
       this.operationName()
     )
     span.setTag(COMPONENT, this.constructor.id)
+    span._integrationName = this.constructor.id
 
     this._parentStore = store
     this.enter(span, { ...store, req, res })

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

@@ -40,6 +40,7 @@ class Http2ClientPlugin extends ClientPlugin {
     const childOf = store && allowed ? store.span : null
     const span = this.startSpan(this.operationName(), {
       childOf,
+      integrationName: this.constructor.id,
       meta: {
         [COMPONENT]: this.constructor.id,
         [SPAN_KIND]: CLIENT,

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

@@ -30,6 +30,7 @@ class Http2ServerPlugin extends ServerPlugin {
     )
 
     span.setTag(COMPONENT, this.constructor.id)
+    span._integrationName = this.constructor.id
 
     this.enter(span, { ...store, req, res })
 

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

@@ -252,7 +252,8 @@ class JestPlugin extends CiPlugin {
           [COMPONENT]: this.constructor.id,
           ...this.testEnvironmentMetadata,
           ...testSuiteMetadata
-        }
+        },
+        integrationName: this.constructor.id
       })
       this.telemetry.ciVisEvent(TELEMETRY_EVENT_CREATED, 'suite')
       if (_ddTestCodeCoverageEnabled) {
@@ -318,11 +319,11 @@ class JestPlugin extends CiPlugin {
      * because this subscription happens in a different process from the one
      * fetching the ITR config.
      */
-    this.addSub('ci:jest:test-suite:code-coverage', ({ coverageFiles, testSuite }) => {
+    this.addSub('ci:jest:test-suite:code-coverage', ({ coverageFiles, testSuite, mockedFiles }) => {
       if (!coverageFiles.length) {
         this.telemetry.count(TELEMETRY_CODE_COVERAGE_EMPTY)
       }
-      const files = [...coverageFiles, testSuite]
+      const files = [...coverageFiles, ...mockedFiles, testSuite]
 
       const { _traceId, _spanId } = this.testSuiteSpan.context()
       const formattedCoverage = {

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

@@ -6,7 +6,9 @@ const CachePlugin = require('../../dd-trace/src/plugins/cache')
 class MemcachedPlugin extends CachePlugin {
   static get id () { return 'memcached' }
 
-  start ({ client, server, query }) {
+  bindStart (ctx) {
+    const { client, server, query } = ctx
+
     const address = getAddress(client, server, query)
 
     const meta = {
@@ -23,7 +25,9 @@ class MemcachedPlugin extends CachePlugin {
       resource: query.type,
       type: 'memcached',
       meta
-    })
+    }, ctx)
+
+    return ctx.currentStore
   }
 }
 

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

@@ -152,7 +152,8 @@ class MochaPlugin extends CiPlugin {
           [COMPONENT]: this.constructor.id,
           ...this.testEnvironmentMetadata,
           ...testSuiteMetadata
-        }
+        },
+        integrationName: this.constructor.id
       })
       this.telemetry.ciVisEvent(TELEMETRY_EVENT_CREATED, 'suite')
       if (this.libraryConfig?.isCodeCoverageEnabled) {
@@ -450,7 +451,7 @@ class MochaPlugin extends CiPlugin {
             const testSuite = span.meta[TEST_SUITE]
             const testSuiteSpan = this._testSuites.get(testSuite)
             if (!testSuiteSpan) {
-              log.warn(`Test suite span not found for test span with test suite ${testSuite}`)
+              log.warn('Test suite span not found for test span with test suite', testSuite)
               return formattedSpan
             }
             const suiteTags = getTestSuiteLevelVisibilityTags(testSuiteSpan)

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

@@ -7,29 +7,35 @@ class MoleculerClientPlugin extends ClientPlugin {
   static get id () { return 'moleculer' }
   static get operation () { return 'call' }
 
-  start ({ actionName, opts }) {
+  bindStart (ctx) {
+    const { actionName, opts } = ctx
+
     const span = this.startSpan(this.operationName(), {
       service: this.config.service || this.serviceName(),
       resource: actionName,
       kind: 'client'
-    })
+    }, ctx)
 
     this.tracer.inject(span, 'text_map', opts.meta)
+
+    return ctx.currentStore
   }
 
-  finish ({ broker, ctx }) {
-    const span = this.activeSpan
+  finish (ctx) {
+    const { promiseCtx, broker } = ctx
+
+    const span = ctx.currentStore.span || this.activeSpan
 
-    if (ctx) {
-      const endpoint = ctx.endpoint || {}
+    if (promiseCtx) {
+      const endpoint = promiseCtx.endpoint || {}
       const node = endpoint.node || {}
 
-      this.addHost(node.hostname, node.port)
+      this.addHost({ hostname: node.hostname, port: node.port })
 
-      span.addTags(moleculerTags(broker, ctx, this.config))
+      span.addTags(moleculerTags(broker, promiseCtx, this.config))
     }
 
-    super.finish()
+    super.finish(ctx)
   }
 }
 

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

@@ -7,19 +7,23 @@ class MoleculerServerPlugin extends ServerPlugin {
   static get id () { return 'moleculer' }
   static get operation () { return 'action' }
 
-  start ({ action, ctx, broker }) {
-    const followsFrom = this.tracer.extract('text_map', ctx.meta)
+  bindStart (ctx) {
+    const { action, middlewareCtx, broker } = ctx
+
+    const followsFrom = this.tracer.extract('text_map', middlewareCtx.meta)
     this.startSpan(this.operationName(), {
-      childOf: followsFrom || this.activeSpan,
+      childOf: followsFrom || ctx?.currentStore?.span || this.activeSpan,
       service: this.config.service || this.serviceName(),
       resource: action.name,
       kind: 'server',
       type: 'web',
       meta: {
         'resource.name': action.name,
-        ...moleculerTags(broker, ctx, this.config)
+        ...moleculerTags(broker, middlewareCtx, this.config)
       }
-    })
+    }, ctx)
+
+    return ctx.currentStore
   }
 }
 

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

@@ -31,7 +31,8 @@ class NextPlugin extends ServerPlugin {
         'span.type': 'web',
         'span.kind': 'server',
         'http.method': req.method
-      }
+      },
+      integrationName: this.constructor.id
     })
 
     analyticsSampler.sample(span, this.config.measured, true)

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

@@ -13,6 +13,8 @@ const makeUtilities = require('../../dd-trace/src/plugins/util/llm')
 
 let normalize
 
+const { DD_MAJOR } = require('../../../version')
+
 function safeRequire (path) {
   try {
     return require(path)
@@ -59,11 +61,19 @@ class OpenAiTracingPlugin extends TracingPlugin {
   bindStart (ctx) {
     const { methodName, args, basePath, apiKey } = ctx
     const payload = normalizeRequestPayload(methodName, args)
+    const normalizedMethodName = normalizeMethodName(methodName)
+
     const store = storage('legacy').getStore() || {}
 
+    // hold onto these to make response extraction matching efficient
+    // the original method name corresponds to the SDK method name (e.g. createChatCompletion, chat.completions.create)
+    // the normalized method name corresponds to the resource name (e.g. createChatCompletion, createCompletion)
+    store.originalMethodName = methodName
+    store.normalizedMethodName = normalizedMethodName
+
     const span = this.startSpan('openai.request', {
       service: this.config.service,
-      resource: methodName,
+      resource: DD_MAJOR >= 6 ? normalizedMethodName : methodName,
       type: 'openai',
       kind: 'client',
       meta: {
@@ -129,62 +139,43 @@ class OpenAiTracingPlugin extends TracingPlugin {
       tags['openai.request.stream'] = payload.stream
     }
 
-    switch (methodName) {
+    switch (normalizedMethodName) {
       case 'createFineTune':
-      case 'fine_tuning.jobs.create':
-      case 'fine-tune.create':
         createFineTuneRequestExtraction(tags, payload)
         break
 
       case 'createImage':
-      case 'images.generate':
       case 'createImageEdit':
-      case 'images.edit':
       case 'createImageVariation':
-      case 'images.createVariation':
         commonCreateImageRequestExtraction(tags, payload, openaiStore)
         break
 
       case 'createChatCompletion':
-      case 'chat.completions.create':
         createChatCompletionRequestExtraction(tags, payload, openaiStore)
         break
 
       case 'createFile':
-      case 'files.create':
       case 'retrieveFile':
-      case 'files.retrieve':
         commonFileRequestExtraction(tags, payload)
         break
 
       case 'createTranscription':
-      case 'audio.transcriptions.create':
       case 'createTranslation':
-      case 'audio.translations.create':
         commonCreateAudioRequestExtraction(tags, payload, openaiStore)
         break
 
       case 'retrieveModel':
-      case 'models.retrieve':
         retrieveModelRequestExtraction(tags, payload)
         break
 
       case 'listFineTuneEvents':
-      case 'fine_tuning.jobs.listEvents':
-      case 'fine-tune.listEvents':
       case 'retrieveFineTune':
-      case 'fine_tuning.jobs.retrieve':
-      case 'fine-tune.retrieve':
       case 'deleteModel':
-      case 'models.del':
       case 'cancelFineTune':
-      case 'fine_tuning.jobs.cancel':
-      case 'fine-tune.cancel':
         commonLookupFineTuneRequestExtraction(tags, payload)
         break
 
       case 'createEdit':
-      case 'edits.create':
         createEditRequestExtraction(tags, payload, openaiStore)
         break
     }
@@ -216,9 +207,11 @@ class OpenAiTracingPlugin extends TracingPlugin {
     if (!error && headers?.constructor.name === 'Headers') {
       headers = Object.fromEntries(headers)
     }
-    const methodName = span._spanContext._tags['resource.name']
 
-    body = coerceResponseBody(body, methodName)
+    const resource = span._spanContext._tags['resource.name']
+    const normalizedMethodName = store.normalizedMethodName
+
+    body = coerceResponseBody(body, normalizedMethodName)
 
     const openaiStore = store.openai
 
@@ -227,7 +220,9 @@ class OpenAiTracingPlugin extends TracingPlugin {
       // not using a full regex as it will likely be "https://api.openai.com/..."
       path = new URL(path).pathname
     }
-    const endpoint = lookupOperationEndpoint(methodName, path)
+
+    const originalMethodName = store.originalMethodName
+    const endpoint = lookupOperationEndpoint(normalizedMethodName, originalMethodName, path)
 
     const tags = error
       ? {}
@@ -248,11 +243,11 @@ class OpenAiTracingPlugin extends TracingPlugin {
           'openai.response.created_at': body.created_at
         }
 
-    responseDataExtractionByMethod(methodName, tags, body, openaiStore)
+    responseDataExtractionByMethod(normalizedMethodName, tags, body, openaiStore)
     span.addTags(tags)
 
     span.finish()
-    this.sendLog(methodName, span, tags, openaiStore, error)
+    this.sendLog(resource, span, tags, openaiStore, error)
     this.sendMetrics(headers, body, endpoint, span._duration, error, tags)
   }
 
@@ -343,10 +338,90 @@ class OpenAiTracingPlugin extends TracingPlugin {
   }
 }
 
+function normalizeMethodName (methodName) {
+  switch (methodName) {
+    // moderations
+    case 'moderations.create':
+      return 'createModeration'
+
+    // completions
+    case 'completions.create':
+      return 'createCompletion'
+
+    // chat completions
+    case 'chat.completions.create':
+      return 'createChatCompletion'
+
+    // edits
+    case 'edits.create':
+      return 'createEdit'
+
+    // embeddings
+    case 'embeddings.create':
+      return 'createEmbedding'
+
+    // files
+    case 'files.create':
+      return 'createFile'
+    case 'files.retrieve':
+      return 'retrieveFile'
+    case 'files.del':
+    case 'files.delete':
+      return 'deleteFile'
+    case 'files.retrieveContent':
+    case 'files.content':
+      return 'downloadFile'
+    case 'files.list':
+      return 'listFiles'
+
+    // fine-tuning
+    case 'fine_tuning.jobs.list':
+    case 'fine-tune.list':
+      return 'listFineTunes'
+    case 'fine_tuning.jobs.listEvents':
+    case 'fine-tune.listEvents':
+      return 'listFineTuneEvents'
+    case 'fine_tuning.jobs.create':
+    case 'fine-tune.create':
+      return 'createFineTune'
+    case 'fine_tuning.jobs.retrieve':
+    case 'fine-tune.retrieve':
+      return 'retrieveFineTune'
+    case 'fine_tuning.jobs.cancel':
+    case 'fine-tune.cancel':
+      return 'cancelFineTune'
+
+    // audio
+    case 'audio.transcriptions.create':
+      return 'createTranscription'
+    case 'audio.translations.create':
+      return 'createTranslation'
+
+    // images
+    case 'images.generate':
+      return 'createImage'
+    case 'images.edit':
+      return 'createImageEdit'
+    case 'images.createVariation':
+      return 'createImageVariation'
+
+    // models
+    case 'models.list':
+      return 'listModels'
+    case 'models.retrieve':
+      return 'retrieveModel'
+    case 'models.del':
+    case 'models.delete':
+      return 'deleteModel'
+    default:
+      return methodName
+  }
+}
+
 function countPromptTokens (methodName, payload, model) {
   let promptTokens = 0
   let promptEstimated = false
-  if (methodName === 'chat.completions.create') {
+  if (methodName === 'createChatCompletion') {
     const messages = payload.messages
     for (const message of messages) {
       const content = message.content
@@ -367,7 +442,7 @@ function countPromptTokens (methodName, payload, model) {
         }
       }
     }
-  } else if (methodName === 'completions.create') {
+  } else if (methodName === 'createCompletion') {
     let prompt = payload.prompt
     if (!Array.isArray(prompt)) prompt = [prompt]
 
@@ -468,88 +543,60 @@ function commonCreateImageRequestExtraction (tags, payload, openaiStore) {
 function responseDataExtractionByMethod (methodName, tags, body, openaiStore) {
   switch (methodName) {
     case 'createModeration':
-    case 'moderations.create':
       createModerationResponseExtraction(tags, body)
       break
 
     case 'createCompletion':
-    case 'completions.create':
     case 'createChatCompletion':
-    case 'chat.completions.create':
     case 'createEdit':
-    case 'edits.create':
       commonCreateResponseExtraction(tags, body, openaiStore, methodName)
       break
 
     case 'listFiles':
-    case 'files.list':
     case 'listFineTunes':
-    case 'fine_tuning.jobs.list':
-    case 'fine-tune.list':
     case 'listFineTuneEvents':
-    case 'fine_tuning.jobs.listEvents':
-    case 'fine-tune.listEvents':
       commonListCountResponseExtraction(tags, body)
       break
 
     case 'createEmbedding':
-    case 'embeddings.create':
       createEmbeddingResponseExtraction(tags, body, openaiStore)
       break
 
     case 'createFile':
-    case 'files.create':
     case 'retrieveFile':
-    case 'files.retrieve':
       createRetrieveFileResponseExtraction(tags, body)
       break
 
     case 'deleteFile':
-    case 'files.del':
       deleteFileResponseExtraction(tags, body)
       break
 
     case 'downloadFile':
-    case 'files.retrieveContent':
-    case 'files.content':
       downloadFileResponseExtraction(tags, body)
       break
 
     case 'createFineTune':
-    case 'fine_tuning.jobs.create':
-    case 'fine-tune.create':
     case 'retrieveFineTune':
-    case 'fine_tuning.jobs.retrieve':
-    case 'fine-tune.retrieve':
     case 'cancelFineTune':
-    case 'fine_tuning.jobs.cancel':
-    case 'fine-tune.cancel':
       commonFineTuneResponseExtraction(tags, body)
       break
 
     case 'createTranscription':
-    case 'audio.transcriptions.create':
     case 'createTranslation':
-    case 'audio.translations.create':
       createAudioResponseExtraction(tags, body)
       break
 
     case 'createImage':
-    case 'images.generate':
     case 'createImageEdit':
-    case 'images.edit':
     case 'createImageVariation':
-    case 'images.createVariation':
       commonImageResponseExtraction(tags, body)
       break
 
     case 'listModels':
-    case 'models.list':
       listModelsResponseExtraction(tags, body)
       break
 
     case 'retrieveModel':
-    case 'models.retrieve':
       retrieveModelResponseExtraction(tags, body)
       break
   }
@@ -641,8 +688,8 @@ function commonFineTuneResponseExtraction (tags, body) {
 
 // the OpenAI package appears to stream the content download then provide it all as a singular string
 function downloadFileResponseExtraction (tags, body) {
-  if (!body.file) return
-  tags['openai.response.total_bytes'] = body.file.length
+  if (typeof body.file !== 'string') return
+  tags['openai.response.total_bytes'] = Buffer.byteLength(body.file)
 }
 
 function deleteFileResponseExtraction (tags, body) {
@@ -763,7 +810,7 @@ function usageExtraction (tags, body, methodName, openaiStore) {
     promptTokens = body.usage.prompt_tokens
     completionTokens = body.usage.completion_tokens
     totalTokens = body.usage.total_tokens
-  } else if (body.model && ['chat.completions.create', 'completions.create'].includes(methodName)) {
+  } else if (body.model && ['createChatCompletion', 'createCompletion'].includes(methodName)) {
     // estimate tokens based on method name for completions and chat completions
     const { model } = body
 
@@ -819,8 +866,6 @@ function tagChatCompletionRequestContent (contents, messageIdx, tags) {
 function coerceResponseBody (body, methodName) {
   switch (methodName) {
     case 'downloadFile':
-    case 'files.retrieveContent':
-    case 'files.content':
       return { file: body }
   }
 
@@ -839,42 +884,42 @@ function coerceResponseBody (body, methodName) {
 }
 
 // This method is used to replace a dynamic URL segment with an asterisk
-function lookupOperationEndpoint (operationId, url) {
+function lookupOperationEndpoint (operationId, methodName, url) {
   switch (operationId) {
     case 'deleteModel':
-    case 'models.del':
     case 'retrieveModel':
-    case 'models.retrieve':
       return '/v1/models/*'
 
     case 'deleteFile':
-    case 'files.del':
     case 'retrieveFile':
-    case 'files.retrieve':
       return '/v1/files/*'
 
     case 'downloadFile':
-    case 'files.retrieveContent':
-    case 'files.content':
       return '/v1/files/*/content'
 
     case 'retrieveFineTune':
-    case 'fine-tune.retrieve':
-      return '/v1/fine-tunes/*'
-    case 'fine_tuning.jobs.retrieve':
-      return '/v1/fine_tuning/jobs/*'
+      switch (methodName) {
+        case 'fine_tuning.jobs.retrieve':
+          return '/v1/fine_tuning/jobs/*'
+        default:
+          return '/v1/fine-tunes/*'
+      }
 
     case 'listFineTuneEvents':
-    case 'fine-tune.listEvents':
-      return '/v1/fine-tunes/*/events'
-    case 'fine_tuning.jobs.listEvents':
-      return '/v1/fine_tuning/jobs/*/events'
+      switch (methodName) {
+        case 'fine_tuning.jobs.listEvents':
+          return '/v1/fine_tuning/jobs/*/events'
+        default:
+          return '/v1/fine-tunes/*/events'
+      }
 
     case 'cancelFineTune':
-    case 'fine-tune.cancel':
-      return '/v1/fine-tunes/*/cancel'
-    case 'fine_tuning.jobs.cancel':
-      return '/v1/fine_tuning/jobs/*/cancel'
+      switch (methodName) {
+        case 'fine_tuning.jobs.cancel':
+          return '/v1/fine_tuning/jobs/*/cancel'
+        default:
+          return '/v1/fine-tunes/*/cancel'
+      }
   }
 
   return url
@@ -909,6 +954,7 @@ function normalizeRequestPayload (methodName, args) {
 
     case 'deleteFile':
     case 'files.del':
+    case 'files.delete':
     case 'retrieveFile':
     case 'files.retrieve':
     case 'downloadFile':
@@ -929,6 +975,7 @@ function normalizeRequestPayload (methodName, args) {
     case 'fine-tune.retrieve':
     case 'deleteModel':
     case 'models.del':
+    case 'models.delete':
     case 'cancelFineTune':
     case 'fine_tuning.jobs.cancel':
     case 'fine-tune.cancel':

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

@@ -1,8 +1,8 @@
 'use strict'
 
-const LogPlugin = require('../../dd-trace/src/plugins/log_plugin')
+const StructuredLogPlugin = require('../../dd-trace/src/plugins/structured_log_plugin')
 
-class PinoPlugin extends LogPlugin {
+class PinoPlugin extends StructuredLogPlugin {
   static get id () {
     return 'pino'
   }

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

@@ -0,0 +1,62 @@
+'use strict'
+const DatabasePlugin = require('../../dd-trace/src/plugins/database')
+class PrismaClientPlugin extends DatabasePlugin {
+  static get id () { return 'prisma' }
+  static get operation () { return 'client' }
+  static get system () { return 'prisma' }
+  static get prefix () {
+    return 'tracing:apm:prisma:client'
+  }
+
+  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: this.operation })
+    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
+}
+
+module.exports = PrismaClientPlugin