Transform watchQuery.fetch results

apollographql · Aug 11, 2016 · 9fc6bc0 · 9fc6bc0
1 parent 6d9e501
commit 9fc6bc0
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 11 deletions.
diff --git a/src/ObservableQuery.ts b/src/ObservableQuery.ts
@@ -104,7 +104,8 @@ export class ObservableQuery extends Observable<ApolloQueryResult> {
       return this.queryManager.fetchQuery(this.queryId, assign(this.options, {
         forceFetch: true,
         variables,
-      }) as WatchQueryOptions);
+      }) as WatchQueryOptions)
+      .then(result => this.queryManager.transformResult(result));
     };
 
     this.fetchMore = (fetchMoreOptions: WatchQueryOptions & FetchMoreOptions) => {

diff --git a/src/QueryManager.ts b/src/QueryManager.ts
@@ -601,6 +601,15 @@ export class QueryManager {
     };
   }
 
+  // Give the result transformer a chance to observe or modify result data before it is passed on.
+  public transformResult(result: ApolloQueryResult): ApolloQueryResult {
+    if (!this.resultTransformer) {
+      return result;
+    } else {
+      return this.resultTransformer(result);
+    }
+  }
+
   private collectResultBehaviorsFromUpdateQueries(
     updateQueries: MutationQueryReducersMap,
     mutationResult: Object,
@@ -936,15 +945,6 @@ export class QueryManager {
     return !comparator(this.queryResults[queryId], result);
   }
 
-  // Give the result transformer a chance to observe or modify result data before it is passed on.
-  private transformResult(result: ApolloQueryResult): ApolloQueryResult {
-    if (!this.resultTransformer) {
-      return result;
-    } else {
-      return this.resultTransformer(result);
-    }
-  }
-
   private broadcastQueries() {
     const queries = this.getApolloState().queries;
     forOwn(this.queryListeners, (listener: QueryListener, queryId: string) => {

diff --git a/typings/main/globals/node/index.d.ts b/typings/main/globals/node/index.d.ts
@@ -2298,4 +2298,4 @@ declare module "constants" {
     export var W_OK: number;
     export var X_OK: number;
     export var UV_UDP_REUSEADDR: number;
-}
+}