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

Datadog APM tracing client for JavaScript

'use strict'

require('./mongodb-core')

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

// collection methods with filter
const collectionMethodsWithFilter = [
  'count',
  'countDocuments',
  'deleteMany',
  'deleteOne',
  'find',
  'findOneAndDelete',
  'findOneAndReplace',
  'replaceOne'
] // findOne is ignored because it calls to find

const collectionMethodsWithTwoFilters = [
  'findOneAndUpdate',
  'updateMany',
  'updateOne'
]

const startCh = channel('datadog:mongodb:collection:filter:start')

addHook({ name: 'mongodb', versions: ['>=3.3 <5', '5', '>=6'] }, mongodb => {
  [...collectionMethodsWithFilter, ...collectionMethodsWithTwoFilters].forEach(methodName => {
    if (!(methodName in mongodb.Collection.prototype)) return

    const useTwoArguments = collectionMethodsWithTwoFilters.includes(methodName)

    shimmer.wrap(mongodb.Collection.prototype, methodName, method => {
      return function () {
        if (!startCh.hasSubscribers) {
          return method.apply(this, arguments)
        }

        const ctx = {
          filters: [arguments[0]],
          methodName
        }

        if (useTwoArguments) {
          ctx.filters.push(arguments[1])
        }

        return startCh.runStores(ctx, () => {
          return method.apply(this, arguments)
        })
      }
    })
  })
  return mongodb
})