feat(opentelemetry): Add getters for root spans contexts

Author: EmrysMyrddin

packages/plugins/opentelemetry/src/plugin.ts

@@ -218,6 +218,11 @@ export type OpenTelemetryContextExtension = {
   openTelemetry: {
     tracer: Tracer;
     activeContext: () => Context;
+    httpContext: (request?: Request) => Context | undefined;
+    operationContext: (context?: unknown) => Context | undefined;
+    executionRequestContext: (
+      ExecutionRequest: ExecutionRequest,
+    ) => Context | undefined;
   };
 };
 
@@ -706,7 +711,20 @@ export function useOpenTelemetry(
       extendContext({
         openTelemetry: {
           tracer,
-          activeContext: () => getContext(state),
+          httpContext: (request) => {
+            const { forRequest } = request ? getState({ request }) : state;
+            return forRequest.otel?.root;
+          },
+          operationContext: (context) => {
+            const { forOperation } = context ? getState({ context }) : state;
+            return forOperation.otel?.root;
+          },
+          executionRequestContext: (executionRequest) => {
+            return getState({ executionRequest }).forSubgraphExecution.otel
+              ?.root;
+          },
+          activeContext: (contextMatcher?: Parameters<typeof getState>[0]) =>
+            getContext(contextMatcher ? getState(contextMatcher) : state),
         },
       });
     },

packages/plugins/opentelemetry/tests/useOpenTelemetry.spec.ts

@@ -481,11 +481,11 @@ describe('useOpenTelemetry', () => {
               const createSpan =
                 (name: string) =>
                 (
-                  matcher: Parameters<(typeof otelPlugin)['getOtelContext']>[0],
+                  matcher: Parameters<(typeof otelPlugin)['getActiveContext']>[0],
                 ) =>
                   otelPlugin
                     .getTracer()
-                    .startSpan(name, {}, otelPlugin.getOtelContext(matcher))
+                    .startSpan(name, {}, otelPlugin.getActiveContext(matcher))
                     .end();
 
               return [

packages/plugins/opentelemetry/tests/yoga.spec.ts

@@ -114,11 +114,11 @@ describe('useOpenTelemetry', () => {
               const createSpan =
                 (name: string) =>
                 (
-                  matcher: Parameters<(typeof otelPlugin)['getOtelContext']>[0],
+                  matcher: Parameters<(typeof otelPlugin)['getActiveContext']>[0],
                 ) =>
                   otelPlugin
                     .getTracer()
-                    .startSpan(name, {}, otelPlugin.getOtelContext(matcher))
+                    .startSpan(name, {}, otelPlugin.getActiveContext(matcher))
                     .end();
 
               return [