fix(tracing): automatically instrument server-side requests (#514)

nuxt-community · Mar 1, 2023 · 1d96f8b · 1d96f8b
1 parent 7933761
commit 1d96f8b
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 0 deletions.
diff --git a/lib/core/hooks.js b/lib/core/hooks.js
@@ -168,6 +168,9 @@ export async function initializeServerSentry (moduleContainer, moduleOptions, se
     Sentry.init(config)
     sentryHandlerProxy.errorHandler = Sentry.Handlers.errorHandler()
     sentryHandlerProxy.requestHandler = Sentry.Handlers.requestHandler(moduleOptions.requestHandlerConfig)
+    if (serverOptions.tracing) {
+      sentryHandlerProxy.tracingHandler = Sentry.Handlers.tracingHandler()
+    }
   }
 
   process.sentry = Sentry

diff --git a/lib/core/options.js b/lib/core/options.js
@@ -116,6 +116,8 @@ function resolveTracingOptions (options, config) {
     config.tracesSampleRate = tracingOptions.tracesSampleRate
   }
   options.tracing = tracingOptions
+  // Enable tracing for `Http` integration.
+  options.serverIntegrations = merge({ Http: { tracing: true } }, options.serverIntegrations || {})
 }
 
 /**

diff --git a/lib/module.js b/lib/module.js
@@ -109,9 +109,14 @@ export default async function SentryModule (moduleOptions) {
     const sentryHandlerProxy = {
       errorHandler: (error, req, res, next) => { next(error) },
       requestHandler: (req, res, next) => { next() },
+      tracingHandler: (req, res, next) => { next() },
     }
     // @ts-ignore
     this.nuxt.hook('render:setupMiddleware', app => app.use((req, res, next) => { sentryHandlerProxy.requestHandler(req, res, next) }))
+    if (options.tracing) {
+      // @ts-ignore
+      this.nuxt.hook('render:setupMiddleware', app => app.use((req, res, next) => { sentryHandlerProxy.tracingHandler(req, res, next) }))
+    }
     // @ts-ignore
     this.nuxt.hook('render:errorMiddleware', app => app.use((error, req, res, next) => { sentryHandlerProxy.errorHandler(error, req, res, next) }))
     // @ts-ignore

diff --git a/types/sentry.d.ts b/types/sentry.d.ts
@@ -10,6 +10,7 @@ import { NodeOptions, Handlers } from '@sentry/node'
 export interface SentryHandlerProxy {
     errorHandler: (error: any, req: IncomingMessage, res: ServerResponse, next: (error: any) => void) => void
     requestHandler: (req: IncomingMessage, res: ServerResponse, next: (error?: any) => void) => void
+    tracingHandler: (req: IncomingMessage, res: ServerResponse, next: (error?: any) => void) => void
 }
 
 export type IntegrationsConfiguration = Record<string, unknown>