dd-trace/packages/dd-trace/src/opentelemetry/tracer_provider.js

Datadog APM tracing client for JavaScript

'use strict'

const { trace, context, propagation } = require('@opentelemetry/api')
const { W3CTraceContextPropagator } = require('@opentelemetry/core')

const tracer = require('../../')

const ContextManager = require('./context_manager')
const { MultiSpanProcessor, NoopSpanProcessor } = require('./span_processor')
const Tracer = require('./tracer')

class TracerProvider {
  constructor (config = {}) {
    this.config = config
    this.resource = config.resource

    this._processors = []
    this._tracers = new Map()
    this._activeProcessor = new NoopSpanProcessor()
    this._contextManager = new ContextManager()
  }

  getTracer (name = 'opentelemetry', version = '0.0.0', options) {
    const key = `${name}@${version}`
    if (!this._tracers.has(key)) {
      this._tracers.set(key, new Tracer(
        { ...options, name, version },
        this.config,
        this
      ))
    }
    return this._tracers.get(key)
  }

  addSpanProcessor (spanProcessor) {
    if (!this._processors.length) {
      this._activeProcessor.shutdown()
    }
    this._processors.push(spanProcessor)
    this._activeProcessor = new MultiSpanProcessor(
      this._processors
    )
  }

  getActiveSpanProcessor () {
    return this._activeProcessor
  }

  // Not actually required by the SDK spec, but the official Node.js SDK does
  // this and the docs reflect that so we should do this too for familiarity.
  register (config = {}) {
    context.setGlobalContextManager(this._contextManager)
    if (!trace.setGlobalTracerProvider(this)) {
      trace.getTracerProvider().setDelegate(this)
    }
    // The default propagator used is the W3C Trace Context propagator, users should be able to pass in others
    // as needed
    if (config.propagator) {
      propagation.setGlobalPropagator(config.propagator)
    } else {
      propagation.setGlobalPropagator(new W3CTraceContextPropagator())
    }
  }

  forceFlush () {
    const exporter = tracer._tracer._exporter
    if (!exporter) {
      return Promise.reject(new Error('Not started'))
    }

    exporter._writer.flush()
    return this._activeProcessor.forceFlush()
  }

  shutdown () {
    return this._activeProcessor.shutdown()
  }
}

module.exports = TracerProvider