dd-trace/packages/datadog-instrumentations/src/zlib.js

Datadog APM tracing client for JavaScript

'use strict'

const shimmer = require('../../datadog-shimmer')
const { addHook } = require('./helpers/instrument')
const { createCallbackInstrumentor } = require('./helpers/callback-instrumentor')

const asyncMethods = [
  'brotliCompress',
  'brotliDecompress',
  'deflate',
  'deflateRaw',
  'gunzip',
  'gzip',
  'inflate',
  'inflateRaw',
  'unzip',
  'zstdCompress', // Node 22.15+ / 23.8+ / 24+
  'zstdDecompress', // Node 22.15+ / 23.8+ / 24+
]

addHook({ name: 'zlib' }, zlib => {
  const instrument = createCallbackInstrumentor('apm:zlib:operation')
  for (const method of asyncMethods) {
    if (typeof zlib[method] === 'function') {
      shimmer.wrap(zlib, method, instrument(() => ({ operation: method })))
    }
  }
  return zlib
})