@sentry/node 10.59.0 → 10.61.0

package/build/esm/integrations/tracing/dataloader/vendored/instrumentation.js.map

@@ -1 +1 @@
-{"version":3,"file":"instrumentation.js","sources":["../../../../../../src/integrations/tracing/dataloader/vendored/instrumentation.ts"],"sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\n * SPDX-License-Identifier: Apache-2.0\n *\n * NOTICE from the Sentry authors:\n * - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-dataloader\n * - Upstream version: @opentelemetry/instrumentation-dataloader@0.35.0\n * - Minor TypeScript strictness adjustments for this repository's compiler settings\n */\n\nimport { InstrumentationBase, InstrumentationNodeModuleDefinition, isWrapped } from '@opentelemetry/instrumentation';\nimport { SpanKind } from '@opentelemetry/api';\nimport type { BatchLoadFn, DataLoader, DataLoaderConstructor } from './types';\nimport { SDK_VERSION, SEMANTIC_ATTRIBUTE_SENTRY_OP, SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, startSpan } from '@sentry/core';\n\nconst MODULE_NAME = 'dataloader';\nconst PACKAGE_NAME = '@sentry/instrumentation-dataloader';\nconst ORIGIN = 'auto.db.otel.dataloader';\n\ntype LoadFn = DataLoader['load'];\ntype LoadManyFn = DataLoader['loadMany'];\ntype PrimeFn = DataLoader['prime'];\ntype ClearFn = DataLoader['clear'];\ntype ClearAllFn = DataLoader['clearAll'];\n\nfunction isModule(module: unknown): module is { [Symbol.toStringTag]: 'Module'; default: DataLoaderConstructor } {\n  return (module as { [Symbol.toStringTag]: string })[Symbol.toStringTag] === 'Module';\n}\n\nfunction extractModuleExports(module: unknown): DataLoaderConstructor {\n  return isModule(module)\n    ? module.default // ESM\n    : (module as DataLoaderConstructor); // CommonJS\n}\n\nfunction getSpanName(\n  dataloader: DataLoader,\n  operation: 'load' | 'loadMany' | 'batch' | 'prime' | 'clear' | 'clearAll',\n): string {\n  const dataloaderName = dataloader.name;\n  if (dataloaderName) {\n    return `${MODULE_NAME}.${operation} ${dataloaderName}`;\n  }\n\n  return `${MODULE_NAME}.${operation}`;\n}\n\n// `load`, `loadMany` and `batch` are all cache reads\nfunction getSpanOp(operation: 'load' | 'loadMany' | 'batch' | 'prime' | 'clear' | 'clearAll'): string | undefined {\n  if (operation === 'load' || operation === 'loadMany' || operation === 'batch') {\n    return 'cache.get';\n  }\n\n  return undefined;\n}\n\nexport class DataloaderInstrumentation extends InstrumentationBase {\n  constructor(config = {}) {\n    super(PACKAGE_NAME, SDK_VERSION, config);\n  }\n\n  protected init() {\n    return [\n      new InstrumentationNodeModuleDefinition(\n        MODULE_NAME,\n        ['>=2.0.0 <3'],\n        module => {\n          const dataloader = extractModuleExports(module);\n          this._patchLoad(dataloader.prototype);\n          this._patchLoadMany(dataloader.prototype);\n          this._patchPrime(dataloader.prototype);\n          this._patchClear(dataloader.prototype);\n          this._patchClearAll(dataloader.prototype);\n\n          return this._getPatchedConstructor(dataloader);\n        },\n        module => {\n          const dataloader = extractModuleExports(module);\n          ['load', 'loadMany', 'prime', 'clear', 'clearAll'].forEach(method => {\n            if (isWrapped(dataloader.prototype[method])) {\n              this._unwrap(dataloader.prototype, method);\n            }\n          });\n        },\n      ),\n    ];\n  }\n\n  private _wrapBatchLoadFn(batchLoadFn: BatchLoadFn<unknown, unknown>): BatchLoadFn<unknown, unknown> {\n    // oxlint-disable-next-line typescript/no-this-alias\n    const instrumentation = this;\n\n    return function patchedBatchLoadFn(this: DataLoader, ...args: Parameters<BatchLoadFn<unknown, unknown>>) {\n      if (!instrumentation.isEnabled()) {\n        return batchLoadFn.call(this, ...args);\n      }\n\n      return startSpan(\n        {\n          name: getSpanName(this, 'batch'),\n          links: this._batch?.spanLinks,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('batch'),\n          },\n          onlyIfParent: true,\n        },\n        () => batchLoadFn.apply(this, args),\n      );\n    };\n  }\n\n  private _getPatchedConstructor(constructor: DataLoaderConstructor): DataLoaderConstructor {\n    // oxlint-disable-next-line typescript/no-this-alias\n    const instrumentation = this;\n    const prototype = constructor.prototype;\n\n    if (!instrumentation.isEnabled()) {\n      return constructor;\n    }\n\n    // oxlint-disable-next-line typescript/no-explicit-any\n    function PatchedDataloader(this: DataLoader, ...args: any[]) {\n      // BatchLoadFn is the first constructor argument\n      // https://github.com/graphql/dataloader/blob/77c2cd7ca97e8795242018ebc212ce2487e729d2/src/index.js#L47\n      if (typeof args[0] === 'function') {\n        if (isWrapped(args[0])) {\n          instrumentation._unwrap(args, 0);\n        }\n\n        args[0] = instrumentation._wrapBatchLoadFn(args[0]);\n      }\n\n      return constructor.apply(this, args);\n    }\n\n    PatchedDataloader.prototype = prototype;\n    return PatchedDataloader as unknown as DataLoaderConstructor;\n  }\n\n  private _patchLoad(proto: DataLoader) {\n    // oxlint-disable-next-line typescript/unbound-method\n    if (isWrapped(proto.load)) {\n      this._unwrap(proto, 'load');\n    }\n\n    this._wrap(proto, 'load', this._getPatchedLoad.bind(this));\n  }\n\n  private _getPatchedLoad(original: LoadFn): LoadFn {\n    return function patchedLoad(this: DataLoader, ...args: Parameters<typeof original>) {\n      return startSpan(\n        {\n          name: getSpanName(this, 'load'),\n          kind: SpanKind.CLIENT,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('load'),\n          },\n          onlyIfParent: true,\n        },\n        span => {\n          const result = original.call(this, ...args);\n\n          if (this._batch && span.isRecording()) {\n            if (!this._batch.spanLinks) {\n              this._batch.spanLinks = [];\n            }\n\n            this._batch.spanLinks.push({ context: span.spanContext() });\n          }\n\n          return result;\n        },\n      );\n    };\n  }\n\n  private _patchLoadMany(proto: DataLoader) {\n    // oxlint-disable-next-line typescript/unbound-method\n    if (isWrapped(proto.loadMany)) {\n      this._unwrap(proto, 'loadMany');\n    }\n\n    this._wrap(proto, 'loadMany', this._getPatchedLoadMany.bind(this));\n  }\n\n  private _getPatchedLoadMany(original: LoadManyFn): LoadManyFn {\n    return function patchedLoadMany(this: DataLoader, ...args: Parameters<typeof original>) {\n      return startSpan(\n        {\n          name: getSpanName(this, 'loadMany'),\n          kind: SpanKind.CLIENT,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('loadMany'),\n          },\n          onlyIfParent: true,\n        },\n        () => original.call(this, ...args),\n      );\n    };\n  }\n\n  private _patchPrime(proto: DataLoader) {\n    // oxlint-disable-next-line typescript/unbound-method\n    if (isWrapped(proto.prime)) {\n      this._unwrap(proto, 'prime');\n    }\n\n    this._wrap(proto, 'prime', this._getPatchedPrime.bind(this));\n  }\n\n  private _getPatchedPrime(original: PrimeFn): PrimeFn {\n    return function patchedPrime(this: DataLoader, ...args: Parameters<typeof original>) {\n      return startSpan(\n        {\n          name: getSpanName(this, 'prime'),\n          kind: SpanKind.CLIENT,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('prime'),\n          },\n          onlyIfParent: true,\n        },\n        () => original.call(this, ...args),\n      );\n    };\n  }\n\n  private _patchClear(proto: DataLoader) {\n    // oxlint-disable-next-line typescript/unbound-method\n    if (isWrapped(proto.clear)) {\n      this._unwrap(proto, 'clear');\n    }\n\n    this._wrap(proto, 'clear', this._getPatchedClear.bind(this));\n  }\n\n  private _getPatchedClear(original: ClearFn): ClearFn {\n    return function patchedClear(this: DataLoader, ...args: Parameters<typeof original>) {\n      return startSpan(\n        {\n          name: getSpanName(this, 'clear'),\n          kind: SpanKind.CLIENT,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('clear'),\n          },\n          onlyIfParent: true,\n        },\n        () => original.call(this, ...args),\n      );\n    };\n  }\n\n  private _patchClearAll(proto: DataLoader) {\n    // oxlint-disable-next-line typescript/unbound-method\n    if (isWrapped(proto.clearAll)) {\n      this._unwrap(proto, 'clearAll');\n    }\n\n    this._wrap(proto, 'clearAll', this._getPatchedClearAll.bind(this));\n  }\n\n  private _getPatchedClearAll(original: ClearAllFn): ClearAllFn {\n    return function patchedClearAll(this: DataLoader, ...args: Parameters<typeof original>) {\n      return startSpan(\n        {\n          name: getSpanName(this, 'clearAll'),\n          kind: SpanKind.CLIENT,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('clearAll'),\n          },\n          onlyIfParent: true,\n        },\n        () => original.call(this, ...args),\n      );\n    };\n  }\n}\n"],"names":[],"mappings":";;;;AAeA,MAAM,WAAA,GAAc,YAAA;AACpB,MAAM,YAAA,GAAe,oCAAA;AACrB,MAAM,MAAA,GAAS,yBAAA;AAQf,SAAS,SAAS,MAAA,EAA+F;AAC/G,EAAA,OAAQ,MAAA,CAA4C,MAAA,CAAO,WAAW,CAAA,KAAM,QAAA;AAC9E;AAEA,SAAS,qBAAqB,MAAA,EAAwC;AACpE,EAAA,OAAO,QAAA,CAAS,MAAM,CAAA,GAClB,MAAA,CAAO,OAAA,GACN,MAAA;AACP;AAEA,SAAS,WAAA,CACP,YACA,SAAA,EACQ;AACR,EAAA,MAAM,iBAAiB,UAAA,CAAW,IAAA;AAClC,EAAA,IAAI,cAAA,EAAgB;AAClB,IAAA,OAAO,CAAA,EAAG,WAAW,CAAA,CAAA,EAAI,SAAS,IAAI,cAAc,CAAA,CAAA;AAAA,EACtD;AAEA,EAAA,OAAO,CAAA,EAAG,WAAW,CAAA,CAAA,EAAI,SAAS,CAAA,CAAA;AACpC;AAGA,SAAS,UAAU,SAAA,EAA+F;AAChH,EAAA,IAAI,SAAA,KAAc,MAAA,IAAU,SAAA,KAAc,UAAA,IAAc,cAAc,OAAA,EAAS;AAC7E,IAAA,OAAO,WAAA;AAAA,EACT;AAEA,EAAA,OAAO,MAAA;AACT;AAEO,MAAM,kCAAkC,mBAAA,CAAoB;AAAA,EACjE,WAAA,CAAY,MAAA,GAAS,EAAC,EAAG;AACvB,IAAA,KAAA,CAAM,YAAA,EAAc,aAAa,MAAM,CAAA;AAAA,EACzC;AAAA,EAEU,IAAA,GAAO;AACf,IAAA,OAAO;AAAA,MACL,IAAI,mCAAA;AAAA,QACF,WAAA;AAAA,QACA,CAAC,YAAY,CAAA;AAAA,QACb,CAAA,MAAA,KAAU;AACR,UAAA,MAAM,UAAA,GAAa,qBAAqB,MAAM,CAAA;AAC9C,UAAA,IAAA,CAAK,UAAA,CAAW,WAAW,SAAS,CAAA;AACpC,UAAA,IAAA,CAAK,cAAA,CAAe,WAAW,SAAS,CAAA;AACxC,UAAA,IAAA,CAAK,WAAA,CAAY,WAAW,SAAS,CAAA;AACrC,UAAA,IAAA,CAAK,WAAA,CAAY,WAAW,SAAS,CAAA;AACrC,UAAA,IAAA,CAAK,cAAA,CAAe,WAAW,SAAS,CAAA;AAExC,UAAA,OAAO,IAAA,CAAK,uBAAuB,UAAU,CAAA;AAAA,QAC/C,CAAA;AAAA,QACA,CAAA,MAAA,KAAU;AACR,UAAA,MAAM,UAAA,GAAa,qBAAqB,MAAM,CAAA;AAC9C,UAAA,CAAC,QAAQ,UAAA,EAAY,OAAA,EAAS,SAAS,UAAU,CAAA,CAAE,QAAQ,CAAA,MAAA,KAAU;AACnE,YAAA,IAAI,SAAA,CAAU,UAAA,CAAW,SAAA,CAAU,MAAM,CAAC,CAAA,EAAG;AAC3C,cAAA,IAAA,CAAK,OAAA,CAAQ,UAAA,CAAW,SAAA,EAAW,MAAM,CAAA;AAAA,YAC3C;AAAA,UACF,CAAC,CAAA;AAAA,QACH;AAAA;AACF,KACF;AAAA,EACF;AAAA,EAEQ,iBAAiB,WAAA,EAA2E;AAElG,IAAA,MAAM,eAAA,GAAkB,IAAA;AAExB,IAAA,OAAO,SAAS,sBAAwC,IAAA,EAAiD;AACvG,MAAA,IAAI,CAAC,eAAA,CAAgB,SAAA,EAAU,EAAG;AAChC,QAAA,OAAO,WAAA,CAAY,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI,CAAA;AAAA,MACvC;AAEA,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,OAAO,CAAA;AAAA,UAC/B,KAAA,EAAO,KAAK,MAAA,EAAQ,SAAA;AAAA,UACpB,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,OAAO;AAAA,WACnD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,MAAM,WAAA,CAAY,KAAA,CAAM,IAAA,EAAM,IAAI;AAAA,OACpC;AAAA,IACF,CAAA;AAAA,EACF;AAAA,EAEQ,uBAAuB,WAAA,EAA2D;AAExF,IAAA,MAAM,eAAA,GAAkB,IAAA;AACxB,IAAA,MAAM,YAAY,WAAA,CAAY,SAAA;AAE9B,IAAA,IAAI,CAAC,eAAA,CAAgB,SAAA,EAAU,EAAG;AAChC,MAAA,OAAO,WAAA;AAAA,IACT;AAGA,IAAA,SAAS,qBAAuC,IAAA,EAAa;AAG3D,MAAA,IAAI,OAAO,IAAA,CAAK,CAAC,CAAA,KAAM,UAAA,EAAY;AACjC,QAAA,IAAI,SAAA,CAAU,IAAA,CAAK,CAAC,CAAC,CAAA,EAAG;AACtB,UAAA,eAAA,CAAgB,OAAA,CAAQ,MAAM,CAAC,CAAA;AAAA,QACjC;AAEA,QAAA,IAAA,CAAK,CAAC,CAAA,GAAI,eAAA,CAAgB,gBAAA,CAAiB,IAAA,CAAK,CAAC,CAAC,CAAA;AAAA,MACpD;AAEA,MAAA,OAAO,WAAA,CAAY,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AAAA,IACrC;AAEA,IAAA,iBAAA,CAAkB,SAAA,GAAY,SAAA;AAC9B,IAAA,OAAO,iBAAA;AAAA,EACT;AAAA,EAEQ,WAAW,KAAA,EAAmB;AAEpC,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,IAAI,CAAA,EAAG;AACzB,MAAA,IAAA,CAAK,OAAA,CAAQ,OAAO,MAAM,CAAA;AAAA,IAC5B;AAEA,IAAA,IAAA,CAAK,MAAM,KAAA,EAAO,MAAA,EAAQ,KAAK,eAAA,CAAgB,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,EAC3D;AAAA,EAEQ,gBAAgB,QAAA,EAA0B;AAChD,IAAA,OAAO,SAAS,eAAiC,IAAA,EAAmC;AAClF,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,MAAM,CAAA;AAAA,UAC9B,MAAM,QAAA,CAAS,MAAA;AAAA,UACf,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,MAAM;AAAA,WAClD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,CAAA,IAAA,KAAQ;AACN,UAAA,MAAM,MAAA,GAAS,QAAA,CAAS,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI,CAAA;AAE1C,UAAA,IAAI,IAAA,CAAK,MAAA,IAAU,IAAA,CAAK,WAAA,EAAY,EAAG;AACrC,YAAA,IAAI,CAAC,IAAA,CAAK,MAAA,CAAO,SAAA,EAAW;AAC1B,cAAA,IAAA,CAAK,MAAA,CAAO,YAAY,EAAC;AAAA,YAC3B;AAEA,YAAA,IAAA,CAAK,MAAA,CAAO,UAAU,IAAA,CAAK,EAAE,SAAS,IAAA,CAAK,WAAA,IAAe,CAAA;AAAA,UAC5D;AAEA,UAAA,OAAO,MAAA;AAAA,QACT;AAAA,OACF;AAAA,IACF,CAAA;AAAA,EACF;AAAA,EAEQ,eAAe,KAAA,EAAmB;AAExC,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,QAAQ,CAAA,EAAG;AAC7B,MAAA,IAAA,CAAK,OAAA,CAAQ,OAAO,UAAU,CAAA;AAAA,IAChC;AAEA,IAAA,IAAA,CAAK,MAAM,KAAA,EAAO,UAAA,EAAY,KAAK,mBAAA,CAAoB,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,EACnE;AAAA,EAEQ,oBAAoB,QAAA,EAAkC;AAC5D,IAAA,OAAO,SAAS,mBAAqC,IAAA,EAAmC;AACtF,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,UAAU,CAAA;AAAA,UAClC,MAAM,QAAA,CAAS,MAAA;AAAA,UACf,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,UAAU;AAAA,WACtD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,MAAM,QAAA,CAAS,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI;AAAA,OACnC;AAAA,IACF,CAAA;AAAA,EACF;AAAA,EAEQ,YAAY,KAAA,EAAmB;AAErC,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,KAAK,CAAA,EAAG;AAC1B,MAAA,IAAA,CAAK,OAAA,CAAQ,OAAO,OAAO,CAAA;AAAA,IAC7B;AAEA,IAAA,IAAA,CAAK,MAAM,KAAA,EAAO,OAAA,EAAS,KAAK,gBAAA,CAAiB,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,EAC7D;AAAA,EAEQ,iBAAiB,QAAA,EAA4B;AACnD,IAAA,OAAO,SAAS,gBAAkC,IAAA,EAAmC;AACnF,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,OAAO,CAAA;AAAA,UAC/B,MAAM,QAAA,CAAS,MAAA;AAAA,UACf,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,OAAO;AAAA,WACnD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,MAAM,QAAA,CAAS,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI;AAAA,OACnC;AAAA,IACF,CAAA;AAAA,EACF;AAAA,EAEQ,YAAY,KAAA,EAAmB;AAErC,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,KAAK,CAAA,EAAG;AAC1B,MAAA,IAAA,CAAK,OAAA,CAAQ,OAAO,OAAO,CAAA;AAAA,IAC7B;AAEA,IAAA,IAAA,CAAK,MAAM,KAAA,EAAO,OAAA,EAAS,KAAK,gBAAA,CAAiB,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,EAC7D;AAAA,EAEQ,iBAAiB,QAAA,EAA4B;AACnD,IAAA,OAAO,SAAS,gBAAkC,IAAA,EAAmC;AACnF,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,OAAO,CAAA;AAAA,UAC/B,MAAM,QAAA,CAAS,MAAA;AAAA,UACf,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,OAAO;AAAA,WACnD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,MAAM,QAAA,CAAS,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI;AAAA,OACnC;AAAA,IACF,CAAA;AAAA,EACF;AAAA,EAEQ,eAAe,KAAA,EAAmB;AAExC,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,QAAQ,CAAA,EAAG;AAC7B,MAAA,IAAA,CAAK,OAAA,CAAQ,OAAO,UAAU,CAAA;AAAA,IAChC;AAEA,IAAA,IAAA,CAAK,MAAM,KAAA,EAAO,UAAA,EAAY,KAAK,mBAAA,CAAoB,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,EACnE;AAAA,EAEQ,oBAAoB,QAAA,EAAkC;AAC5D,IAAA,OAAO,SAAS,mBAAqC,IAAA,EAAmC;AACtF,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,UAAU,CAAA;AAAA,UAClC,MAAM,QAAA,CAAS,MAAA;AAAA,UACf,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,UAAU;AAAA,WACtD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,MAAM,QAAA,CAAS,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI;AAAA,OACnC;AAAA,IACF,CAAA;AAAA,EACF;AACF;;;;"}
+{"version":3,"file":"instrumentation.js","sources":["../../../../../../src/integrations/tracing/dataloader/vendored/instrumentation.ts"],"sourcesContent":["/*\n * Copyright The OpenTelemetry Authors\n * SPDX-License-Identifier: Apache-2.0\n *\n * NOTICE from the Sentry authors:\n * - Vendored from: https://github.com/open-telemetry/opentelemetry-js-contrib/tree/15ef7506553f631ea4181391e0c5725a56f0d082/packages/instrumentation-dataloader\n * - Upstream version: @opentelemetry/instrumentation-dataloader@0.35.0\n * - Minor TypeScript strictness adjustments for this repository's compiler settings\n */\n\nimport { InstrumentationBase, InstrumentationNodeModuleDefinition, isWrapped } from '@opentelemetry/instrumentation';\nimport type { BatchLoadFn, DataLoader, DataLoaderConstructor } from './types';\nimport {\n  SDK_VERSION,\n  SEMANTIC_ATTRIBUTE_SENTRY_OP,\n  SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN,\n  SPAN_KIND,\n  startSpan,\n} from '@sentry/core';\n\nconst MODULE_NAME = 'dataloader';\nconst PACKAGE_NAME = '@sentry/instrumentation-dataloader';\nconst ORIGIN = 'auto.db.otel.dataloader';\n\ntype LoadFn = DataLoader['load'];\ntype LoadManyFn = DataLoader['loadMany'];\ntype PrimeFn = DataLoader['prime'];\ntype ClearFn = DataLoader['clear'];\ntype ClearAllFn = DataLoader['clearAll'];\n\nfunction isModule(module: unknown): module is { [Symbol.toStringTag]: 'Module'; default: DataLoaderConstructor } {\n  return (module as { [Symbol.toStringTag]: string })[Symbol.toStringTag] === 'Module';\n}\n\nfunction extractModuleExports(module: unknown): DataLoaderConstructor {\n  return isModule(module)\n    ? module.default // ESM\n    : (module as DataLoaderConstructor); // CommonJS\n}\n\nfunction getSpanName(\n  dataloader: DataLoader,\n  operation: 'load' | 'loadMany' | 'batch' | 'prime' | 'clear' | 'clearAll',\n): string {\n  const dataloaderName = dataloader.name;\n  if (dataloaderName) {\n    return `${MODULE_NAME}.${operation} ${dataloaderName}`;\n  }\n\n  return `${MODULE_NAME}.${operation}`;\n}\n\n// `load`, `loadMany` and `batch` are all cache reads\nfunction getSpanOp(operation: 'load' | 'loadMany' | 'batch' | 'prime' | 'clear' | 'clearAll'): string | undefined {\n  if (operation === 'load' || operation === 'loadMany' || operation === 'batch') {\n    return 'cache.get';\n  }\n\n  return undefined;\n}\n\nexport class DataloaderInstrumentation extends InstrumentationBase {\n  constructor(config = {}) {\n    super(PACKAGE_NAME, SDK_VERSION, config);\n  }\n\n  protected init() {\n    return [\n      new InstrumentationNodeModuleDefinition(\n        MODULE_NAME,\n        ['>=2.0.0 <3'],\n        module => {\n          const dataloader = extractModuleExports(module);\n          this._patchLoad(dataloader.prototype);\n          this._patchLoadMany(dataloader.prototype);\n          this._patchPrime(dataloader.prototype);\n          this._patchClear(dataloader.prototype);\n          this._patchClearAll(dataloader.prototype);\n\n          return this._getPatchedConstructor(dataloader);\n        },\n        module => {\n          const dataloader = extractModuleExports(module);\n          ['load', 'loadMany', 'prime', 'clear', 'clearAll'].forEach(method => {\n            if (isWrapped(dataloader.prototype[method])) {\n              this._unwrap(dataloader.prototype, method);\n            }\n          });\n        },\n      ),\n    ];\n  }\n\n  private _wrapBatchLoadFn(batchLoadFn: BatchLoadFn<unknown, unknown>): BatchLoadFn<unknown, unknown> {\n    // oxlint-disable-next-line typescript/no-this-alias\n    const instrumentation = this;\n\n    return function patchedBatchLoadFn(this: DataLoader, ...args: Parameters<BatchLoadFn<unknown, unknown>>) {\n      if (!instrumentation.isEnabled()) {\n        return batchLoadFn.call(this, ...args);\n      }\n\n      return startSpan(\n        {\n          name: getSpanName(this, 'batch'),\n          links: this._batch?.spanLinks,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('batch'),\n          },\n          onlyIfParent: true,\n        },\n        () => batchLoadFn.apply(this, args),\n      );\n    };\n  }\n\n  private _getPatchedConstructor(constructor: DataLoaderConstructor): DataLoaderConstructor {\n    // oxlint-disable-next-line typescript/no-this-alias\n    const instrumentation = this;\n    const prototype = constructor.prototype;\n\n    if (!instrumentation.isEnabled()) {\n      return constructor;\n    }\n\n    // oxlint-disable-next-line typescript/no-explicit-any\n    function PatchedDataloader(this: DataLoader, ...args: any[]) {\n      // BatchLoadFn is the first constructor argument\n      // https://github.com/graphql/dataloader/blob/77c2cd7ca97e8795242018ebc212ce2487e729d2/src/index.js#L47\n      if (typeof args[0] === 'function') {\n        if (isWrapped(args[0])) {\n          instrumentation._unwrap(args, 0);\n        }\n\n        args[0] = instrumentation._wrapBatchLoadFn(args[0]);\n      }\n\n      return constructor.apply(this, args);\n    }\n\n    PatchedDataloader.prototype = prototype;\n    return PatchedDataloader as unknown as DataLoaderConstructor;\n  }\n\n  private _patchLoad(proto: DataLoader) {\n    // oxlint-disable-next-line typescript/unbound-method\n    if (isWrapped(proto.load)) {\n      this._unwrap(proto, 'load');\n    }\n\n    this._wrap(proto, 'load', this._getPatchedLoad.bind(this));\n  }\n\n  private _getPatchedLoad(original: LoadFn): LoadFn {\n    return function patchedLoad(this: DataLoader, ...args: Parameters<typeof original>) {\n      return startSpan(\n        {\n          name: getSpanName(this, 'load'),\n          kind: SPAN_KIND.CLIENT,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('load'),\n          },\n          onlyIfParent: true,\n        },\n        span => {\n          const result = original.call(this, ...args);\n\n          if (this._batch && span.isRecording()) {\n            if (!this._batch.spanLinks) {\n              this._batch.spanLinks = [];\n            }\n\n            this._batch.spanLinks.push({ context: span.spanContext() });\n          }\n\n          return result;\n        },\n      );\n    };\n  }\n\n  private _patchLoadMany(proto: DataLoader) {\n    // oxlint-disable-next-line typescript/unbound-method\n    if (isWrapped(proto.loadMany)) {\n      this._unwrap(proto, 'loadMany');\n    }\n\n    this._wrap(proto, 'loadMany', this._getPatchedLoadMany.bind(this));\n  }\n\n  private _getPatchedLoadMany(original: LoadManyFn): LoadManyFn {\n    return function patchedLoadMany(this: DataLoader, ...args: Parameters<typeof original>) {\n      return startSpan(\n        {\n          name: getSpanName(this, 'loadMany'),\n          kind: SPAN_KIND.CLIENT,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('loadMany'),\n          },\n          onlyIfParent: true,\n        },\n        () => original.call(this, ...args),\n      );\n    };\n  }\n\n  private _patchPrime(proto: DataLoader) {\n    // oxlint-disable-next-line typescript/unbound-method\n    if (isWrapped(proto.prime)) {\n      this._unwrap(proto, 'prime');\n    }\n\n    this._wrap(proto, 'prime', this._getPatchedPrime.bind(this));\n  }\n\n  private _getPatchedPrime(original: PrimeFn): PrimeFn {\n    return function patchedPrime(this: DataLoader, ...args: Parameters<typeof original>) {\n      return startSpan(\n        {\n          name: getSpanName(this, 'prime'),\n          kind: SPAN_KIND.CLIENT,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('prime'),\n          },\n          onlyIfParent: true,\n        },\n        () => original.call(this, ...args),\n      );\n    };\n  }\n\n  private _patchClear(proto: DataLoader) {\n    // oxlint-disable-next-line typescript/unbound-method\n    if (isWrapped(proto.clear)) {\n      this._unwrap(proto, 'clear');\n    }\n\n    this._wrap(proto, 'clear', this._getPatchedClear.bind(this));\n  }\n\n  private _getPatchedClear(original: ClearFn): ClearFn {\n    return function patchedClear(this: DataLoader, ...args: Parameters<typeof original>) {\n      return startSpan(\n        {\n          name: getSpanName(this, 'clear'),\n          kind: SPAN_KIND.CLIENT,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('clear'),\n          },\n          onlyIfParent: true,\n        },\n        () => original.call(this, ...args),\n      );\n    };\n  }\n\n  private _patchClearAll(proto: DataLoader) {\n    // oxlint-disable-next-line typescript/unbound-method\n    if (isWrapped(proto.clearAll)) {\n      this._unwrap(proto, 'clearAll');\n    }\n\n    this._wrap(proto, 'clearAll', this._getPatchedClearAll.bind(this));\n  }\n\n  private _getPatchedClearAll(original: ClearAllFn): ClearAllFn {\n    return function patchedClearAll(this: DataLoader, ...args: Parameters<typeof original>) {\n      return startSpan(\n        {\n          name: getSpanName(this, 'clearAll'),\n          kind: SPAN_KIND.CLIENT,\n          attributes: {\n            [SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: ORIGIN,\n            [SEMANTIC_ATTRIBUTE_SENTRY_OP]: getSpanOp('clearAll'),\n          },\n          onlyIfParent: true,\n        },\n        () => original.call(this, ...args),\n      );\n    };\n  }\n}\n"],"names":[],"mappings":";;;AAoBA,MAAM,WAAA,GAAc,YAAA;AACpB,MAAM,YAAA,GAAe,oCAAA;AACrB,MAAM,MAAA,GAAS,yBAAA;AAQf,SAAS,SAAS,MAAA,EAA+F;AAC/G,EAAA,OAAQ,MAAA,CAA4C,MAAA,CAAO,WAAW,CAAA,KAAM,QAAA;AAC9E;AAEA,SAAS,qBAAqB,MAAA,EAAwC;AACpE,EAAA,OAAO,QAAA,CAAS,MAAM,CAAA,GAClB,MAAA,CAAO,OAAA,GACN,MAAA;AACP;AAEA,SAAS,WAAA,CACP,YACA,SAAA,EACQ;AACR,EAAA,MAAM,iBAAiB,UAAA,CAAW,IAAA;AAClC,EAAA,IAAI,cAAA,EAAgB;AAClB,IAAA,OAAO,CAAA,EAAG,WAAW,CAAA,CAAA,EAAI,SAAS,IAAI,cAAc,CAAA,CAAA;AAAA,EACtD;AAEA,EAAA,OAAO,CAAA,EAAG,WAAW,CAAA,CAAA,EAAI,SAAS,CAAA,CAAA;AACpC;AAGA,SAAS,UAAU,SAAA,EAA+F;AAChH,EAAA,IAAI,SAAA,KAAc,MAAA,IAAU,SAAA,KAAc,UAAA,IAAc,cAAc,OAAA,EAAS;AAC7E,IAAA,OAAO,WAAA;AAAA,EACT;AAEA,EAAA,OAAO,MAAA;AACT;AAEO,MAAM,kCAAkC,mBAAA,CAAoB;AAAA,EACjE,WAAA,CAAY,MAAA,GAAS,EAAC,EAAG;AACvB,IAAA,KAAA,CAAM,YAAA,EAAc,aAAa,MAAM,CAAA;AAAA,EACzC;AAAA,EAEU,IAAA,GAAO;AACf,IAAA,OAAO;AAAA,MACL,IAAI,mCAAA;AAAA,QACF,WAAA;AAAA,QACA,CAAC,YAAY,CAAA;AAAA,QACb,CAAA,MAAA,KAAU;AACR,UAAA,MAAM,UAAA,GAAa,qBAAqB,MAAM,CAAA;AAC9C,UAAA,IAAA,CAAK,UAAA,CAAW,WAAW,SAAS,CAAA;AACpC,UAAA,IAAA,CAAK,cAAA,CAAe,WAAW,SAAS,CAAA;AACxC,UAAA,IAAA,CAAK,WAAA,CAAY,WAAW,SAAS,CAAA;AACrC,UAAA,IAAA,CAAK,WAAA,CAAY,WAAW,SAAS,CAAA;AACrC,UAAA,IAAA,CAAK,cAAA,CAAe,WAAW,SAAS,CAAA;AAExC,UAAA,OAAO,IAAA,CAAK,uBAAuB,UAAU,CAAA;AAAA,QAC/C,CAAA;AAAA,QACA,CAAA,MAAA,KAAU;AACR,UAAA,MAAM,UAAA,GAAa,qBAAqB,MAAM,CAAA;AAC9C,UAAA,CAAC,QAAQ,UAAA,EAAY,OAAA,EAAS,SAAS,UAAU,CAAA,CAAE,QAAQ,CAAA,MAAA,KAAU;AACnE,YAAA,IAAI,SAAA,CAAU,UAAA,CAAW,SAAA,CAAU,MAAM,CAAC,CAAA,EAAG;AAC3C,cAAA,IAAA,CAAK,OAAA,CAAQ,UAAA,CAAW,SAAA,EAAW,MAAM,CAAA;AAAA,YAC3C;AAAA,UACF,CAAC,CAAA;AAAA,QACH;AAAA;AACF,KACF;AAAA,EACF;AAAA,EAEQ,iBAAiB,WAAA,EAA2E;AAElG,IAAA,MAAM,eAAA,GAAkB,IAAA;AAExB,IAAA,OAAO,SAAS,sBAAwC,IAAA,EAAiD;AACvG,MAAA,IAAI,CAAC,eAAA,CAAgB,SAAA,EAAU,EAAG;AAChC,QAAA,OAAO,WAAA,CAAY,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI,CAAA;AAAA,MACvC;AAEA,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,OAAO,CAAA;AAAA,UAC/B,KAAA,EAAO,KAAK,MAAA,EAAQ,SAAA;AAAA,UACpB,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,OAAO;AAAA,WACnD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,MAAM,WAAA,CAAY,KAAA,CAAM,IAAA,EAAM,IAAI;AAAA,OACpC;AAAA,IACF,CAAA;AAAA,EACF;AAAA,EAEQ,uBAAuB,WAAA,EAA2D;AAExF,IAAA,MAAM,eAAA,GAAkB,IAAA;AACxB,IAAA,MAAM,YAAY,WAAA,CAAY,SAAA;AAE9B,IAAA,IAAI,CAAC,eAAA,CAAgB,SAAA,EAAU,EAAG;AAChC,MAAA,OAAO,WAAA;AAAA,IACT;AAGA,IAAA,SAAS,qBAAuC,IAAA,EAAa;AAG3D,MAAA,IAAI,OAAO,IAAA,CAAK,CAAC,CAAA,KAAM,UAAA,EAAY;AACjC,QAAA,IAAI,SAAA,CAAU,IAAA,CAAK,CAAC,CAAC,CAAA,EAAG;AACtB,UAAA,eAAA,CAAgB,OAAA,CAAQ,MAAM,CAAC,CAAA;AAAA,QACjC;AAEA,QAAA,IAAA,CAAK,CAAC,CAAA,GAAI,eAAA,CAAgB,gBAAA,CAAiB,IAAA,CAAK,CAAC,CAAC,CAAA;AAAA,MACpD;AAEA,MAAA,OAAO,WAAA,CAAY,KAAA,CAAM,IAAA,EAAM,IAAI,CAAA;AAAA,IACrC;AAEA,IAAA,iBAAA,CAAkB,SAAA,GAAY,SAAA;AAC9B,IAAA,OAAO,iBAAA;AAAA,EACT;AAAA,EAEQ,WAAW,KAAA,EAAmB;AAEpC,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,IAAI,CAAA,EAAG;AACzB,MAAA,IAAA,CAAK,OAAA,CAAQ,OAAO,MAAM,CAAA;AAAA,IAC5B;AAEA,IAAA,IAAA,CAAK,MAAM,KAAA,EAAO,MAAA,EAAQ,KAAK,eAAA,CAAgB,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,EAC3D;AAAA,EAEQ,gBAAgB,QAAA,EAA0B;AAChD,IAAA,OAAO,SAAS,eAAiC,IAAA,EAAmC;AAClF,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,MAAM,CAAA;AAAA,UAC9B,MAAM,SAAA,CAAU,MAAA;AAAA,UAChB,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,MAAM;AAAA,WAClD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,CAAA,IAAA,KAAQ;AACN,UAAA,MAAM,MAAA,GAAS,QAAA,CAAS,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI,CAAA;AAE1C,UAAA,IAAI,IAAA,CAAK,MAAA,IAAU,IAAA,CAAK,WAAA,EAAY,EAAG;AACrC,YAAA,IAAI,CAAC,IAAA,CAAK,MAAA,CAAO,SAAA,EAAW;AAC1B,cAAA,IAAA,CAAK,MAAA,CAAO,YAAY,EAAC;AAAA,YAC3B;AAEA,YAAA,IAAA,CAAK,MAAA,CAAO,UAAU,IAAA,CAAK,EAAE,SAAS,IAAA,CAAK,WAAA,IAAe,CAAA;AAAA,UAC5D;AAEA,UAAA,OAAO,MAAA;AAAA,QACT;AAAA,OACF;AAAA,IACF,CAAA;AAAA,EACF;AAAA,EAEQ,eAAe,KAAA,EAAmB;AAExC,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,QAAQ,CAAA,EAAG;AAC7B,MAAA,IAAA,CAAK,OAAA,CAAQ,OAAO,UAAU,CAAA;AAAA,IAChC;AAEA,IAAA,IAAA,CAAK,MAAM,KAAA,EAAO,UAAA,EAAY,KAAK,mBAAA,CAAoB,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,EACnE;AAAA,EAEQ,oBAAoB,QAAA,EAAkC;AAC5D,IAAA,OAAO,SAAS,mBAAqC,IAAA,EAAmC;AACtF,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,UAAU,CAAA;AAAA,UAClC,MAAM,SAAA,CAAU,MAAA;AAAA,UAChB,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,UAAU;AAAA,WACtD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,MAAM,QAAA,CAAS,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI;AAAA,OACnC;AAAA,IACF,CAAA;AAAA,EACF;AAAA,EAEQ,YAAY,KAAA,EAAmB;AAErC,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,KAAK,CAAA,EAAG;AAC1B,MAAA,IAAA,CAAK,OAAA,CAAQ,OAAO,OAAO,CAAA;AAAA,IAC7B;AAEA,IAAA,IAAA,CAAK,MAAM,KAAA,EAAO,OAAA,EAAS,KAAK,gBAAA,CAAiB,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,EAC7D;AAAA,EAEQ,iBAAiB,QAAA,EAA4B;AACnD,IAAA,OAAO,SAAS,gBAAkC,IAAA,EAAmC;AACnF,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,OAAO,CAAA;AAAA,UAC/B,MAAM,SAAA,CAAU,MAAA;AAAA,UAChB,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,OAAO;AAAA,WACnD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,MAAM,QAAA,CAAS,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI;AAAA,OACnC;AAAA,IACF,CAAA;AAAA,EACF;AAAA,EAEQ,YAAY,KAAA,EAAmB;AAErC,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,KAAK,CAAA,EAAG;AAC1B,MAAA,IAAA,CAAK,OAAA,CAAQ,OAAO,OAAO,CAAA;AAAA,IAC7B;AAEA,IAAA,IAAA,CAAK,MAAM,KAAA,EAAO,OAAA,EAAS,KAAK,gBAAA,CAAiB,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,EAC7D;AAAA,EAEQ,iBAAiB,QAAA,EAA4B;AACnD,IAAA,OAAO,SAAS,gBAAkC,IAAA,EAAmC;AACnF,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,OAAO,CAAA;AAAA,UAC/B,MAAM,SAAA,CAAU,MAAA;AAAA,UAChB,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,OAAO;AAAA,WACnD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,MAAM,QAAA,CAAS,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI;AAAA,OACnC;AAAA,IACF,CAAA;AAAA,EACF;AAAA,EAEQ,eAAe,KAAA,EAAmB;AAExC,IAAA,IAAI,SAAA,CAAU,KAAA,CAAM,QAAQ,CAAA,EAAG;AAC7B,MAAA,IAAA,CAAK,OAAA,CAAQ,OAAO,UAAU,CAAA;AAAA,IAChC;AAEA,IAAA,IAAA,CAAK,MAAM,KAAA,EAAO,UAAA,EAAY,KAAK,mBAAA,CAAoB,IAAA,CAAK,IAAI,CAAC,CAAA;AAAA,EACnE;AAAA,EAEQ,oBAAoB,QAAA,EAAkC;AAC5D,IAAA,OAAO,SAAS,mBAAqC,IAAA,EAAmC;AACtF,MAAA,OAAO,SAAA;AAAA,QACL;AAAA,UACE,IAAA,EAAM,WAAA,CAAY,IAAA,EAAM,UAAU,CAAA;AAAA,UAClC,MAAM,SAAA,CAAU,MAAA;AAAA,UAChB,UAAA,EAAY;AAAA,YACV,CAAC,gCAAgC,GAAG,MAAA;AAAA,YACpC,CAAC,4BAA4B,GAAG,SAAA,CAAU,UAAU;AAAA,WACtD;AAAA,UACA,YAAA,EAAc;AAAA,SAChB;AAAA,QACA,MAAM,QAAA,CAAS,IAAA,CAAK,IAAA,EAAM,GAAG,IAAI;AAAA,OACnC;AAAA,IACF,CAAA;AAAA,EACF;AACF;;;;"}

