dd-trace/packages/dd-trace/src/profiling/exporter_cli.js

Datadog APM tracing client for JavaScript

'use strict'

const { AgentExporter } = require('./exporters/agent')
const { FileExporter } = require('./exporters/file')

const { SourceMapper, heap, encode } = require('@datadog/pprof')
const { ConsoleLogger } = require('./loggers/console')
const { tagger } = require('./tagger')
const fs = require('fs')
const { fileURLToPath } = require('url')

const logger = new ConsoleLogger()
const timeoutMs = 10 * 1000

function exporterFromURL (url) {
  if (url.protocol === 'file:') {
    return new FileExporter({ pprofPrefix: fileURLToPath(url) })
  } else {
    const injectionEnabled = (process.env.DD_INJECTION_ENABLED || '').split(',')
    const libraryInjected = injectionEnabled.length > 0
    const profilingEnabled = (process.env.DD_PROFILING_ENABLED || '').toLowerCase()
    const activation = ['true', '1'].includes(profilingEnabled)
      ? 'manual'
      : profilingEnabled === 'auto'
        ? 'auto'
        : injectionEnabled.includes('profiling')
          ? 'injection'
          : 'unknown'
    return new AgentExporter({
      url,
      logger,
      uploadTimeout: timeoutMs,
      libraryInjected,
      activation
    })
  }
}

async function exportProfile (urls, tags, profileType, profile) {
  let mapper
  try {
    mapper = await SourceMapper.create([process.cwd()])
  } catch (err) {
    logger.error(err)
  }

  const encodedProfile = await encode(heap.convertProfile(profile, undefined, mapper))
  const start = new Date()
  for (const url of urls) {
    const exporter = exporterFromURL(url)

    await exporter.export({
      profiles: {
        [profileType]: encodedProfile
      },
      start,
      end: start,
      tags
    })
  }
}

/** Expected command line arguments are:
* - Comma separated list of URLs (eg. "http://127.0.0.1:8126/,file:///tmp/foo.pprof")
* - Tags (eg. "service:nodejs_oom_test,version:1.0.0")
* - Profiletype (eg. space,wall,cpu)
* - JSON profile filepath
**/
const urls = process.argv[2].split(',').map(s => new URL(s))
const tags = tagger.parse(process.argv[3])
const profileType = process.argv[4]
const profile = JSON.parse(fs.readFileSync(process.argv[5]))

exportProfile(urls, tags, profileType, profile)