Marc Mintel
5397309103
Some checks failed
Build & Deploy / 🔍 Prepare (push) Successful in 20s
Build & Deploy / 🧪 QA (push) Failing after 34s
Build & Deploy / 🏗️ Build (push) Has started running
Build & Deploy / 🚀 Deploy (push) Has been cancelled
Build & Deploy / 🧪 Smoke Test (push) Has been cancelled
Build & Deploy / ⚡ Lighthouse (push) Has been cancelled
Build & Deploy / 🔔 Notify (push) Has been cancelled
70 lines
2.4 KiB
Plaintext
70 lines
2.4 KiB
Plaintext
import { DataloaderInstrumentation } from '@opentelemetry/instrumentation-dataloader';
|
|
import { defineIntegration, spanToJSON, SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, SEMANTIC_ATTRIBUTE_SENTRY_OP } from '@sentry/core';
|
|
import { generateInstrumentOnce, instrumentWhenWrapped } from '@sentry/node-core';
|
|
|
|
const INTEGRATION_NAME = 'Dataloader';
|
|
|
|
const instrumentDataloader = generateInstrumentOnce(
|
|
INTEGRATION_NAME,
|
|
() =>
|
|
new DataloaderInstrumentation({
|
|
requireParentSpan: true,
|
|
}),
|
|
);
|
|
|
|
const _dataloaderIntegration = (() => {
|
|
let instrumentationWrappedCallback;
|
|
|
|
return {
|
|
name: INTEGRATION_NAME,
|
|
setupOnce() {
|
|
const instrumentation = instrumentDataloader();
|
|
instrumentationWrappedCallback = instrumentWhenWrapped(instrumentation);
|
|
},
|
|
|
|
setup(client) {
|
|
// This is called either immediately or when the instrumentation is wrapped
|
|
instrumentationWrappedCallback?.(() => {
|
|
client.on('spanStart', span => {
|
|
const spanJSON = spanToJSON(span);
|
|
if (spanJSON.description?.startsWith('dataloader')) {
|
|
span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN, 'auto.db.otel.dataloader');
|
|
}
|
|
|
|
// These are all possible dataloader span descriptions
|
|
// Still checking for the future versions
|
|
// in case they add support for `clear` and `prime`
|
|
if (
|
|
spanJSON.description === 'dataloader.load' ||
|
|
spanJSON.description === 'dataloader.loadMany' ||
|
|
spanJSON.description === 'dataloader.batch'
|
|
) {
|
|
span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, 'cache.get');
|
|
// TODO: We can try adding `key` to the `data` attribute upstream.
|
|
// Or alternatively, we can add `requestHook` to the dataloader instrumentation.
|
|
}
|
|
});
|
|
});
|
|
},
|
|
};
|
|
}) ;
|
|
|
|
/**
|
|
* Adds Sentry tracing instrumentation for the [dataloader](https://www.npmjs.com/package/dataloader) library.
|
|
*
|
|
* For more information, see the [`dataloaderIntegration` documentation](https://docs.sentry.io/platforms/javascript/guides/node/configuration/integrations/dataloader/).
|
|
*
|
|
* @example
|
|
* ```javascript
|
|
* const Sentry = require('@sentry/node');
|
|
*
|
|
* Sentry.init({
|
|
* integrations: [Sentry.dataloaderIntegration()],
|
|
* });
|
|
* ```
|
|
*/
|
|
const dataloaderIntegration = defineIntegration(_dataloaderIntegration);
|
|
|
|
export { dataloaderIntegration, instrumentDataloader };
|
|
//# sourceMappingURL=dataloader.js.map
|