dd-trace 5.61.1 → 5.63.0

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

@@ -13,9 +13,9 @@ const LangChainLanguageModelHandler = require('./handlers/language_models')
 const LangChainEmbeddingHandler = require('./handlers/embedding')
 
 class BaseLangChainTracingPlugin extends TracingPlugin {
-  static get id () { return 'langchain' }
-  static get operation () { return 'invoke' }
-  static get system () { return 'langchain' }
+  static id = 'langchain'
+  static operation = 'invoke'
+  static system = 'langchain'
 
   constructor () {
     super(...arguments)
@@ -85,75 +85,57 @@ class BaseLangChainTracingPlugin extends TracingPlugin {
 }
 
 class RunnableSequenceInvokePlugin extends BaseLangChainTracingPlugin {
-  static get id () { return 'langchain_rs_invoke' }
-  static get lcType () { return 'chain' }
-  static get prefix () {
-    return 'tracing:orchestrion:@langchain/core:RunnableSequence_invoke'
-  }
+  static id = 'langchain_rs_invoke'
+  static lcType = 'chain'
+  static prefix = 'tracing:orchestrion:@langchain/core:RunnableSequence_invoke'
 }
 
 class RunnableSequenceBatchPlugin extends BaseLangChainTracingPlugin {
-  static get id () { return 'langchain_rs_batch' }
-  static get lcType () { return 'chain' }
-  static get prefix () {
-    return 'tracing:orchestrion:@langchain/core:RunnableSequence_batch'
-  }
+  static id = 'langchain_rs_batch'
+  static lcType = 'chain'
+  static prefix = 'tracing:orchestrion:@langchain/core:RunnableSequence_batch'
 }
 
 class BaseChatModelGeneratePlugin extends BaseLangChainTracingPlugin {
-  static get id () { return 'langchain_chat_model_generate' }
-  static get lcType () { return 'chat_model' }
-  static get prefix () {
-    return 'tracing:orchestrion:@langchain/core:BaseChatModel_generate'
-  }
+  static id = 'langchain_chat_model_generate'
+  static lcType = 'chat_model'
+  static prefix = 'tracing:orchestrion:@langchain/core:BaseChatModel_generate'
 }
 
 class BaseLLMGeneratePlugin extends BaseLangChainTracingPlugin {
-  static get id () { return 'langchain_llm_generate' }
-  static get lcType () { return 'llm' }
-  static get prefix () {
-    return 'tracing:orchestrion:@langchain/core:BaseLLM_generate'
-  }
+  static id = 'langchain_llm_generate'
+  static lcType = 'llm'
+  static prefix = 'tracing:orchestrion:@langchain/core:BaseLLM_generate'
 }
 
 class EmbeddingsEmbedQueryPlugin extends BaseLangChainTracingPlugin {
-  static get id () { return 'langchain_embeddings_embed_query' }
-  static get lcType () { return 'embedding' }
-  static get prefix () {
-    return 'tracing:apm:@langchain/core:Embeddings_embedQuery'
-  }
+  static id = 'langchain_embeddings_embed_query'
+  static lcType = 'embedding'
+  static prefix = 'tracing:apm:@langchain/core:Embeddings_embedQuery'
 }
 
 class EmbeddingsEmbedDocumentsPlugin extends BaseLangChainTracingPlugin {
-  static get id () { return 'langchain_embeddings_embed_documents' }
-  static get lcType () { return 'embedding' }
-  static get prefix () {
-    return 'tracing:apm:@langchain/core:Embeddings_embedDocuments'
-  }
+  static id = 'langchain_embeddings_embed_documents'
+  static lcType = 'embedding'
+  static prefix = 'tracing:apm:@langchain/core:Embeddings_embedDocuments'
 }
 
 class ToolInvokePlugin extends BaseLangChainTracingPlugin {
-  static get id () { return 'langchain_tool_invoke' }
-  static get lcType () { return 'tool' }
-  static get prefix () {
-    return 'tracing:orchestrion:@langchain/core:Tool_invoke'
-  }
+  static id = 'langchain_tool_invoke'
+  static lcType = 'tool'
+  static prefix = 'tracing:orchestrion:@langchain/core:Tool_invoke'
 }
 
 class VectorStoreSimilaritySearchPlugin extends BaseLangChainTracingPlugin {
-  static get id () { return 'langchain_vectorstore_similarity_search' }
-  static get lcType () { return 'similarity_search' }
-  static get prefix () {
-    return 'tracing:orchestrion:@langchain/core:VectorStore_similaritySearch'
-  }
+  static id = 'langchain_vectorstore_similarity_search'
+  static lcType = 'similarity_search'
+  static prefix = 'tracing:orchestrion:@langchain/core:VectorStore_similaritySearch'
 }
 
 class VectorStoreSimilaritySearchWithScorePlugin extends BaseLangChainTracingPlugin {
-  static get id () { return 'langchain_vectorstore_similarity_search_with_score' }
-  static get lcType () { return 'similarity_search' }
-  static get prefix () {
-    return 'tracing:orchestrion:@langchain/core:VectorStore_similaritySearchWithScore'
-  }
+  static id = 'langchain_vectorstore_similarity_search_with_score'
+  static lcType = 'similarity_search'
+  static prefix = 'tracing:orchestrion:@langchain/core:VectorStore_similaritySearchWithScore'
 }
 
 module.exports = [

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

@@ -4,8 +4,8 @@ const { storage } = require('../../datadog-core')
 const MySQLPlugin = require('../../datadog-plugin-mysql/src')
 
 class MariadbPlugin extends MySQLPlugin {
-  static get id () { return 'mariadb' }
-  static get system () { return 'mariadb' }
+  static id = 'mariadb'
+  static system = 'mariadb'
 
   constructor (...args) {
     super(...args)

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

@@ -4,7 +4,7 @@ const { CLIENT_PORT_KEY } = require('../../dd-trace/src/constants')
 const CachePlugin = require('../../dd-trace/src/plugins/cache')
 
 class MemcachedPlugin extends CachePlugin {
-  static get id () { return 'memcached' }
+  static id = 'memcached'
 
   bindStart (ctx) {
     const { client, server, query } = ctx

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

@@ -4,9 +4,7 @@ const RouterPlugin = require('../../datadog-plugin-router/src')
 const web = require('../../dd-trace/src/plugins/util/web')
 
 class MicrogatewayCorePlugin extends RouterPlugin {
-  static get id () {
-    return 'microgateway-core'
-  }
+  static id = 'microgateway-core'
 
   constructor (...args) {
     super(...args)
@@ -19,7 +17,9 @@ class MicrogatewayCorePlugin extends RouterPlugin {
       web.setRoute(req, route)
     })
 
-    this.addSub('apm:microgateway-core:request:error', this.addError)
+    this.addSub('apm:microgateway-core:request:error', ({ error }) => {
+      this.addError(error)
+    })
   }
 }
 

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

@@ -74,9 +74,7 @@ function getTestSuiteLevelVisibilityTags (testSuiteSpan) {
 }
 
 class MochaPlugin extends CiPlugin {
-  static get id () {
-    return 'mocha'
-  }
+  static id = 'mocha'
 
   constructor (...args) {
     super(...args)

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

@@ -4,8 +4,8 @@ const ClientPlugin = require('../../dd-trace/src/plugins/client')
 const { moleculerTags } = require('./util')
 
 class MoleculerClientPlugin extends ClientPlugin {
-  static get id () { return 'moleculer' }
-  static get operation () { return 'call' }
+  static id = 'moleculer'
+  static operation = 'call'
 
   bindStart (ctx) {
     const { actionName, opts } = ctx

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

@@ -7,7 +7,7 @@ const MoleculerClientPlugin = require('./client')
 const CompositePlugin = require('../../dd-trace/src/plugins/composite')
 
 class MoleculerPlugin extends CompositePlugin {
-  static get id () { return 'moleculer' }
+  static id = 'moleculer'
   static get plugins () {
     return {
       server: MoleculerServerPlugin,

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

@@ -4,8 +4,8 @@ const ServerPlugin = require('../../dd-trace/src/plugins/server')
 const { moleculerTags } = require('./util')
 
 class MoleculerServerPlugin extends ServerPlugin {
-  static get id () { return 'moleculer' }
-  static get operation () { return 'action' }
+  static id = 'moleculer'
+  static operation = 'action'
 
   bindStart (ctx) {
     const { action, middlewareCtx, broker } = ctx

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

@@ -6,11 +6,11 @@ const coalesce = require('koalas')
 const { getEnvironmentVariable } = require('../../dd-trace/src/config-helper')
 
 class MongodbCorePlugin extends DatabasePlugin {
-  static get id () { return 'mongodb-core' }
-  static get component () { return 'mongodb' }
+  static id = 'mongodb-core'
+  static component = 'mongodb'
   // avoid using db.name for peer.service since it includes the collection name
   // should be removed if one day this will be fixed
-  static get peerServicePrecursors () { return [] }
+  static peerServicePrecursors = []
 
   configure (config) {
     super.configure(config)
@@ -24,7 +24,9 @@ class MongodbCorePlugin extends DatabasePlugin {
     )
   }
 
-  start ({ ns, ops, options = {}, name }) {
+  bindStart (ctx) {
+    const { ns, ops, options = {}, name } = ctx
+
     // heartbeat commands can be disabled if this.config.heartbeatEnabled is false
     if (!this.config.heartbeatEnabled && isHeartbeat(ops, this.config)) {
       return
@@ -44,11 +46,13 @@ class MongodbCorePlugin extends DatabasePlugin {
         'out.host': options.host,
         'out.port': options.port
       }
-    })
+    }, ctx)
     const comment = this.injectDbmComment(span, ops.comment, service)
     if (comment) {
       ops.comment = comment
     }
+
+    return ctx.currentStore
   }
 
   getPeerService (tags) {

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

@@ -0,0 +1,20 @@
+'use strict'
+
+const DatabasePlugin = require('../../dd-trace/src/plugins/database')
+const { storage } = require('../../datadog-core')
+
+class MongoosePlugin extends DatabasePlugin {
+  static id = 'mongoose'
+  static operation = 'exec'
+
+  bindStart (ctx) {
+    ctx.parentStore = storage('legacy').getStore()
+    return ctx.parentStore
+  }
+
+  bindFinish (ctx) {
+    return ctx.parentStore
+  }
+}
+
+module.exports = MongoosePlugin

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

@@ -5,8 +5,8 @@ const CLIENT_PORT_KEY = require('../../dd-trace/src/constants')
 const DatabasePlugin = require('../../dd-trace/src/plugins/database')
 
 class MySQLPlugin extends DatabasePlugin {
-  static get id () { return 'mysql' }
-  static get system () { return 'mysql' }
+  static id = 'mysql'
+  static system = 'mysql'
 
   constructor () {
     super(...arguments)

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

@@ -4,7 +4,7 @@ const { storage } = require('../../datadog-core')
 const MySQLPlugin = require('../../datadog-plugin-mysql/src')
 
 class MySQL2Plugin extends MySQLPlugin {
-  static get id () { return 'mysql2' }
+  static id = 'mysql2'
 
   constructor () {
     super(...arguments)

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

@@ -5,7 +5,7 @@ const NetIPCPlugin = require('./ipc')
 const NetTCPPlugin = require('./tcp')
 
 class NetPlugin extends Plugin {
-  static get id () { return 'net' }
+  static id = 'net'
 
   constructor (...args) {
     super(...args)

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

@@ -3,8 +3,8 @@
 const ClientPlugin = require('../../dd-trace/src/plugins/client')
 
 class NetIPCPlugin extends ClientPlugin {
-  static get id () { return 'net' }
-  static get operation () { return 'ipc' }
+  static id = 'net'
+  static operation = 'ipc'
 
   bindStart (ctx) {
     this.startSpan('ipc.connect', {

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

@@ -4,8 +4,8 @@ const { CLIENT_PORT_KEY } = require('../../dd-trace/src/constants')
 const ClientPlugin = require('../../dd-trace/src/plugins/client')
 
 class NetTCPPlugin extends ClientPlugin {
-  static get id () { return 'net' }
-  static get operation () { return 'tcp' }
+  static id = 'net'
+  static operation = 'tcp'
 
   constructor (...args) {
     super(...args)

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

@@ -9,9 +9,7 @@ const web = require('../../dd-trace/src/plugins/util/web')
 const errorPages = new Set(['/404', '/500', '/_error', '/_not-found', '/_not-found/page'])
 
 class NextPlugin extends ServerPlugin {
-  static get id () {
-    return 'next'
-  }
+  static id = 'next'
 
   constructor (...args) {
     super(...args)

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

@@ -3,9 +3,7 @@
 const CiPlugin = require('../../dd-trace/src/plugins/ci_plugin')
 
 class NycPlugin extends CiPlugin {
-  static get id () {
-    return 'nyc'
-  }
+  static id = 'nyc'
 
   constructor (...args) {
     super(...args)

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

@@ -5,7 +5,7 @@ const OpenAiTracingPlugin = require('./tracing')
 const OpenAiLLMObsPlugin = require('../../dd-trace/src/llmobs/plugins/openai')
 
 class OpenAiPlugin extends CompositePlugin {
-  static get id () { return 'openai' }
+  static id = 'openai'
   static get plugins () {
     return {
       llmobs: OpenAiLLMObsPlugin,