diff --git a/packages/core/helpers/context-id-factory.ts b/packages/core/helpers/context-id-factory.ts index 0cb80d3117f..99506220bfd 100644 --- a/packages/core/helpers/context-id-factory.ts +++ b/packages/core/helpers/context-id-factory.ts @@ -27,16 +27,19 @@ export class ContextIdFactory { */ public static getByRequest = any>( request: T, + propsToInspect: string[] = ['raw'], ): ContextId { if (!request) { - return createContextId(); + return ContextIdFactory.create(); } if (request[REQUEST_CONTEXT_ID as any]) { return request[REQUEST_CONTEXT_ID as any]; } - if (request.raw && request.raw[REQUEST_CONTEXT_ID]) { - return request.raw[REQUEST_CONTEXT_ID]; + for (const key of propsToInspect) { + if (request[key]?.[REQUEST_CONTEXT_ID]) { + return request[key][REQUEST_CONTEXT_ID]; + } } - return createContextId(); + return ContextIdFactory.create(); } }