atproto/services/pds/tracer.js

/* eslint-env node */

'use strict'

const { registerInstrumentations } = require('@opentelemetry/instrumentation')

const {
  BetterSqlite3Instrumentation,
} = require('opentelemetry-plugin-better-sqlite3')

const { TracerProvider } = require('dd-trace') // Only works with commonjs
  .init({ logInjection: true })
  .use('express', {
    hooks: { request: maintainXrpcResource },
  })

const tracer = new TracerProvider()
tracer.register()

registerInstrumentations({
  tracerProvider: tracer,
  instrumentations: [new BetterSqlite3Instrumentation()],
})

const path = require('path')

function maintainXrpcResource(span, req) {
  // Show actual xrpc method as resource rather than the route pattern
  if (span && req.originalUrl?.startsWith('/xrpc/')) {
    span.setTag(
      'resource.name',
      [
        req.method,
        path.posix.join(req.baseUrl || '', req.path || '', '/').slice(0, -1), // Ensures no trailing slash
      ]
        .filter(Boolean)
        .join(' '),
    )
  }
}