dd-trace/packages/datadog-instrumentations/src/limitd-client.js

Datadog APM tracing client for JavaScript

'use strict'

const shimmer = require('../../datadog-shimmer')
const { addHook, AsyncResource } = require('./helpers/instrument')

function wrapRequest (original) {
  return function () {
    const id = arguments.length - 1
    arguments[id] = AsyncResource.bind(arguments[id])
    return original.apply(this, arguments)
  }
}

addHook({
  name: 'limitd-client',
  versions: ['>=2.8'],
  file: ['client.js']
}, LimitdClient => {
  shimmer.wrap(LimitdClient.prototype, '_directRequest', wrapRequest)
  shimmer.wrap(LimitdClient.prototype, '_retriedRequest', wrapRequest)
  return LimitdClient
})