[react-relay]: Export ProfilerContext

To match facebook/relay#4049
tomgasson · Sep 27, 2022 · 3dc25b3 · 3dc25b3
1 parent e1b1a23
commit 3dc25b3
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/types/react-relay/hooks.d.ts b/types/react-relay/hooks.d.ts
@@ -62,6 +62,8 @@ export { requestSubscription } from 'relay-runtime';
 export { RelayEnvironmentProvider } from './relay-hooks/RelayEnvironmentProvider.react';
 export { EntryPointContainer } from './relay-hooks/EntryPointContainer.react';
 
+export { ProfilerContext } from './relay-hooks/ProfilerContext';
+
 export { fetchQuery } from 'relay-runtime';
 
 export { loadQuery } from './relay-hooks/loadQuery';

diff --git a/types/react-relay/test/react-relay-tests.tsx b/types/react-relay/test/react-relay-tests.tsx
@@ -26,6 +26,7 @@ import {
     RelayProp,
     RelayRefetchProp,
     requestSubscription,
+    ProfilerContext,
 } from 'react-relay';
 
 // ~~~~~~~~~~~~~~~~~~~~~
@@ -759,3 +760,17 @@ const MyRelayContextConsumer: React.FunctionComponent = () => {
         />
     );
 };
+
+const MyRelayProfilerContextProvider: React.FunctionComponent = () => {
+    const context = React.useMemo(
+        () => ({
+            wrapPrepareQueryResource<T>(cb: () => T): T {
+                return cb();
+            },
+        }),
+        [],
+    );
+    return (
+        <ProfilerContext.Provider value={context}><div /></ProfilerContext.Provider>
+    );
+};