Skip to content

feat(react-router): Add otel instrumentation for server requests #16147

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 13 commits into
base: develop
Choose a base branch
from
Original file line number Diff line number Diff line change
Expand Up @@ -15,5 +15,6 @@ export default [
route('ssr', 'routes/performance/ssr.tsx'),
route('with/:param', 'routes/performance/dynamic-param.tsx'),
route('static', 'routes/performance/static.tsx'),
route('server-loader', 'routes/performance/server-loader.tsx'),
]),
] satisfies RouteConfig;
Original file line number Diff line number Diff line change
@@ -1,5 +1,10 @@
import type { Route } from './+types/dynamic-param';

export async function loader() {
await new Promise(resolve => setTimeout(resolve, 500));
return { data: 'burritos' };
}

export default function DynamicParamPage({ params }: Route.ComponentProps) {
const { param } = params;

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@ export default function PerformancePage() {
<nav>
<Link to="/performance/ssr">SSR Page</Link>
<Link to="/performance/with/sentry">With Param Page</Link>
<Link to="/performance/server-loader">Server Loader</Link>
</nav>
</div>
);
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
import type { Route } from './+types/server-loader';

export async function loader() {
await new Promise(resolve => setTimeout(resolve, 500));
return { data: 'burritos' };
}

export default function ServerLoaderPage({ loaderData }: Route.ComponentProps) {
const { data } = loaderData;
return (
<div>
<h1>Server Loader Page</h1>
<div>{data}</div>
</div>
);
}
Original file line number Diff line number Diff line change
Expand Up @@ -104,4 +104,32 @@ test.describe('servery - performance', () => {
},
});
});

test('should automatically instrument server loader', async ({ page }) => {
const txPromise = waitForTransaction(APP_NAME, async transactionEvent => {
return transactionEvent.transaction === 'GET /performance/server-loader.data';
});

await page.goto(`/performance`); // initial ssr pageloads do not contain .data requests
await page.waitForTimeout(500); // quick breather before navigation
await page.getByRole('link', { name: 'Server Loader' }).click(); // this will actually trigger a .data request

const transaction = await txPromise;

expect(transaction?.spans?.[transaction.spans?.length - 1]).toMatchObject({
span_id: expect.any(String),
trace_id: expect.any(String),
data: {
'sentry.origin': 'auto.http.react-router',
'sentry.op': 'function.react-router.loader',
},
description: 'Executing Server Loader',
parent_span_id: expect.any(String),
start_timestamp: expect.any(Number),
timestamp: expect.any(Number),
status: 'ok',
op: 'function.react-router.loader',
origin: 'auto.http.react-router',
});
});
});
1 change: 1 addition & 0 deletions packages/react-router/package.json
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@
"@opentelemetry/api": "^1.9.0",
"@opentelemetry/core": "^1.30.1",
"@opentelemetry/semantic-conventions": "^1.30.0",
"@opentelemetry/instrumentation": "0.57.2",
"@sentry/browser": "9.17.0",
"@sentry/cli": "^2.43.0",
"@sentry/core": "9.17.0",
Expand Down
111 changes: 111 additions & 0 deletions packages/react-router/src/server/instrumentation/reactRouter.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,111 @@
import type { InstrumentationConfig } from '@opentelemetry/instrumentation';
import { InstrumentationBase, InstrumentationNodeModuleDefinition } from '@opentelemetry/instrumentation';
import {
getActiveSpan,
getRootSpan,
logger,
SDK_VERSION,
SEMANTIC_ATTRIBUTE_SENTRY_OP,
SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN,
SEMANTIC_ATTRIBUTE_SENTRY_SOURCE,
startSpan,
} from '@sentry/core';
import type * as reactRouter from 'react-router';
import { DEBUG_BUILD } from '../../common/debug-build';
import { getOpName, getSpanName, isDataRequest, SEMANTIC_ATTRIBUTE_SENTRY_OVERWRITE } from './util';

type ReactRouterModuleExports = typeof reactRouter;

const supportedVersions = ['>=7.0.0'];
const COMPONENT = 'react-router';

