dd-trace/packages/datadog-instrumentations/src/bluebird.js

Datadog APM tracing client for JavaScript

'use strict'

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

function createGetNewLibraryCopyWrap (originalLib) {
  return function wrapGetNewLibraryCopy (getNewLibraryCopy) {
    return function getNewLibraryCopyWithTrace () {
      const libraryCopy = getNewLibraryCopy.apply(this, arguments)
      shimmer.wrap(libraryCopy.prototype, '_then', wrapThen)
      shimmer.wrap(libraryCopy, 'getNewLibraryCopy', createGetNewLibraryCopyWrap(originalLib))
      return libraryCopy
    }
  }
}

addHook({ name: 'bluebird', versions: ['>=2.0.2'] }, Promise => {
  shimmer.wrap(Promise.prototype, '_then', wrapThen)
  return Promise
})

addHook({ name: 'bluebird', versions: ['^2.11.0', '^3.4.1'] }, Promise => {
  shimmer.wrap(Promise, 'getNewLibraryCopy', createGetNewLibraryCopyWrap(Promise))
  return Promise
})