dd-trace 2.0.1 → 2.1.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dd-trace",
-  "version": "2.0.1",
+  "version": "2.1.0",
   "description": "Datadog APM tracing client for JavaScript",
   "main": "index.js",
   "typings": "index.d.ts",

package/packages/datadog-instrumentations/index.js

@@ -3,6 +3,7 @@
 require('./src/dns')
 require('./src/memcached')
 require('./src/mysql')
+require('./src/mysql2')
 require('./src/bluebird')
 require('./src/when')
 require('./src/promise')

package/packages/datadog-instrumentations/src/mysql2.js

@@ -0,0 +1,76 @@
+'use strict'
+
+const {
+  channel,
+  addHook,
+  AsyncResource
+} = require('./helpers/instrument')
+const shimmer = require('../../datadog-shimmer')
+
+addHook({ name: 'mysql2', file: 'lib/connection.js', versions: ['>=1'] }, Connection => {
+  const startCh = channel('apm:mysql:query:start')
+  const asyncEndCh = channel('apm:mysql:query:async-end')
+  const endCh = channel('apm:mysql:query:end')
+  const errorCh = channel('apm:mysql:query:error')
+
+  shimmer.wrap(Connection.prototype, 'addCommand', addCommand => function (cmd) {
+    if (!startCh.hasSubscribers) return addCommand.apply(this, arguments)
+
+    const asyncResource = new AsyncResource('bound-anonymous-fn')
+    const name = cmd && cmd.constructor && cmd.constructor.name
+    const isCommand = typeof cmd.execute === 'function'
+    const isQuery = isCommand && (name === 'Execute' || name === 'Query')
+
+    // TODO: consider supporting all commands and not just queries
+    cmd.execute = isQuery
+      ? wrapExecute(cmd, cmd.execute, asyncResource, this.config)
+      : bindExecute(cmd, cmd.execute, asyncResource)
+
+    return asyncResource.bind(addCommand, this).apply(this, arguments)
+  })
+
+  return Connection
+
+  function bindExecute (cmd, execute, asyncResource) {
+    return asyncResource.bind(function executeWithTrace (packet, connection) {
+      if (this.onResult) {
+        this.onResult = asyncResource.bind(this.onResult)
+      }
+
+      return execute.apply(this, arguments)
+    }, cmd)
+  }
+
+  function wrapExecute (cmd, execute, asyncResource, config) {
+    return asyncResource.bind(function executeWithTrace (packet, connection) {
+      const sql = cmd.statement ? cmd.statement.query : cmd.sql
+
+      startCh.publish([sql, config])
+
+      if (this.onResult) {
+        const onResult = asyncResource.bind(this.onResult)
+
+        this.onResult = AsyncResource.bind(function (error) {
+          if (error) {
+            errorCh.publish(error)
+          }
+          asyncEndCh.publish(undefined)
+          onResult.apply(this, arguments)
+        }, 'bound-anonymous-fn', this)
+      } else {
+        this.on('error', AsyncResource.bind(error => errorCh.publish(error)))
+        this.on('end', AsyncResource.bind(() => asyncEndCh.publish(undefined)))
+      }
+
+      this.execute = execute
+
+      try {
+        return execute.apply(this, arguments)
+      } catch (err) {
+        errorCh.publish(err)
+      } finally {
+        endCh.publish(undefined)
+      }
+    }, cmd)
+  }
+})

package/packages/datadog-plugin-aws-sdk/src/helpers.js