/**
* Instrumentation for React Router's server request handler.
* This patches the requestHandler function to add Sentry performance monitoring for data loaders.
*/
export class ReactRouterInstrumentation extends InstrumentationBase<InstrumentationConfig> {
public constructor(config: InstrumentationConfig = {}) {
super('ReactRouterInstrumentation', SDK_VERSION, config);
}

/**
* Initializes the instrumentation by defining the React Router server modules to be patched.
*/
// eslint-disable-next-line @typescript-eslint/naming-convention
protected init(): InstrumentationNodeModuleDefinition {
const reactRouterServerModule = new InstrumentationNodeModuleDefinition(
COMPONENT,
supportedVersions,
(moduleExports: ReactRouterModuleExports) => {
return this._createPatchedModuleProxy(moduleExports);
},
(_moduleExports: unknown) => {
// nothing to unwrap here
return _moduleExports;
},
);

return reactRouterServerModule;
}

/**
* Creates a proxy around the React Router module exports that patches the createRequestHandler function.
* This allows us to wrap the request handler to add performance monitoring for data loaders and actions.
*/
private _createPatchedModuleProxy(moduleExports: ReactRouterModuleExports): ReactRouterModuleExports {
return new Proxy(moduleExports, {
get(target, prop, receiver) {
if (prop === 'createRequestHandler') {
const original = target[prop];
return function wrappedCreateRequestHandler(this: unknown, ...args: unknown[]) {
const originalRequestHandler = original.apply(this, args);

return async function wrappedRequestHandler(request: Request, initialContext?: unknown) {
let url: URL;
try {
url = new URL(request.url);
} catch (error) {
return originalRequestHandler(request, initialContext);
}

// We currently just want to trace loaders and actions
if (!isDataRequest(url.pathname)) {
return originalRequestHandler(request, initialContext);
}

const activeSpan = getActiveSpan();
const rootSpan = activeSpan && getRootSpan(activeSpan);

if (!rootSpan) {
DEBUG_BUILD && logger.debug('No active root span found, skipping tracing for data request');
return originalRequestHandler(request, initialContext);
}

// Set the source and overwrite attributes on the root span to ensure the transaction name
// is derived from the raw URL pathname rather than any parameterized route that may be set later
// TODO: try to set derived parameterized route from build here (args[0])
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

want to ship this in a follow-up

rootSpan.setAttributes({
[SEMANTIC_ATTRIBUTE_SENTRY_SOURCE]: 'url',
[SEMANTIC_ATTRIBUTE_SENTRY_OVERWRITE]: `${request.method} ${url.pathname}`,
});

return startSpan(
{
name: getSpanName(url.pathname, request.method),
attributes: {
[SEMANTIC_ATTRIBUTE_SENTRY_ORIGIN]: 'auto.http.react-router',
[SEMANTIC_ATTRIBUTE_SENTRY_OP]: getOpName(url.pathname, request.method),
},
},
() => {
return originalRequestHandler(request, initialContext);
},
);
};
};
}
return Reflect.get(target, prop, receiver);
},
});
}
}
53 changes: 53 additions & 0 deletions packages/react-router/src/server/instrumentation/util.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,53 @@
/**
* Gets the op name for a request based on whether it's a loader or action request.
* @param pathName The URL pathname to check
* @param requestMethod The HTTP request method
*/
export function getOpName(pathName: string, requestMethod: string): string {
return isLoaderRequest(pathName, requestMethod)
? 'function.react-router.loader'
: isActionRequest(pathName, requestMethod)
? 'function.react-router.action'
: 'function.react-router';
}

/**
* Gets the span name for a request based on whether it's a loader or action request.
* @param pathName The URL pathname to check
* @param requestMethod The HTTP request method
*/
export function getSpanName(pathName: string, requestMethod: string): string {
return isLoaderRequest(pathName, requestMethod)
? 'Executing Server Loader'
: isActionRequest(pathName, requestMethod)
? 'Executing Server Action'
: 'Unknown Data Request';
}

/**
* Checks if the request is a server loader request
* @param pathname The URL pathname to check
* @param requestMethod The HTTP request method
*/
export function isLoaderRequest(pathname: string, requestMethod: string): boolean {
return isDataRequest(pathname) && requestMethod === 'GET';
}

/**
* Checks if the request is a server action request
* @param pathname The URL pathname to check
* @param requestMethod The HTTP request method
*/
export function isActionRequest(pathname: string, requestMethod: string): boolean {
return isDataRequest(pathname) && requestMethod === 'POST';
}

/**
* Checks if the request is a react-router data request
* @param pathname The URL pathname to check
*/
export function isDataRequest(pathname: string): boolean {
return pathname.endsWith('.data');
}

export const SEMANTIC_ATTRIBUTE_SENTRY_OVERWRITE = 'sentry.overwrite-route';
28 changes: 28 additions & 0 deletions packages/react-router/src/server/integration/reactRouterServer.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,28 @@
import { defineIntegration } from '@sentry/core';
import { generateInstrumentOnce } from '@sentry/node';
import { ReactRouterInstrumentation } from '../instrumentation/reactRouter';

const INTEGRATION_NAME = 'ReactRouterServer';

const instrumentReactRouter = generateInstrumentOnce('React-Router-Server', () => {
return new ReactRouterInstrumentation();
});

export const instrumentReactRouterServer = Object.assign(
(): void => {
instrumentReactRouter();
},
{ id: INTEGRATION_NAME },
);

