diff --git a/packages/remix/src/utils/instrumentServer.ts b/packages/remix/src/utils/instrumentServer.ts index 96acde1db2aa..7fbff4bb6bd8 100644 --- a/packages/remix/src/utils/instrumentServer.ts +++ b/packages/remix/src/utils/instrumentServer.ts @@ -113,6 +113,7 @@ function makeWrappedDocumentRequestFunction( responseStatusCode: number, responseHeaders: Headers, context: Record, + loadContext?: Record, ): Promise { let res: Response; @@ -120,7 +121,7 @@ function makeWrappedDocumentRequestFunction( const currentScope = getCurrentHub().getScope(); if (!currentScope) { - return origDocumentRequestFunction.call(this, request, responseStatusCode, responseHeaders, context); + return origDocumentRequestFunction.call(this, request, responseStatusCode, responseHeaders, context, loadContext); } try { @@ -133,7 +134,14 @@ function makeWrappedDocumentRequestFunction( }, }); - res = await origDocumentRequestFunction.call(this, request, responseStatusCode, responseHeaders, context); + res = await origDocumentRequestFunction.call( + this, + request, + responseStatusCode, + responseHeaders, + context, + loadContext, + ); span?.finish(); } catch (err) { diff --git a/packages/remix/src/utils/types.ts b/packages/remix/src/utils/types.ts index e57b8ad15e34..642f6eef76cb 100644 --- a/packages/remix/src/utils/types.ts +++ b/packages/remix/src/utils/types.ts @@ -127,9 +127,13 @@ export interface ServerBuild { } export interface HandleDocumentRequestFunction { - (request: RemixRequest, responseStatusCode: number, responseHeaders: Headers, context: EntryContext): - | Promise - | Response; + ( + request: RemixRequest, + responseStatusCode: number, + responseHeaders: Headers, + context: EntryContext, + loadContext?: AppLoadContext, + ): Promise | Response; } export interface HandleDataRequestFunction {