Use ModuleImport module providers in runtime

Reviewed By: voideanvalue Differential Revision: D38007238 fbshipit-source-id: b405e4c122e365f46371f1896f48d230cb3a728f
facebook · Jul 25, 2022 · 4d92104 · 4d92104
1 parent 3c94dec
commit 4d92104
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/packages/relay-runtime/index.js b/packages/relay-runtime/index.js
@@ -173,6 +173,7 @@ export type {
   NormalizationLinkedHandle,
   NormalizationLocalArgumentDefinition,
   NormalizationModuleImport,
+  NormalizationRootNode,
   NormalizationScalarField,
   NormalizationSelection,
   NormalizationSplitOperation,

diff --git a/packages/relay-runtime/store/RelayResponseNormalizer.js b/packages/relay-runtime/store/RelayResponseNormalizer.js
@@ -444,14 +444,16 @@ class RelayResponseNormalizer {
     );
     const typeName: string = RelayModernRecord.getType(record);
     const componentKey = getModuleComponentKey(moduleImport.documentName);
-    const componentReference = data[componentKey];
+    const componentReference =
+      moduleImport.componentModuleProvider || data[componentKey];
     RelayModernRecord.setValue(
       record,
       componentKey,
       componentReference ?? null,
     );
     const operationKey = getModuleOperationKey(moduleImport.documentName);
-    const operationReference = data[operationKey];
+    const operationReference =
+      moduleImport.operationModuleProvider || data[operationKey];
     RelayModernRecord.setValue(
       record,
       operationKey,

diff --git a/packages/relay-runtime/util/NormalizationNode.js b/packages/relay-runtime/util/NormalizationNode.js
@@ -11,6 +11,7 @@
 'use strict';
 
 import type {ConcreteRequest} from './RelayConcreteNode';
+import type {JSResourceReference} from 'JSResourceReference';
 
 /**
  * Represents a single operation used to processing and normalize runtime
@@ -114,6 +115,14 @@ export type NormalizationModuleImport = {
   +documentName: string,
   +fragmentPropName: string,
   +fragmentName: string,
+  +componentModuleProvider?: () =>
+    | mixed
+    | Promise<mixed>
+    | JSResourceReference<mixed>,
+  +operationModuleProvider?: () =>
+    | NormalizationRootNode
+    | Promise<NormalizationRootNode>
+    | JSResourceReference<NormalizationRootNode>,
 };
 
 export type NormalizationListValueArgument = {