Remove mutation tracking from the store

apollographql · Jul 5, 2017 · aea2442 · aea2442
1 parent ce2649a
commit aea2442
Show file tree

Hide file tree

Showing 7 changed files with 27 additions and 78 deletions.
diff --git a/src/core/QueryManager.ts b/src/core/QueryManager.ts
@@ -103,6 +103,10 @@ import {
   MutationQueryReducer,
 } from '../data/mutationResults';
 
+import {
+  MutationStore,
+} from '../mutations/store';
+
 import {
   QueryScheduler,
 } from '../scheduler/scheduler';
@@ -137,6 +141,7 @@ export class QueryManager {
   public store: ApolloStore;
   public networkInterface: NetworkInterface;
   public ssrMode: boolean;
+  public mutationStore: MutationStore = {};
 
   private addTypename: boolean;
   private deduplicator: Deduplicator;
@@ -309,6 +314,13 @@ export class QueryManager {
       update: updateWithProxyFn,
     });
 
+    this.mutationStore[mutationId] = {
+      mutationString: mutationString,
+      variables: variables || {},
+      loading: true,
+      error: null,
+    };
+
     return new Promise<ExecutionResult>((resolve, reject) => {
       this.networkInterface.query(request)
         .then((result) => {
@@ -322,6 +334,9 @@ export class QueryManager {
               mutationId,
             });
 
+            this.mutationStore[mutationId].loading = false;
+            this.mutationStore[mutationId].error = error;
+
             delete this.queryDocuments[mutationId];
             reject(error);
             return;
@@ -339,6 +354,9 @@ export class QueryManager {
             update: updateWithProxyFn,
           });
 
+          this.mutationStore[mutationId].loading = false;
+          this.mutationStore[mutationId].error = null;
+
           // If there was an error in our reducers, reject this promise!
           const { reducerError } = this.getApolloState();
           if (reducerError && reducerError.mutationId === mutationId) {
@@ -833,6 +851,8 @@ export class QueryManager {
       observableQueryIds: Object.keys(this.observableQueries),
     });
 
+    this.mutationStore = {};
+
     // Similarly, we have to have to refetch each of the queries currently being
     // observed. We refetch instead of error'ing on these since the assumption is that
     // resetting the store doesn't eliminate the need for the queries currently being

diff --git a/src/mutations/store.ts b/src/mutations/store.ts
@@ -1,15 +1,3 @@
-import {
-  ApolloAction,
-  isMutationInitAction,
-  isMutationResultAction,
-  isMutationErrorAction,
-  isStoreResetAction,
-} from '../actions';
-
-import {
-  SelectionSetNode,
-} from 'graphql';
-
 export interface MutationStore {
   [mutationId: string]: MutationStoreValue;
 }
@@ -21,50 +9,3 @@ export interface MutationStoreValue {
   error: Error | null;
 }
 
-export interface SelectionSetWithRoot {
-  id: string;
-  typeName: string;
-  selectionSet: SelectionSetNode;
-}
-
-export function mutations(
-  previousState: MutationStore = {},
-  action: ApolloAction,
-): MutationStore {
-  if (isMutationInitAction(action)) {
-    const newState = { ...previousState } as MutationStore;
-
-    newState[action.mutationId] = {
-      mutationString: action.mutationString,
-      variables: action.variables,
-      loading: true,
-      error: null,
-    };
-
-    return newState;
-  } else if (isMutationResultAction(action)) {
-    const newState = { ...previousState } as MutationStore;
-
-    newState[action.mutationId] = {
-      ...previousState[action.mutationId],
-      loading: false,
-      error: null,
-    } as MutationStoreValue;
-
-    return newState;
-  } else if (isMutationErrorAction(action)) {
-    const newState = { ...previousState } as MutationStore;
-
-    newState[action.mutationId] = {
-      ...previousState[action.mutationId],
-      loading: false,
-      error: action.error,
-    } as MutationStoreValue;
-  } else if (isStoreResetAction(action)) {
-    // if we are resetting the store, we no longer need information about the mutations
-    // that are currently in the store so we can just throw them all away.
-    return {};
-  }
-
-  return previousState;
-}
diff --git a/src/store.ts b/src/store.ts
@@ -24,7 +24,7 @@ import {
 } from './queries/store';
 
 import {
-  mutations,
+  // mutations,
   MutationStore,
 } from './mutations/store';
 
@@ -62,7 +62,6 @@ export interface ReducerError {
 export interface Store {
   data: NormalizedCache;
   queries: QueryStore;
-  mutations: MutationStore;
   optimistic: OptimisticStore;
   reducerError: ReducerError | null;
 }
@@ -110,7 +109,6 @@ export function createApolloReducer(config: ApolloReducerConfig): (state: Store,
     try {
       const newState: Store = {
         queries: queries(state.queries, action),
-        mutations: mutations(state.mutations, action),
 
         data: data(state.data, action, config),
         optimistic: [] as any,
@@ -134,7 +132,6 @@ export function createApolloReducer(config: ApolloReducerConfig): (state: Store,
       );
 
       if (state.data === newState.data &&
-      state.mutations === newState.mutations &&
       state.queries === newState.queries &&
       state.optimistic === newState.optimistic &&
       state.reducerError === newState.reducerError) {

diff --git a/test/QueryManager.ts b/test/QueryManager.ts
@@ -2442,7 +2442,6 @@ describe('QueryManager', () => {
       const currentState = queryManager.getApolloState();
       const expectedState: any = {
         data: {},
-        mutations: {},
         queries: {},
         optimistic: [],
         reducerError: null,

diff --git a/test/client.ts b/test/client.ts
@@ -182,7 +182,6 @@ describe('client', () => {
       {
         apollo: {
           queries: {},
-          mutations: {},
           data: {},
           optimistic: [],
           reducerError: null,
@@ -585,7 +584,6 @@ describe('client', () => {
           metadata: null,
         },
       },
-      mutations: {},
       reducerError: null,
     }) };
 

diff --git a/test/optimistic.ts b/test/optimistic.ts
@@ -318,7 +318,7 @@ describe('optimistic mutation results', () => {
             updateQueries,
           }).then((res) => {
             checkBothMutationsAreApplied('This one was created with a mutation.', 'Optimistically generated 2');
-            const mutationsState = client.store.getState().apollo.mutations;
+            const mutationsState = client.queryManager.mutationStore;
             assert.equal(mutationsState['5'].loading, false);
             assert.equal(mutationsState['6'].loading, true);
 
@@ -331,14 +331,14 @@ describe('optimistic mutation results', () => {
             updateQueries,
           }).then((res) => {
             checkBothMutationsAreApplied('This one was created with a mutation.', 'Second mutation.');
-            const mutationsState = client.store.getState().apollo.mutations;
+            const mutationsState = client.queryManager.mutationStore;
             assert.equal(mutationsState[5].loading, false);
             assert.equal(mutationsState[6].loading, false);
 
             return res;
           });
 
-          const mutationsState = client.store.getState().apollo.mutations;
+          const mutationsState = client.queryManager.mutationStore;
           assert.equal(mutationsState[5].loading, true);
           assert.equal(mutationsState[6].loading, true);
 
@@ -486,7 +486,7 @@ describe('optimistic mutation results', () => {
             update,
           }).then((res) => {
             checkBothMutationsAreApplied('This one was created with a mutation.', 'Optimistically generated 2');
-            const mutationsState = client.store.getState().apollo.mutations;
+            const mutationsState = client.queryManager.mutationStore;
             assert.equal(mutationsState['5'].loading, false);
             assert.equal(mutationsState['6'].loading, true);
 
@@ -499,14 +499,14 @@ describe('optimistic mutation results', () => {
             update,
           }).then((res) => {
             checkBothMutationsAreApplied('This one was created with a mutation.', 'Second mutation.');
-            const mutationsState = client.store.getState().apollo.mutations;
+            const mutationsState = client.queryManager.mutationStore;
             assert.equal(mutationsState[5].loading, false);
             assert.equal(mutationsState[6].loading, false);
 
             return res;
           });
 
-          const mutationsState = client.store.getState().apollo.mutations;
+          const mutationsState = client.queryManager.mutationStore;
           assert.equal(mutationsState[5].loading, true);
           assert.equal(mutationsState[6].loading, true);
 

diff --git a/test/store.ts b/test/store.ts
@@ -22,7 +22,6 @@ describe('createApolloStore', () => {
       store.getState()['apollo'],
       {
         queries: {},
-        mutations: {},
         data: {},
         optimistic: [],
         reducerError: null,
@@ -39,7 +38,6 @@ describe('createApolloStore', () => {
       store.getState()['test'],
       {
         queries: {},
-        mutations: {},
         data: {},
         optimistic: [],
         reducerError: null,
@@ -64,7 +62,6 @@ describe('createApolloStore', () => {
     assert.deepEqual(store.getState(), {
       apollo: {
         queries: {},
-        mutations: {},
         data: initialState.apollo.data,
         optimistic: initialState.apollo.optimistic,
         reducerError: null,
@@ -119,7 +116,6 @@ describe('createApolloStore', () => {
 
     const emptyState: Store = {
       queries: { },
-      mutations: { },
       data: { },
       optimistic: ([] as any[]),
       reducerError: null,
@@ -163,7 +159,6 @@ describe('createApolloStore', () => {
           'metadata': null,
         },
       },
-      mutations: {},
       data: {},
       optimistic: ([] as any[]),
       reducerError: null,
@@ -238,7 +233,6 @@ describe('createApolloStore', () => {
 
     const resetState = {
       queries: {},
-      mutations: {},
       data: {},
       optimistic: [
         {