dd-trace/packages/datadog-plugin-fastify/src/tracing.js

Datadog APM tracing client for JavaScript

'use strict'

const RouterPlugin = require('../../datadog-plugin-router/src')
const { storage } = require('../../datadog-core')

class FastifyTracingPlugin extends RouterPlugin {
  static id = 'fastify'

  constructor (...args) {
    super(...args)

    this.addSub('apm:fastify:request:handle', ({ req }) => {
      this.setFramework(req, 'fastify', this.config)
    })

    this.addBind('datadog:fastify:pre-parsing:start', getParentStore)
    this.addBind('datadog:fastify:pre-validation:start', getParentStore)

    this.addSub('datadog:fastify:pre-parsing:finish', (ctx) => {
      return ctx.parentStore
    })
    this.addSub('datadog:fastify:pre-validation:finish', (ctx) => {
      return ctx.parentStore
    })
    this.addSub('datadog:fastify:callback:execute', getParentStore)
  }
}

function getParentStore (ctx) {
  ctx.parentStore = ctx.parentStore ?? storage('legacy').getStore()
  return ctx.parentStore
}

module.exports = FastifyTracingPlugin