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

Datadog APM tracing client for JavaScript

'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