use documentKeySet() and documentMap().

Author: ehsannas

packages/firestore/src/local/local_documents_view.ts

@@ -32,7 +32,7 @@ import {
   newMutationMap,
   newOverlayMap,
   documentMap,
-  mutableDocumentMap
+  mutableDocumentMap, documentKeySet
 } from '../model/collections';
 import { Document, MutableDocument } from '../model/document';
 import { DocumentKey } from '../model/document_key';
@@ -116,7 +116,7 @@ export class LocalDocumentsView {
         this.getLocalViewOfDocuments(
           transaction,
           docs,
-          new SortedSet(DocumentKey.comparator)
+          documentKeySet()
         ).next(() => docs as DocumentMap)
       );
   }
@@ -209,7 +209,7 @@ export class LocalDocumentsView {
     let documentsByBatchId = new SortedMap<number, DocumentKeySet>(
       (key1: number, key2: number) => key1 - key2
     );
-    let processed = new SortedSet(DocumentKey.comparator);
+    let processed = documentKeySet();
     return this.mutationQueue
       .getAllMutationBatchesAffectingDocumentKeys(transaction, docs)
       .next(batches => {
@@ -223,7 +223,7 @@ export class LocalDocumentsView {
             if (documentsByBatchId.get(batch.batchId) === null) {
               documentsByBatchId = documentsByBatchId.insert(
                 batch.batchId,
-                new SortedSet(DocumentKey.comparator)
+                documentKeySet()
               );
             }
             const newSet = documentsByBatchId.get(batch.batchId)!.add(key);
@@ -394,9 +394,7 @@ export class LocalDocumentsView {
         });
 
         // Apply the overlays and match against the query.
-        let results = new SortedMap<DocumentKey, Document>(
-          DocumentKey.comparator
-        );
+        let results = documentMap();
         remoteDocuments.forEach((key, document) => {
           const overlay = overlays.get(key);
           if (overlay !== undefined) {