dd-trace/packages/dd-trace/src/scope.js

Datadog APM tracing client for JavaScript

'use strict'

const { storage } = require('../../datadog-core')

const legacyStorage = storage('legacy')

// TODO: refactor bind to use shimmer once the new internal tracer lands
class Scope {
  active () {
    const store = legacyStorage.getStore()

    return store?.span ?? null
  }

  activate (span, callback) {
    if (typeof callback !== 'function') return callback

    const oldStore = legacyStorage.getStore()
    const newStore = span ? legacyStorage.getStore(span._store) : oldStore

    legacyStorage.enterWith({ ...newStore, span })

    try {
      return callback()
    } catch (e) {
      if (span && typeof span.setTag === 'function') {
        span.setTag('error', e)
      }

      throw e
    } finally {
      legacyStorage.enterWith(oldStore)
    }
  }

  bind (fn, span) {
    if (typeof fn !== 'function') return fn

    const scope = this
    const spanOrActive = this._spanOrActive(span)

    return function (...args) {
      return scope.activate(spanOrActive, () => {
        return fn.apply(this, args)
      })
    }
  }

  _spanOrActive (span) {
    return span === undefined ? this.active() : span
  }

  _isPromise (promise) {
    return promise && typeof promise.then === 'function'
  }
}

module.exports = Scope