@@ -10,7 +10,8 @@ const services = {
   s3: getService(require('./services/s3')),
   redshift: getService(require('./services/redshift')),
   sns: getService(require('./services/sns')),
-  sqs: getService(require('./services/sqs'))
+  sqs: getService(require('./services/sqs')),
+  eventbridge: getService(require('./services/eventbridge'))
 }
 
 function getService (Service) {
@@ -78,8 +79,8 @@ const helpers = {
   requestInject (span, request, serviceName, tracer) {
     if (!span) return
 
-    const inject = services[serviceName] && services[serviceName].requestInject
-    if (inject) inject(span, request, tracer)
+    const service = services[serviceName] && services[serviceName]
+    if (service.requestInject) service.requestInject(span, request, tracer)
   },
 
   wrapCb (cb, serviceName, tags, request, tracer, childOf) {

package/packages/datadog-plugin-aws-sdk/src/services/eventbridge.js

@@ -0,0 +1,48 @@
+'use strict'
+const log = require('../../../dd-trace/src/log')
+class EventBridge {
+  generateTags (params, operation, response) {
+    if (!params || !params.source) return {}
+
+    return {
+      'resource.name': `${operation} ${params.source}`,
+      'aws.eventbridge.source': params.source
+    }
+  }
+
+  /**
+   * requestInject
+   * @param {*} span
+   * @param {*} request
+   * @param {*} tracer
+   *
+   * Docs: https://docs.aws.amazon.com/eventbridge/latest/APIReference/API_PutEventsRequestEntry.html
+   * We cannot use the traceHeader field as that's reserved for X-Ray.
+   * Detail must be a valid JSON string
+   * Max size per event is 256kb (https://docs.aws.amazon.com/eventbridge/latest/userguide/eb-putevent-size.html)
+   */
+  requestInject (span, request, tracer) {
+    const operation = request.operation
+    if (operation === 'putEvents' &&
+      request.params &&
+      request.params.Entries &&
+      request.params.Entries.length > 0 &&
+      request.params.Entries[0].Detail) {
+      try {
+        const details = JSON.parse(request.params.Entries[0].Detail)
+        details._datadog = {}
+        tracer.inject(span, 'text_map', details._datadog)
+        const finalData = JSON.stringify(details)
+        const byteSize = Buffer.byteLength(finalData)
+        if (byteSize >= (1024 * 256)) {
+          log.info('Payload size too large to pass context')
+          return
+        }
+        request.params.Entries[0].Detail = finalData
+      } catch (e) {
+        log.error(e)
+      }
+    }
+  }
+}
+module.exports = EventBridge

package/packages/datadog-plugin-aws-sdk/src/services/kinesis.js

@@ -1,15 +1,65 @@
 'use strict'
-
+const log = require('../../../dd-trace/src/log')
 class Kinesis {
   generateTags (params, operation, response) {
-    const tags = {}
-
-    if (!params || !params.StreamName) return tags
+    if (!params || !params.StreamName) return {}
 
-    return Object.assign(tags, {
+    return {
       'resource.name': `${operation} ${params.StreamName}`,
       'aws.kinesis.stream_name': params.StreamName
-    })
+    }
+  }
+
+  // AWS-SDK will b64 kinesis payloads
+  // or will accept an already b64 encoded payload
+  // This method handles both
+  _tryParse (body) {
+    try {
+      return JSON.parse(body)
+    } catch (e) {
+      log.info('Not JSON string. Trying Base64 encoded JSON string')
+    }
+    try {
+      return JSON.parse(Buffer.from(body, 'base64').toString('ascii'), true)
+    } catch (e) {
+      return null
+    }
+  }
+
+  requestInject (span, request, tracer) {
+    const operation = request.operation
+    if (operation === 'putRecord' || operation === 'putRecords') {
+      if (!request.params) {
+        return
+      }
+
+      const traceData = {}
+      tracer.inject(span, 'text_map', traceData)
+      let injectPath
+      if (request.params.Records && request.params.Records.length > 0) {
+        injectPath = request.params.Records[0]
+      } else if (request.params.Data) {
+        injectPath = request.params
+      } else {
+        log.error('No valid payload passed, unable to pass trace context')
+        return
+      }
+      const parsedData = this._tryParse(injectPath.Data)
+      if (parsedData) {
+        parsedData._datadog = traceData
+        const finalData = JSON.stringify(parsedData)
+        const byteSize = Buffer.byteLength(finalData, 'ascii')
+        // Kinesis max payload size is 1MB
+        // So we must ensure adding DD context won't go over that (512b is an estimate)
+        if (byteSize >= 1048576) {
+          log.info('Payload size too large to pass context')
+          return
+        }
+        injectPath.Data = finalData
+      } else {
+        log.error('Unable to parse payload, unable to pass trace context')
+      }
+    }
   }
 }
 

package/packages/datadog-plugin-aws-sdk/src/services/sns.js

@@ -1,21 +1,48 @@
 'use strict'
+const log = require('../../../dd-trace/src/log')
 
 class Sns {
   generateTags (params, operation, response) {
-    const tags = {}
+    if (!params) return {}
 
-    if (!params) return tags
+    if (!params.TopicArn && !(response.data && response.data.TopicArn)) return {}
 
-    if (!params.TopicArn && !(response.data && response.data.TopicArn)) return tags
-
-    return Object.assign(tags, {
+    return {
       'resource.name': `${operation} ${params.TopicArn || response.data.TopicArn}`,
       'aws.sns.topic_arn': params.TopicArn || response.data.TopicArn
-    })
+    }
 
     // TODO: should arn be sanitized or quantized in some way here,
     // for example if it contains a phone number?
   }
+
+  requestInject (span, request, tracer) {
+    const operation = request.operation
+    if (operation === 'publish' || operation === 'publishBatch') {
+      if (!request.params) {
+        request.params = {}
+      }
+      let injectPath
+      if (request.params.PublishBatchRequestEntries && request.params.PublishBatchRequestEntries.length > 0) {
+        injectPath = request.params.PublishBatchRequestEntries[0]
+      } else if (request.params.Message) {
+        injectPath = request.params
+      }
+      if (!injectPath.MessageAttributes) {
+        injectPath.MessageAttributes = {}
+      }
+      if (Object.keys(injectPath.MessageAttributes).length >= 10) { // SNS quota
+        log.info('Message attributes full, skipping trace context injection')
+        return
+      }
+      const ddInfo = {}
+      tracer.inject(span, 'text_map', ddInfo)
+      injectPath.MessageAttributes._datadog = {
+        DataType: 'String',
+        StringValue: JSON.stringify(ddInfo)
+      }
+    }
+  }
 }
 
 module.exports = Sns

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

@@ -4,6 +4,7 @@ const Plugin = require('../../dd-trace/src/plugins/plugin')
 const { storage } = require('../../datadog-core')
 const analyticsSampler = require('../../dd-trace/src/analytics_sampler')
 
+// This plugin supports both mysql and mysql2
 class MySQLPlugin extends Plugin {
   static get name () {
     return 'mysql'

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

@@ -115,10 +115,10 @@ function trace (tracer, config, req, res, handler) {
   // TODO: Use CLS when it will be available in core.
   span._nextReq = req
 
-  promise.then(() => finish(span, config, req, res), err => {
-    span.setTag('error', err)
-    finish(span, config, req, res)
-  })
+  promise.then(
+    () => finish(span, config, req, res),
+    err => finish(span, config, req, res, err)
+  )
 
   return promise
 }
@@ -132,7 +132,8 @@ function addPage (req, page) {
   })
 }
 
-function finish (span, config, req, res) {
+function finish (span, config, req, res, err) {
+  span.setTag('error', err || !config.validateStatus(res.statusCode))
   span.addTags({
     'http.status_code': res.statusCode
   })
@@ -142,8 +143,11 @@ function finish (span, config, req, res) {
 
 function normalizeConfig (config) {
   const hooks = getHooks(config)
+  const validateStatus = typeof config.validateStatus === 'function'
+    ? config.validateStatus
+    : code => code < 500
 
-  return Object.assign({}, config, { hooks })
+  return Object.assign({}, config, { hooks, validateStatus })
 }
 
 function getHooks (config) {

package/packages/dd-trace/lib/version.js

@@ -1 +1 @@
-module.exports = '2.0.1'
+module.exports = '2.1.0'

package/packages/dd-trace/src/plugins/index.js

@@ -35,7 +35,7 @@ module.exports = {
   'mongodb-core': require('../../../datadog-plugin-mongodb-core/src'),
   'mongoose': require('../../../datadog-plugin-mongoose/src'),
   'mysql': require('../../../datadog-plugin-mysql/src'),
-  'mysql2': require('../../../datadog-plugin-mysql2/src'),
+  'mysql2': require('../../../datadog-plugin-mysql/src'),
   'net': require('../../../datadog-plugin-net/src'),
   'next': require('../../../datadog-plugin-next/src'),
   'oracledb': require('../../../datadog-plugin-oracledb/src'),

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

@@ -1,94 +0,0 @@
-'use strict'
-
-const Tags = require('opentracing').Tags
-const analyticsSampler = require('../../dd-trace/src/analytics_sampler')
-
-function createWrapAddCommand (tracer, config) {
-  return function wrapAddCommand (addCommand) {
-    return function addCommandWithTrace (cmd) {
-      const name = cmd && cmd.constructor && cmd.constructor.name
-      const isCommand = typeof cmd.execute === 'function'
-      const isSupported = name === 'Execute' || name === 'Query'
-
-      if (isCommand && isSupported) {
-        cmd.execute = wrapExecute(tracer, config, cmd.execute)
-      }
-
-      return addCommand.apply(this, arguments)
-    }
-  }
-}
-
-function wrapExecute (tracer, config, execute) {
-  const scope = tracer.scope()
-  const childOf = scope.active()
-
-  return function executeWithTrace (packet, connection) {
-    const connectionConfig = (connection && connection.config) || {}
-    const sql = this.statement ? this.statement.query : this.sql
-    const span = tracer.startSpan('mysql.query', {
-      childOf,
-      tags: {
-        [Tags.SPAN_KIND]: Tags.SPAN_KIND_RPC_CLIENT,
-        'service.name': config.service || `${tracer._service}-mysql`,
-        'resource.name': sql,
-        'span.type': 'sql',
-        'span.kind': 'client',
-        'db.type': 'mysql',
-        'db.user': connectionConfig.user,
-        'db.name': connectionConfig.database,
-        'out.host': connectionConfig.host,
-        'out.port': connectionConfig.port
-      }
-    })
-
-    analyticsSampler.sample(span, config.measured)
-
-    if (typeof this.onResult === 'function') {
-      this.onResult = wrapCallback(tracer, span, childOf, this.onResult)
-    } else {
-      this.on('error', error => span.addTags({ error }))
-      this.on('end', () => span.finish())
-    }
-
-    this.execute = execute
-
-    return scope.bind(execute, span).apply(this, arguments)
-  }
-}
-
-function wrapCallback (tracer, span, parent, done) {
-  return tracer.scope().bind((...args) => {
-    const [ error ] = args
-    span.addTags({ error })
-
-    span.finish()
-
-    done(...args)
-  }, parent)
-}
-
-module.exports = [
-  {
-    name: 'mysql2',
-    file: 'lib/connection.js',
-    versions: ['>=1'],
-    patch (Connection, tracer, config) {
-      this.wrap(Connection.prototype, 'addCommand', createWrapAddCommand(tracer, config))
-    },
-    unpatch (Connection) {
-      this.unwrap(Connection.prototype, 'addCommand')
-    }
-  },
-  {
-    name: 'mysql2',
-    file: 'lib/commands/command.js',
-    versions: ['>=1'],
-    patch (Command, tracer, config) {
-      tracer.scope().bind(Command.prototype)
-    },
-    unpatch (Command, tracer) {
-      tracer.scope().unbind(Command.prototype)
-    }
-  }
-]