Add React Apollo changes from PR #3515

apollographql/react-apollo#3515
apollographql · Oct 2, 2019 · 7e87e38 · 7e87e38
1 parent 0b028a1
commit 7e87e38
Show file tree

Hide file tree

Showing 4 changed files with 51 additions and 46 deletions.
diff --git a/src/react/data/OperationData.ts b/src/react/data/OperationData.ts
@@ -26,8 +26,11 @@ export abstract class OperationData<TOptions = any> {
     return this.options;
   }
 
-  public setOptions(newOptions: CommonOptions<TOptions>) {
-    if (!isEqual(this.options, newOptions)) {
+  public setOptions(
+    newOptions: CommonOptions<TOptions>,
+    storePrevious: boolean = false
+  ) {
+    if (storePrevious && !isEqual(this.options, newOptions)) {
       this.previousOptions = this.options;
     }
     this.options = newOptions;

diff --git a/src/react/data/QueryData.ts b/src/react/data/QueryData.ts
@@ -48,7 +48,7 @@ export class QueryData<TData, TVariables> extends OperationData {
   public execute(): QueryResult<TData, TVariables> {
     this.refreshClient();
 
-    const { skip, query, ssr } = this.getOptions();
+    const { skip, query } = this.getOptions();
     if (skip || query !== this.previousData.query) {
       this.removeQuerySubscription();
       this.previousData.query = query;
@@ -58,9 +58,7 @@ export class QueryData<TData, TVariables> extends OperationData {
 
     if (this.isMounted) this.startQuerySubscription();
 
-    const ssrDisabled = ssr === false;
-
-    return this.getExecuteSsrResult(ssrDisabled) || this.getExecuteResult();
+    return this.getExecuteSsrResult() || this.getExecuteResult();
   }
 
   public executeLazy(): QueryTuple<TData, TVariables> {
@@ -90,6 +88,7 @@ export class QueryData<TData, TVariables> extends OperationData {
 
   public afterExecute({ lazy = false }: { lazy?: boolean } = {}) {
     this.isMounted = true;
+
     if (!lazy || this.runLazy) {
       this.handleErrorOrCompleted();
 
@@ -103,6 +102,7 @@ export class QueryData<TData, TVariables> extends OperationData {
       });
     }
 
+    this.previousOptions = this.getOptions();
     return this.unmount.bind(this);
   }
 
@@ -114,25 +114,24 @@ export class QueryData<TData, TVariables> extends OperationData {
 
   public getOptions() {
     const options = super.getOptions();
-    const lazyOptions = this.lazyOptions || {};
-    const updatedOptions = {
-      ...options,
-      variables: {
+
+    if (this.lazyOptions) {
+      options.variables = {
         ...options.variables,
-        ...lazyOptions.variables
-      },
-      context: {
+        ...this.lazyOptions.variables
+      };
+      options.context = {
         ...options.context,
-        ...lazyOptions.context
-      }
-    };
+        ...this.lazyOptions.context
+      };
+    }
 
     // skip is not supported when using lazy query execution.
     if (this.runLazy) {
-      delete updatedOptions.skip;
+      delete options.skip;
     }
 
-    return updatedOptions;
+    return options;
   }
 
   private runLazyQuery = (options?: QueryLazyOptions<TVariables>) => {
@@ -149,29 +148,31 @@ export class QueryData<TData, TVariables> extends OperationData {
     return result;
   };
 
-  private getExecuteSsrResult(ssrDisabled: boolean) {
-    let result;
-
-    if (this.context && this.context.renderPromises) {
-      const ssrLoading = {
-        loading: true,
-        networkStatus: NetworkStatus.loading,
-        called: true,
-        data: undefined
-      } as QueryResult<TData, TVariables>;
-
-      // SSR is disabled, so just return the loading event and leave it in that state.
-      if (ssrDisabled) {
-        return ssrLoading;
-      }
+  private getExecuteSsrResult() {
+    const treeRenderingInitiated = this.context && this.context.renderPromises;
+    const ssrDisabled = this.getOptions().ssr === false;
+    const fetchDisabled = this.refreshClient().client.disableNetworkFetches;
+
+    const ssrLoading = {
+      loading: true,
+      networkStatus: NetworkStatus.loading,
+      called: true,
+      data: undefined
+    } as QueryResult<TData, TVariables>;
+
+    // If SSR has been explicitly disabled, and this function has been called
+    // on the server side, return the default loading state.
+    if (ssrDisabled && (treeRenderingInitiated || fetchDisabled)) {
+      return ssrLoading;
+    }
 
-      result = this.context.renderPromises.addQueryPromise(
-        this,
-        this.getExecuteResult
-      );
-      if (!result) {
-        result = ssrLoading as QueryResult<TData, TVariables>;
-      }
+    let result;
+    if (treeRenderingInitiated) {
+      result =
+        this.context.renderPromises!.addQueryPromise(
+          this,
+          this.getExecuteResult
+        ) || ssrLoading;
     }
 
     return result;
@@ -196,7 +197,7 @@ export class QueryData<TData, TVariables> extends OperationData {
     return {
       ...options,
       displayName,
-      context: options.context || {},
+      context: options.context,
       metadata: { reactComponent: { displayName } }
     };
   }
@@ -213,13 +214,14 @@ export class QueryData<TData, TVariables> extends OperationData {
 
     if (!this.currentObservable.query) {
       const observableQueryOptions = this.prepareObservableQueryOptions();
+
       this.previousData.observableQueryOptions = {
         ...observableQueryOptions,
         children: null
       };
-      this.currentObservable.query = this.refreshClient().client.watchQuery(
-        observableQueryOptions
-      );
+      this.currentObservable.query = this.refreshClient().client.watchQuery({
+        ...observableQueryOptions
+      });
 
       if (this.context && this.context.renderPromises) {
         this.context.renderPromises.registerSSRObservable(

diff --git a/src/react/hooks/__tests__/useLazyQuery.test.tsx b/src/react/hooks/__tests__/useLazyQuery.test.tsx
@@ -417,7 +417,7 @@ describe('useLazyQuery Hook', () => {
           expect(loading).toEqual(false);
           expect(data).toEqual(CAR_RESULT_DATA);
           setTimeout(() => {
-            execute();
+            execute({ variables: { someProp: 'someValue' } });
           });
           break;
         case 3:

diff --git a/src/react/hooks/useSubscription.ts b/src/react/hooks/useSubscription.ts
@@ -33,7 +33,7 @@ export function useSubscription<TData = any, TVariables = OperationVariables>(
   }
 
   const subscriptionData = getSubscriptionDataRef();
-  subscriptionData.setOptions(updatedOptions);
+  subscriptionData.setOptions(updatedOptions, true);
   subscriptionData.context = context;
 
   useEffect(() => subscriptionData.afterExecute());