UNPKG

@sentry/node

Version:

Sentry Node SDK using OpenTelemetry for performance instrumentation

130 lines (118 loc) 3.78 kB
import { KoaInstrumentation } from '@opentelemetry/instrumentation-koa'; import { ATTR_HTTP_ROUTE } from '@opentelemetry/semantic-conventions'; import { spanToJSON, SEMANTIC_ATTRIBUTE_SENTRY_OP, getIsolationScope, getDefaultIsolationScope, debug, defineIntegration, captureException } from '@sentry/core'; import { generateInstrumentOnce, addOriginToSpan, ensureIsWrapped } from '@sentry/node-core'; import { DEBUG_BUILD } from '../../debug-build.js'; const INTEGRATION_NAME = 'Koa'; const instrumentKoa = generateInstrumentOnce( INTEGRATION_NAME, KoaInstrumentation, (options = {}) => { return { ignoreLayersType: options.ignoreLayersType , requestHook(span, info) { addOriginToSpan(span, 'auto.http.otel.koa'); const attributes = spanToJSON(span).data; // this is one of: middleware, router const type = attributes['koa.type']; if (type) { span.setAttribute(SEMANTIC_ATTRIBUTE_SENTRY_OP, `${type}.koa`); } // Also update the name const name = attributes['koa.name']; if (typeof name === 'string') { // Somehow, name is sometimes `''` for middleware spans // See: https://github.com/open-telemetry/opentelemetry-js-contrib/issues/2220 span.updateName(name || '< unknown >'); } if (getIsolationScope() === getDefaultIsolationScope()) { DEBUG_BUILD && debug.warn('Isolation scope is default isolation scope - skipping setting transactionName'); return; } const route = attributes[ATTR_HTTP_ROUTE]; // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access const method = info.context?.request?.method?.toUpperCase() || 'GET'; if (route) { getIsolationScope().setTransactionName(`${method} ${route}`); } }, } ; }, ); const _koaIntegration = ((options = {}) => { return { name: INTEGRATION_NAME, setupOnce() { instrumentKoa(options); }, }; }) ; /** * Adds Sentry tracing instrumentation for [Koa](https://koajs.com/). * * If you also want to capture errors, you need to call `setupKoaErrorHandler(app)` after you set up your Koa server. * * For more information, see the [koa documentation](https://docs.sentry.io/platforms/javascript/guides/koa/). * * @param {KoaOptions} options Configuration options for the Koa integration. * * @example * ```javascript * const Sentry = require('@sentry/node'); * * Sentry.init({ * integrations: [Sentry.koaIntegration()], * }) * ``` * * @example * ```javascript * // To ignore middleware spans * const Sentry = require('@sentry/node'); * * Sentry.init({ * integrations: [ * Sentry.koaIntegration({ * ignoreLayersType: ['middleware'] * }) * ], * }) * ``` */ const koaIntegration = defineIntegration(_koaIntegration); /** * Add an Koa error handler to capture errors to Sentry. * * The error handler must be before any other middleware and after all controllers. * * @param app The Express instances * @param options {ExpressHandlerOptions} Configuration options for the handler * * @example * ```javascript * const Sentry = require('@sentry/node'); * const Koa = require("koa"); * * const app = new Koa(); * * Sentry.setupKoaErrorHandler(app); * * // Add your routes, etc. * * app.listen(3000); * ``` */ // eslint-disable-next-line @typescript-eslint/no-explicit-any const setupKoaErrorHandler = (app) => { app.use(async (ctx, next) => { try { await next(); } catch (error) { captureException(error); throw error; } }); ensureIsWrapped(app.use, 'koa'); }; export { instrumentKoa, koaIntegration, setupKoaErrorHandler }; //# sourceMappingURL=koa.js.map