dd-trace/packages/datadog-plugin-koa/src/index.js

Datadog APM tracing client for JavaScript

'use strict'

const RouterPlugin = require('../../datadog-plugin-router/src')
const web = require('../../dd-trace/src/plugins/util/web')

class KoaPlugin extends RouterPlugin {
  static id = 'koa'

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

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

    this.addSub('apm:koa:request:route', ({ req, route }) => {
      web.setRoute(req, route)
    })
  }
}

module.exports = KoaPlugin