/**
* Integration capturing tracing data for React Router server functions.
*/
export const reactRouterServerIntegration = defineIntegration(() => {
return {
name: INTEGRATION_NAME,
setupOnce() {
instrumentReactRouterServer();
},
};
});
47 changes: 41 additions & 6 deletions packages/react-router/src/server/sdk.ts
Original file line number Diff line number Diff line change
@@ -1,21 +1,32 @@
import type { Integration } from '@sentry/core';
import { applySdkMetadata, logger, setTag } from '@sentry/core';
import { ATTR_HTTP_ROUTE } from '@opentelemetry/semantic-conventions';
import type { EventProcessor, Integration } from '@sentry/core';
import { applySdkMetadata, getGlobalScope, logger, setTag } from '@sentry/core';
import type { NodeClient, NodeOptions } from '@sentry/node';
import { getDefaultIntegrations as getNodeDefaultIntegrations, init as initNodeSdk } from '@sentry/node';
import { DEBUG_BUILD } from '../common/debug-build';
import { SEMANTIC_ATTRIBUTE_SENTRY_OVERWRITE } from './instrumentation/util';
import { reactRouterServerIntegration } from './integration/reactRouterServer';
import { lowQualityTransactionsFilterIntegration } from './lowQualityTransactionsFilterIntegration';

function getDefaultIntegrations(options: NodeOptions): Integration[] {
return [...getNodeDefaultIntegrations(options), lowQualityTransactionsFilterIntegration(options)];
/**
* Returns the default integrations for the React Router SDK.
* @param options The options for the SDK.
*/
export function getDefaultReactRouterServerIntegrations(options: NodeOptions): Integration[] {
return [
...getNodeDefaultIntegrations(options),
lowQualityTransactionsFilterIntegration(options),
reactRouterServerIntegration(),
];
}

/**
* Initializes the server side of the React Router SDK
*/
export function init(options: NodeOptions): NodeClient | undefined {
const opts = {
const opts: NodeOptions = {
...options,
defaultIntegrations: getDefaultIntegrations(options),
defaultIntegrations: getDefaultReactRouterServerIntegrations(options),
};

DEBUG_BUILD && logger.log('Initializing SDK...');
Expand All @@ -26,6 +37,30 @@ export function init(options: NodeOptions): NodeClient | undefined {

setTag('runtime', 'node');

// Overwrite the transaction name for instrumented data loaders because the trace data gets overwritten at a later point.
// We only update the tx in case SEMANTIC_ATTRIBUTE_SENTRY_OVERWRITE got set in our instrumentation before.
getGlobalScope().addEventProcessor(
Object.assign(
(event => {
const overwrite = event.contexts?.trace?.data?.[SEMANTIC_ATTRIBUTE_SENTRY_OVERWRITE];
if (
event.type === 'transaction' &&
event.transaction === 'GET *' &&
event.contexts?.trace?.data?.[ATTR_HTTP_ROUTE] === '*' &&
overwrite
) {
event.transaction = overwrite;
delete event.contexts?.trace?.data?.[SEMANTIC_ATTRIBUTE_SENTRY_OVERWRITE];
event.contexts.trace.data[ATTR_HTTP_ROUTE] = 'url';
return event;
} else {
return event;
}
}) satisfies EventProcessor,
{ id: 'ReactRouterTransactionEnhancer' },
),
);

DEBUG_BUILD && logger.log('SDK successfully initialized');

return client;
Expand Down
11 changes: 10 additions & 1 deletion packages/react-router/src/server/wrapSentryHandleRequest.ts
Original file line number Diff line number Diff line change
@@ -1,7 +1,13 @@
import { context } from '@opentelemetry/api';
import { getRPCMetadata, RPCType } from '@opentelemetry/core';
import { ATTR_HTTP_ROUTE } from '@opentelemetry/semantic-conventions';
import { getActiveSpan, getRootSpan, getTraceMetaTags, SEMANTIC_ATTRIBUTE_SENTRY_SOURCE } from '@sentry/core';
import {
getActiveSpan,
getRootSpan,
getTraceMetaTags,
SEMANTIC_ATTRIBUTE_SENTRY_CUSTOM_SPAN_NAME,
SEMANTIC_ATTRIBUTE_SENTRY_SOURCE,
} from '@sentry/core';
import type { AppLoadContext, EntryContext } from 'react-router';
import type { PassThrough } from 'stream';
import { Transform } from 'stream';
Expand Down Expand Up @@ -30,6 +36,7 @@ export function wrapSentryHandleRequest(originalHandle: OriginalHandleRequest):
) {
const parameterizedPath =
routerContext?.staticHandlerContext?.matches?.[routerContext.staticHandlerContext.matches.length - 1]?.route.path;

if (parameterizedPath) {
const activeSpan = getActiveSpan();
if (activeSpan) {
Expand All @@ -38,6 +45,7 @@ export function wrapSentryHandleRequest(originalHandle: OriginalHandleRequest):

// The express instrumentation writes on the rpcMetadata and that ends up stomping on the `http.route` attribute.
const rpcMetadata = getRPCMetadata(context.active());

if (rpcMetadata?.type === RPCType.HTTP) {
rpcMetadata.route = routeName;
}
Expand All @@ -46,6 +54,7 @@ export function wrapSentryHandleRequest(originalHandle: OriginalHandleRequest):
rootSpan.setAttributes({
[ATTR_HTTP_ROUTE]: routeName,
[SEMANTIC_ATTRIBUTE_SENTRY_SOURCE]: 'route',
[SEMANTIC_ATTRIBUTE_SENTRY_CUSTOM_SPAN_NAME]: `${request.method} ${routeName}`,
});
}
}
Expand Down
Loading
Loading