dd-trace/packages/datadog-plugin-limitd-client/src/index.js

Datadog APM tracing client for JavaScript

'use strict'

function createWrapRequest (tracer) {
  const scope = tracer.scope()

  return function wrapRequest (original) {
    return function requestWithTrace (request, callback) {
      const index = arguments.length - 1

      arguments[index] = scope.bind(arguments[index])

      return original.apply(this, arguments)
    }
  }
}

module.exports = [
  {
    name: 'limitd-client',
    versions: ['>=2.8'],
    patch (LimitdClient, tracer) {
      this.wrap(LimitdClient.prototype, '_directRequest', createWrapRequest(tracer))
      this.wrap(LimitdClient.prototype, '_retriedRequest', createWrapRequest(tracer))
    },
    unpatch (LimitdClient) {
      this.unwrap(LimitdClient.prototype, '_directRequest')
      this.unwrap(LimitdClient.prototype, '_retriedRequest')
    }
  }
]