Merge the Index Engines

Author: schmidt-sebastian

firebase-firestore/src/androidTest/java/com/google/firebase/firestore/remote/RemoteStoreTest.java

@@ -21,11 +21,11 @@
 import com.google.firebase.database.collection.ImmutableSortedSet;
 import com.google.firebase.firestore.auth.User;
 import com.google.firebase.firestore.core.OnlineState;
-import com.google.firebase.firestore.local.DefaultQueryEngine;
 import com.google.firebase.firestore.local.IndexBackfiller;
 import com.google.firebase.firestore.local.LocalStore;
 import com.google.firebase.firestore.local.MemoryPersistence;
 import com.google.firebase.firestore.local.Persistence;
+import com.google.firebase.firestore.local.QueryEngine;
 import com.google.firebase.firestore.model.DocumentKey;
 import com.google.firebase.firestore.model.mutation.MutationBatchResult;
 import com.google.firebase.firestore.testutil.IntegrationTestUtil;
@@ -76,7 +76,7 @@ public ImmutableSortedSet<DocumentKey> getRemoteKeysForTarget(int targetId) {
         };
 
     FakeConnectivityMonitor connectivityMonitor = new FakeConnectivityMonitor();
-    DefaultQueryEngine queryEngine = new DefaultQueryEngine();
+    QueryEngine queryEngine = new QueryEngine();
     Persistence persistence = MemoryPersistence.createEagerGcMemoryPersistence();
     persistence.start();
     IndexBackfiller indexBackfiller = new IndexBackfiller(persistence, new AsyncQueue());

firebase-firestore/src/main/java/com/google/firebase/firestore/core/MemoryComponentProvider.java

@@ -16,11 +16,11 @@
 
 import androidx.annotation.Nullable;
 import com.google.firebase.database.collection.ImmutableSortedSet;
-import com.google.firebase.firestore.local.DefaultQueryEngine;
 import com.google.firebase.firestore.local.IndexBackfiller;
 import com.google.firebase.firestore.local.LocalStore;
 import com.google.firebase.firestore.local.MemoryPersistence;
 import com.google.firebase.firestore.local.Persistence;
+import com.google.firebase.firestore.local.QueryEngine;
 import com.google.firebase.firestore.local.Scheduler;
 import com.google.firebase.firestore.model.DocumentKey;
 import com.google.firebase.firestore.model.mutation.MutationBatchResult;
@@ -54,10 +54,7 @@ protected EventManager createEventManager(Configuration configuration) {
   @Override
   protected LocalStore createLocalStore(Configuration configuration) {
     return new LocalStore(
-        getPersistence(),
-        getIndexBackfiller(),
-        new DefaultQueryEngine(),
-        configuration.getInitialUser());
+        getPersistence(), getIndexBackfiller(), new QueryEngine(), configuration.getInitialUser());
   }
 
   @Override

firebase-firestore/src/main/java/com/google/firebase/firestore/local/DefaultQueryEngine.java

@@ -169,8 +169,7 @@ public LocalStore(
         new LocalDocumentsView(remoteDocuments, mutationQueue, documentOverlayCache, indexManager);
     this.queryEngine = queryEngine;
     this.indexBackfiller = indexBackfiller;
-    queryEngine.setLocalDocumentsView(localDocuments);
-    queryEngine.setIndexManager(indexManager);
+    queryEngine.initialize(localDocuments, indexManager);
 
     localViewReferences = new ReferenceSet();
     persistence.getReferenceDelegate().setInMemoryPins(localViewReferences);
@@ -215,8 +214,7 @@ public ImmutableSortedMap<DocumentKey, Document> handleUserChange(User user) {
     // Recreate our LocalDocumentsView using the new MutationQueue.
     localDocuments =
         new LocalDocumentsView(remoteDocuments, mutationQueue, documentOverlayCache, indexManager);
-    queryEngine.setLocalDocumentsView(localDocuments);
-    queryEngine.setIndexManager(indexManager);
+    queryEngine.initialize(localDocuments, indexManager);
 
     // TODO(indexing): Add spec tests that test these components change after a user change
     remoteDocuments.setIndexManager(indexManager);
@@ -854,8 +852,8 @@ public QueryResult executeQuery(Query query, boolean usePreviousResults) {
     ImmutableSortedMap<DocumentKey, Document> documents =
         queryEngine.getDocumentsMatchingQuery(
             query,
-            usePreviousResults ? lastLimboFreeSnapshotVersion : SnapshotVersion.NONE,
-            usePreviousResults ? remoteKeys : DocumentKey.emptyKeySet());
+            remoteKeys,
+            usePreviousResults ? lastLimboFreeSnapshotVersion : SnapshotVersion.NONE);
     return new QueryResult(documents, remoteKeys);
   }