package/build/esm/integrations/tracing/kafka/index.js

@@ -1,19 +1,9 @@
 import { KafkaJsInstrumentation } from './vendored/instrumentation.js';
 import { defineIntegration } from '@sentry/core';
-import { generateInstrumentOnce, addOriginToSpan } from '@sentry/node-core';
+import { generateInstrumentOnce } from '@sentry/node-core';
 
 const INTEGRATION_NAME = "Kafka";
-const instrumentKafka = generateInstrumentOnce(
-  INTEGRATION_NAME,
-  () => new KafkaJsInstrumentation({
-    consumerHook(span) {
-      addOriginToSpan(span, "auto.kafkajs.otel.consumer");
-    },
-    producerHook(span) {
-      addOriginToSpan(span, "auto.kafkajs.otel.producer");
-    }
-  })
-);
+const instrumentKafka = generateInstrumentOnce(INTEGRATION_NAME, () => new KafkaJsInstrumentation());
 const _kafkaIntegration = (() => {
   return {
     name: INTEGRATION_NAME,

package/build/esm/integrations/tracing/kafka/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sources":["../../../../../src/integrations/tracing/kafka/index.ts"],"sourcesContent":["import { KafkaJsInstrumentation } from './vendored/instrumentation';\nimport type { IntegrationFn } from '@sentry/core';\nimport { defineIntegration } from '@sentry/core';\nimport { addOriginToSpan, generateInstrumentOnce } from '@sentry/node-core';\n\nconst INTEGRATION_NAME = 'Kafka';\n\nexport const instrumentKafka = generateInstrumentOnce(\n  INTEGRATION_NAME,\n  () =>\n    new KafkaJsInstrumentation({\n      consumerHook(span) {\n        addOriginToSpan(span, 'auto.kafkajs.otel.consumer');\n      },\n      producerHook(span) {\n        addOriginToSpan(span, 'auto.kafkajs.otel.producer');\n      },\n    }),\n);\n\nconst _kafkaIntegration = (() => {\n  return {\n    name: INTEGRATION_NAME,\n    setupOnce() {\n      instrumentKafka();\n    },\n  };\n}) satisfies IntegrationFn;\n\n/**\n * Adds Sentry tracing instrumentation for the [kafkajs](https://www.npmjs.com/package/kafkajs) library.\n *\n * For more information, see the [`kafkaIntegration` documentation](https://docs.sentry.io/platforms/javascript/guides/node/configuration/integrations/kafka/).\n *\n * @example\n * ```javascript\n * const Sentry = require('@sentry/node');\n *\n * Sentry.init({\n *  integrations: [Sentry.kafkaIntegration()],\n * });\n */\nexport const kafkaIntegration = defineIntegration(_kafkaIntegration);\n"],"names":[],"mappings":";;;;AAKA,MAAM,gBAAA,GAAmB,OAAA;AAElB,MAAM,eAAA,GAAkB,sBAAA;AAAA,EAC7B,gBAAA;AAAA,EACA,MACE,IAAI,sBAAA,CAAuB;AAAA,IACzB,aAAa,IAAA,EAAM;AACjB,MAAA,eAAA,CAAgB,MAAM,4BAA4B,CAAA;AAAA,IACpD,CAAA;AAAA,IACA,aAAa,IAAA,EAAM;AACjB,MAAA,eAAA,CAAgB,MAAM,4BAA4B,CAAA;AAAA,IACpD;AAAA,GACD;AACL;AAEA,MAAM,qBAAqB,MAAM;AAC/B,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,gBAAA;AAAA,IACN,SAAA,GAAY;AACV,MAAA,eAAA,EAAgB;AAAA,IAClB;AAAA,GACF;AACF,CAAA,CAAA;AAeO,MAAM,gBAAA,GAAmB,kBAAkB,iBAAiB;;;;"}
+{"version":3,"file":"index.js","sources":["../../../../../src/integrations/tracing/kafka/index.ts"],"sourcesContent":["import { KafkaJsInstrumentation } from './vendored/instrumentation';\nimport type { IntegrationFn } from '@sentry/core';\nimport { defineIntegration } from '@sentry/core';\nimport { generateInstrumentOnce } from '@sentry/node-core';\n\nconst INTEGRATION_NAME = 'Kafka';\n\nexport const instrumentKafka = generateInstrumentOnce(INTEGRATION_NAME, () => new KafkaJsInstrumentation());\n\nconst _kafkaIntegration = (() => {\n  return {\n    name: INTEGRATION_NAME,\n    setupOnce() {\n      instrumentKafka();\n    },\n  };\n}) satisfies IntegrationFn;\n\n/**\n * Adds Sentry tracing instrumentation for the [kafkajs](https://www.npmjs.com/package/kafkajs) library.\n *\n * For more information, see the [`kafkaIntegration` documentation](https://docs.sentry.io/platforms/javascript/guides/node/configuration/integrations/kafka/).\n *\n * @example\n * ```javascript\n * const Sentry = require('@sentry/node');\n *\n * Sentry.init({\n *  integrations: [Sentry.kafkaIntegration()],\n * });\n */\nexport const kafkaIntegration = defineIntegration(_kafkaIntegration);\n"],"names":[],"mappings":";;;;AAKA,MAAM,gBAAA,GAAmB,OAAA;AAElB,MAAM,kBAAkB,sBAAA,CAAuB,gBAAA,EAAkB,MAAM,IAAI,wBAAwB;AAE1G,MAAM,qBAAqB,MAAM;AAC/B,EAAA,OAAO;AAAA,IACL,IAAA,EAAM,gBAAA;AAAA,IACN,SAAA,GAAY;AACV,MAAA,eAAA,EAAgB;AAAA,IAClB;AAAA,GACF;AACF,CAAA,CAAA;AAeO,MAAM,gBAAA,GAAmB,kBAAkB,iBAAiB;;;;"}

package/build/esm/integrations/tracing/kafka/vendored/instrumentation.js

@@ -1,43 +1,13 @@
-import { propagation, ROOT_CONTEXT, context, trace, SpanStatusCode, SpanKind } from '@opentelemetry/api';
-import { InstrumentationBase, InstrumentationNodeModuleDefinition, isWrapped, safeExecuteInTheMiddle } from '@opentelemetry/instrumentation';
-import { ATTR_SERVER_PORT, ATTR_SERVER_ADDRESS, ERROR_TYPE_VALUE_OTHER, ATTR_ERROR_TYPE } from '@opentelemetry/semantic-conventions';
-import { EVENT_LISTENERS_SET } from './internal-types.js';
-import { bufferTextMapGetter } from './propagator.js';
-import { METRIC_MESSAGING_CLIENT_OPERATION_DURATION, METRIC_MESSAGING_CLIENT_SENT_MESSAGES, METRIC_MESSAGING_CLIENT_CONSUMED_MESSAGES, METRIC_MESSAGING_PROCESS_DURATION, MESSAGING_SYSTEM_VALUE_KAFKA, ATTR_MESSAGING_OPERATION_NAME, ATTR_MESSAGING_SYSTEM, MESSAGING_OPERATION_TYPE_VALUE_PROCESS, ATTR_MESSAGING_DESTINATION_PARTITION_ID, ATTR_MESSAGING_DESTINATION_NAME, MESSAGING_OPERATION_TYPE_VALUE_RECEIVE, ATTR_MESSAGING_BATCH_MESSAGE_COUNT, ATTR_MESSAGING_KAFKA_OFFSET, ATTR_MESSAGING_KAFKA_MESSAGE_TOMBSTONE, ATTR_MESSAGING_KAFKA_MESSAGE_KEY, ATTR_MESSAGING_OPERATION_TYPE, MESSAGING_OPERATION_TYPE_VALUE_SEND } from './semconv.js';
-import { SDK_VERSION } from '@sentry/core';
+import { InstrumentationBase, InstrumentationNodeModuleDefinition, isWrapped } from '@opentelemetry/instrumentation';
+import { SDK_VERSION, continueTrace, withActiveSpan, startNewTrace, startInactiveSpan, SPAN_STATUS_ERROR, SPAN_STATUS_OK } from '@sentry/core';
+import { ATTR_MESSAGING_DESTINATION_PARTITION_ID, ATTR_MESSAGING_BATCH_MESSAGE_COUNT, MESSAGING_OPERATION_TYPE_VALUE_PROCESS, MESSAGING_OPERATION_TYPE_VALUE_RECEIVE } from './semconv.js';
+import { getHeaderAsString, startConsumerSpan, endSpansOnPromise, getLinksFromHeaders, startProducerSpan } from './utils.js';
 
 const PACKAGE_NAME = "@sentry/instrumentation-kafkajs";
-function prepareCounter(meter, value, attributes) {
-  return (errorType) => {
-    meter.add(value, {
-      ...attributes,
-      ...errorType ? { [ATTR_ERROR_TYPE]: errorType } : {}
-    });
-  };
-}
-function prepareDurationHistogram(meter, value, attributes) {
-  return (errorType) => {
-    meter.record((Date.now() - value) / 1e3, {
-      ...attributes,
-      ...errorType ? { [ATTR_ERROR_TYPE]: errorType } : {}
-    });
-  };
-}
-const HISTOGRAM_BUCKET_BOUNDARIES = [5e-3, 0.01, 0.025, 0.05, 0.075, 0.1, 0.25, 0.5, 0.75, 1, 2.5, 5, 7.5, 10];
 class KafkaJsInstrumentation extends InstrumentationBase {
   constructor(config = {}) {
     super(PACKAGE_NAME, SDK_VERSION, config);
   }
-  _updateMetricInstruments() {
-    this._clientDuration = this.meter.createHistogram(METRIC_MESSAGING_CLIENT_OPERATION_DURATION, {
-      advice: { explicitBucketBoundaries: HISTOGRAM_BUCKET_BOUNDARIES }
-    });
-    this._sentMessages = this.meter.createCounter(METRIC_MESSAGING_CLIENT_SENT_MESSAGES);
-    this._consumedMessages = this.meter.createCounter(METRIC_MESSAGING_CLIENT_CONSUMED_MESSAGES);
-    this._processDuration = this.meter.createHistogram(METRIC_MESSAGING_PROCESS_DURATION, {
-      advice: { explicitBucketBoundaries: HISTOGRAM_BUCKET_BOUNDARIES }
-    });
-  }
   init() {
     const unpatch = (moduleExports) => {
       if (isWrapped(moduleExports?.Kafka?.prototype.producer)) {
@@ -69,27 +39,10 @@ class KafkaJsInstrumentation extends InstrumentationBase {
           instrumentation._unwrap(newConsumer, "run");
         }
         instrumentation._wrap(newConsumer, "run", instrumentation._getConsumerRunPatch());
-        instrumentation._setKafkaEventListeners(newConsumer);
         return newConsumer;
       };
     };
   }
-  _setKafkaEventListeners(kafkaObj) {
-    if (kafkaObj[EVENT_LISTENERS_SET]) return;
-    if (kafkaObj.events?.REQUEST) {
-      kafkaObj.on(kafkaObj.events.REQUEST, this._recordClientDurationMetric.bind(this));
-    }
-    kafkaObj[EVENT_LISTENERS_SET] = true;
-  }
-  _recordClientDurationMetric(event) {
-    const [address = "", port = "0"] = event.payload.broker.split(":");
-    this._clientDuration.record(event.payload.duration / 1e3, {
-      [ATTR_MESSAGING_SYSTEM]: MESSAGING_SYSTEM_VALUE_KAFKA,
-      [ATTR_MESSAGING_OPERATION_NAME]: `${event.payload.apiName}`,
-      [ATTR_SERVER_ADDRESS]: address,
-      [ATTR_SERVER_PORT]: Number.parseInt(port, 10)
-    });
-  }
   _getProducerPatch() {
     const instrumentation = this;
     return (original) => {
@@ -107,7 +60,6 @@ class KafkaJsInstrumentation extends InstrumentationBase {
           instrumentation._unwrap(newProducer, "transaction");
         }
         instrumentation._wrap(newProducer, "transaction", instrumentation._getProducerTransactionPatch());
-        instrumentation._setKafkaEventListeners(newProducer);
         return newProducer;
       };
     };
@@ -134,103 +86,60 @@ class KafkaJsInstrumentation extends InstrumentationBase {
     };
   }
   _getConsumerEachMessagePatch() {
-    const instrumentation = this;
     return (original) => {
       return function eachMessage(...args) {
         const payload = args[0];
-        const propagatedContext = propagation.extract(
-          ROOT_CONTEXT,
-          payload.message.headers,
-          bufferTextMapGetter
-        );
-        const span = instrumentation._startConsumerSpan({
-          topic: payload.topic,
-          message: payload.message,
-          operationType: MESSAGING_OPERATION_TYPE_VALUE_PROCESS,
-          ctx: propagatedContext,
-          attributes: {
-            [ATTR_MESSAGING_DESTINATION_PARTITION_ID]: String(payload.partition)
-          }
-        });
-        const pendingMetrics = [
-          prepareDurationHistogram(instrumentation._processDuration, Date.now(), {
-            [ATTR_MESSAGING_SYSTEM]: MESSAGING_SYSTEM_VALUE_KAFKA,
-            [ATTR_MESSAGING_OPERATION_NAME]: "process",
-            [ATTR_MESSAGING_DESTINATION_NAME]: payload.topic,
-            [ATTR_MESSAGING_DESTINATION_PARTITION_ID]: String(payload.partition)
-          }),
-          prepareCounter(instrumentation._consumedMessages, 1, {
-            [ATTR_MESSAGING_SYSTEM]: MESSAGING_SYSTEM_VALUE_KAFKA,
-            [ATTR_MESSAGING_OPERATION_NAME]: "process",
-            [ATTR_MESSAGING_DESTINATION_NAME]: payload.topic,
-            [ATTR_MESSAGING_DESTINATION_PARTITION_ID]: String(payload.partition)
-          })
-        ];
-        const eachMessagePromise = context.with(trace.setSpan(propagatedContext, span), () => {
-          return original.apply(this, args);
+        const sentryTrace = getHeaderAsString(payload.message.headers, "sentry-trace");
+        const baggage = getHeaderAsString(payload.message.headers, "baggage");
+        return continueTrace({ sentryTrace, baggage }, () => {
+          const span = startConsumerSpan({
+            topic: payload.topic,
+            message: payload.message,
+            operationType: MESSAGING_OPERATION_TYPE_VALUE_PROCESS,
+            attributes: {
+              [ATTR_MESSAGING_DESTINATION_PARTITION_ID]: String(payload.partition)
+            }
+          });
+          const eachMessagePromise = withActiveSpan(span, () => {
+            return original.apply(this, args);
+          });
+          return endSpansOnPromise([span], eachMessagePromise);
         });
-        return instrumentation._endSpansOnPromise([span], pendingMetrics, eachMessagePromise);
       };
     };
   }
   _getConsumerEachBatchPatch() {
     return (original) => {
-      const instrumentation = this;
       return function eachBatch(...args) {
         const payload = args[0];
-        const receivingSpan = instrumentation._startConsumerSpan({
-          topic: payload.batch.topic,
-          message: void 0,
-          operationType: MESSAGING_OPERATION_TYPE_VALUE_RECEIVE,
-          ctx: ROOT_CONTEXT,
-          attributes: {
-            [ATTR_MESSAGING_BATCH_MESSAGE_COUNT]: payload.batch.messages.length,
-            [ATTR_MESSAGING_DESTINATION_PARTITION_ID]: String(payload.batch.partition)
-          }
-        });
-        return context.with(trace.setSpan(context.active(), receivingSpan), () => {
-          const startTime = Date.now();
-          const spans = [];
-          const pendingMetrics = [
-            prepareCounter(instrumentation._consumedMessages, payload.batch.messages.length, {
-              [ATTR_MESSAGING_SYSTEM]: MESSAGING_SYSTEM_VALUE_KAFKA,
-              [ATTR_MESSAGING_OPERATION_NAME]: "process",
-              [ATTR_MESSAGING_DESTINATION_NAME]: payload.batch.topic,
+        const receivingSpan = startNewTrace(
+          () => startConsumerSpan({
+            topic: payload.batch.topic,
+            message: void 0,
+            operationType: MESSAGING_OPERATION_TYPE_VALUE_RECEIVE,
+            attributes: {
+              [ATTR_MESSAGING_BATCH_MESSAGE_COUNT]: payload.batch.messages.length,
               [ATTR_MESSAGING_DESTINATION_PARTITION_ID]: String(payload.batch.partition)
-            })
-          ];
-          payload.batch.messages.forEach((message) => {
-            const propagatedContext = propagation.extract(ROOT_CONTEXT, message.headers, bufferTextMapGetter);
-            const spanContext = trace.getSpan(propagatedContext)?.spanContext();
-            let origSpanLink;
-            if (spanContext) {
-              origSpanLink = {
-                context: spanContext
-              };
             }
+          })
+        );
+        return withActiveSpan(receivingSpan, () => {
+          const spans = [receivingSpan];
+          payload.batch.messages.forEach((message) => {
             spans.push(
-              instrumentation._startConsumerSpan({
+              startConsumerSpan({
                 topic: payload.batch.topic,
                 message,
                 operationType: MESSAGING_OPERATION_TYPE_VALUE_PROCESS,
-                link: origSpanLink,
+                links: getLinksFromHeaders(message.headers),
                 attributes: {
                   [ATTR_MESSAGING_DESTINATION_PARTITION_ID]: String(payload.batch.partition)
                 }
               })
             );
-            pendingMetrics.push(
-              prepareDurationHistogram(instrumentation._processDuration, startTime, {
-                [ATTR_MESSAGING_SYSTEM]: MESSAGING_SYSTEM_VALUE_KAFKA,
-                [ATTR_MESSAGING_OPERATION_NAME]: "process",
-                [ATTR_MESSAGING_DESTINATION_NAME]: payload.batch.topic,
-                [ATTR_MESSAGING_DESTINATION_PARTITION_ID]: String(payload.batch.partition)
-              })
-            );
           });
           const batchMessagePromise = original.apply(this, args);
-          spans.unshift(receivingSpan);
-          return instrumentation._endSpansOnPromise(spans, pendingMetrics, batchMessagePromise);
+          return endSpansOnPromise(spans, batchMessagePromise);
         });
       };
     };
@@ -239,33 +148,31 @@ class KafkaJsInstrumentation extends InstrumentationBase {
     const instrumentation = this;
     return (original) => {
       return function transaction(...args) {
-        const transactionSpan = instrumentation.tracer.startSpan("transaction");
+        const transactionSpan = startInactiveSpan({ name: "transaction" });
         const transactionPromise = original.apply(this, args);
         transactionPromise.then((transaction2) => {
           const originalSend = transaction2.send;
           transaction2.send = function send(...args2) {
-            return context.with(trace.setSpan(context.active(), transactionSpan), () => {
+            return withActiveSpan(transactionSpan, () => {
               const patched = instrumentation._getSendPatch()(originalSend);
               return patched.apply(this, args2).catch((err) => {
                 transactionSpan.setStatus({
-                  code: SpanStatusCode.ERROR,
+                  code: SPAN_STATUS_ERROR,
                   message: err?.message
                 });
-                transactionSpan.recordException(err);
                 throw err;
               });
             });
           };
           const originalSendBatch = transaction2.sendBatch;
           transaction2.sendBatch = function sendBatch(...args2) {
-            return context.with(trace.setSpan(context.active(), transactionSpan), () => {
+            return withActiveSpan(transactionSpan, () => {
               const patched = instrumentation._getSendBatchPatch()(originalSendBatch);
               return patched.apply(this, args2).catch((err) => {
                 transactionSpan.setStatus({
-                  code: SpanStatusCode.ERROR,
+                  code: SPAN_STATUS_ERROR,
                   message: err?.message
                 });
-                transactionSpan.recordException(err);
                 throw err;
               });
             });
@@ -273,21 +180,20 @@ class KafkaJsInstrumentation extends InstrumentationBase {
           const originalCommit = transaction2.commit;
           transaction2.commit = function commit(...args2) {
             const originCommitPromise = originalCommit.apply(this, args2).then(() => {
-              transactionSpan.setStatus({ code: SpanStatusCode.OK });
+              transactionSpan.setStatus({ code: SPAN_STATUS_OK });
             });
-            return instrumentation._endSpansOnPromise([transactionSpan], [], originCommitPromise);
+            return endSpansOnPromise([transactionSpan], originCommitPromise);
           };
           const originalAbort = transaction2.abort;
           transaction2.abort = function abort(...args2) {
             const originAbortPromise = originalAbort.apply(this, args2);
-            return instrumentation._endSpansOnPromise([transactionSpan], [], originAbortPromise);
+            return endSpansOnPromise([transactionSpan], originAbortPromise);
           };
         }).catch((err) => {
           transactionSpan.setStatus({
-            code: SpanStatusCode.ERROR,
+            code: SPAN_STATUS_ERROR,
             message: err?.message
           });
-          transactionSpan.recordException(err);
           transactionSpan.end();
         });
         return transactionPromise;
@@ -295,141 +201,33 @@ class KafkaJsInstrumentation extends InstrumentationBase {
     };
   }
   _getSendBatchPatch() {
-    const instrumentation = this;
     return (original) => {
       return function sendBatch(...args) {
         const batch = args[0];
         const messages = batch.topicMessages || [];
         const spans = [];
-        const pendingMetrics = [];
         messages.forEach((topicMessage) => {
           topicMessage.messages.forEach((message) => {
-            spans.push(instrumentation._startProducerSpan(topicMessage.topic, message));
-            pendingMetrics.push(
-              prepareCounter(instrumentation._sentMessages, 1, {
-                [ATTR_MESSAGING_SYSTEM]: MESSAGING_SYSTEM_VALUE_KAFKA,
-                [ATTR_MESSAGING_OPERATION_NAME]: "send",
-                [ATTR_MESSAGING_DESTINATION_NAME]: topicMessage.topic,
-                ...message.partition !== void 0 ? {
-                  [ATTR_MESSAGING_DESTINATION_PARTITION_ID]: String(message.partition)
-                } : {}
-              })
-            );
+            spans.push(startProducerSpan(topicMessage.topic, message));
           });
         });
         const origSendResult = original.apply(this, args);
-        return instrumentation._endSpansOnPromise(spans, pendingMetrics, origSendResult);
+        return endSpansOnPromise(spans, origSendResult);
       };
     };
   }
   _getSendPatch() {
-    const instrumentation = this;
     return (original) => {
       return function send(...args) {
         const record = args[0];
         const spans = record.messages.map((message) => {
-          return instrumentation._startProducerSpan(record.topic, message);
+          return startProducerSpan(record.topic, message);
         });
-        const pendingMetrics = record.messages.map(
-          (m) => prepareCounter(instrumentation._sentMessages, 1, {
-            [ATTR_MESSAGING_SYSTEM]: MESSAGING_SYSTEM_VALUE_KAFKA,
-            [ATTR_MESSAGING_OPERATION_NAME]: "send",
-            [ATTR_MESSAGING_DESTINATION_NAME]: record.topic,
-            ...m.partition !== void 0 ? {
-              [ATTR_MESSAGING_DESTINATION_PARTITION_ID]: String(m.partition)
-            } : {}
-          })
-        );
         const origSendResult = original.apply(this, args);
-        return instrumentation._endSpansOnPromise(spans, pendingMetrics, origSendResult);
+        return endSpansOnPromise(spans, origSendResult);
       };
     };
   }
-  _endSpansOnPromise(spans, pendingMetrics, sendPromise) {
-    return Promise.resolve(sendPromise).then((result) => {
-      pendingMetrics.forEach((m) => m());
-      return result;
-    }).catch((reason) => {
-      let errorMessage;
-      let errorType = ERROR_TYPE_VALUE_OTHER;
-      if (typeof reason === "string" || reason === void 0) {
-        errorMessage = reason;
-      } else if (typeof reason === "object" && Object.prototype.hasOwnProperty.call(reason, "message")) {
-        errorMessage = reason.message;
-        errorType = reason.constructor.name;
-      }
-      pendingMetrics.forEach((m) => m(errorType));
-      spans.forEach((span) => {
-        span.setAttribute(ATTR_ERROR_TYPE, errorType);
-        span.setStatus({
-          code: SpanStatusCode.ERROR,
-          message: errorMessage
-        });
-      });
-      throw reason;
-    }).finally(() => {
-      spans.forEach((span) => span.end());
-    });
-  }
-  _startConsumerSpan({ topic, message, operationType, ctx, link, attributes }) {
-    const operationName = operationType === MESSAGING_OPERATION_TYPE_VALUE_RECEIVE ? "poll" : operationType;
-    const span = this.tracer.startSpan(
-      `${operationName} ${topic}`,
-      {
-        kind: operationType === MESSAGING_OPERATION_TYPE_VALUE_RECEIVE ? SpanKind.CLIENT : SpanKind.CONSUMER,
-        attributes: {
-          ...attributes,
-          [ATTR_MESSAGING_SYSTEM]: MESSAGING_SYSTEM_VALUE_KAFKA,
-          [ATTR_MESSAGING_DESTINATION_NAME]: topic,
-          [ATTR_MESSAGING_OPERATION_TYPE]: operationType,
-          [ATTR_MESSAGING_OPERATION_NAME]: operationName,
-          [ATTR_MESSAGING_KAFKA_MESSAGE_KEY]: message?.key ? String(message.key) : void 0,
-          [ATTR_MESSAGING_KAFKA_MESSAGE_TOMBSTONE]: message?.key && message.value === null ? true : void 0,
-          [ATTR_MESSAGING_KAFKA_OFFSET]: message?.offset
-        },
-        links: link ? [link] : []
-      },
-      ctx
-    );
-    const { consumerHook } = this.getConfig();
-    if (consumerHook && message) {
-      safeExecuteInTheMiddle(
-        () => consumerHook(span, { topic, message }),
-        (e) => {
-          if (e) this._diag.error("consumerHook error", e);
-        },
-        true
-      );
-    }
-    return span;
-  }
-  _startProducerSpan(topic, message) {
-    const span = this.tracer.startSpan(`send ${topic}`, {
-      kind: SpanKind.PRODUCER,
-      attributes: {
-        [ATTR_MESSAGING_SYSTEM]: MESSAGING_SYSTEM_VALUE_KAFKA,
-        [ATTR_MESSAGING_DESTINATION_NAME]: topic,
-        [ATTR_MESSAGING_KAFKA_MESSAGE_KEY]: message.key ? String(message.key) : void 0,
-        [ATTR_MESSAGING_KAFKA_MESSAGE_TOMBSTONE]: message.key && message.value === null ? true : void 0,
-        [ATTR_MESSAGING_DESTINATION_PARTITION_ID]: message.partition !== void 0 ? String(message.partition) : void 0,
-        [ATTR_MESSAGING_OPERATION_NAME]: "send",
-        [ATTR_MESSAGING_OPERATION_TYPE]: MESSAGING_OPERATION_TYPE_VALUE_SEND
-      }
-    });
-    message.headers = message.headers ?? {};
-    propagation.inject(trace.setSpan(context.active(), span), message.headers);
-    const { producerHook } = this.getConfig();
-    if (producerHook) {
-      safeExecuteInTheMiddle(
-        () => producerHook(span, { topic, message }),
-        (e) => {
-          if (e) this._diag.error("producerHook error", e);
-        },
-        true
-      );
-    }
-    return span;
-  }
 }
 
 export { KafkaJsInstrumentation };