Tree-shake LocalStore

Author: schmidt-sebastian

packages/firestore/src/core/component_provider.ts

@@ -23,9 +23,8 @@ import {
 } from '../local/shared_client_state';
 import {
   LocalStore,
-  MultiTabLocalStore,
   newLocalStore,
-  newMultiTabLocalStore
+  synchronizeLastDocumentChangeReadTime
 } from '../local/local_store';
 import {
   MultiTabSyncEngine,
@@ -125,7 +124,6 @@ export class MemoryComponentProvider implements ComponentProvider {
       );
     this.remoteStore.syncEngine = this.syncEngine;
 
-    await this.localStore.start();
     await this.sharedClientState.start();
     await this.remoteStore.start();
 
@@ -281,7 +279,6 @@ export class IndexedDbComponentProvider extends MemoryComponentProvider {
  * `synchronizeTabs` will be enabled.
  */
 export class MultiTabIndexedDbComponentProvider extends IndexedDbComponentProvider {
-  localStore!: MultiTabLocalStore;
   syncEngine!: MultiTabSyncEngine;
 
   async initialize(cfg: ComponentConfiguration): Promise<void> {
@@ -301,14 +298,8 @@ export class MultiTabIndexedDbComponentProvider extends IndexedDbComponentProvid
         }
       }
     });
-  }
 
-  createLocalStore(cfg: ComponentConfiguration): LocalStore {
-    return newMultiTabLocalStore(
-      this.persistence,
-      new IndexFreeQueryEngine(),
-      cfg.initialUser
-    );
+    await synchronizeLastDocumentChangeReadTime(this.localStore);
   }
 
   createSyncEngine(cfg: ComponentConfiguration): SyncEngine {

packages/firestore/src/core/sync_engine.ts

@@ -17,9 +17,14 @@
 
 import { User } from '../auth/user';
 import {
+  getNewDocumentChanges,
+  getTarget,
   ignoreIfPrimaryLeaseLoss,
   LocalStore,
-  MultiTabLocalStore
+  localStoreGetActiveClients,
+  lookupMutationDocuments,
+  removeCachedMutationBatchMetadata,
+  setNetworkEnabled
 } from '../local/local_store';
 import { LocalViewChanges } from '../local/local_view_changes';
 import { ReferenceSet } from '../local/reference_set';
@@ -1016,35 +1021,17 @@ class MultiTabSyncEngineImpl extends SyncEngineImpl {
   // `isPrimary` is true.
   private _isPrimaryClient: undefined | boolean = undefined;
 
-  constructor(
-    protected localStore: MultiTabLocalStore,
-    remoteStore: RemoteStore,
-    datastore: Datastore,
-    sharedClientState: SharedClientState,
-    currentUser: User,
-    maxConcurrentLimboResolutions: number
-  ) {
-    super(
-      localStore,
-      remoteStore,
-      datastore,
-      sharedClientState,
-      currentUser,
-      maxConcurrentLimboResolutions
-    );
-  }
-
   get isPrimaryClient(): boolean {
     return this._isPrimaryClient === true;
   }
 
   enableNetwork(): Promise<void> {
-    this.localStore.setNetworkEnabled(true);
+    setNetworkEnabled(this.localStore, true);
     return super.enableNetwork();
   }
 
   disableNetwork(): Promise<void> {
-    this.localStore.setNetworkEnabled(false);
+    setNetworkEnabled(this.localStore, false);
     return super.disableNetwork();
   }
 
@@ -1097,7 +1084,7 @@ class MultiTabSyncEngineImpl extends SyncEngineImpl {
     error?: FirestoreError
   ): Promise<void> {
     this.assertSubscribed('applyBatchState()');
-    const documents = await this.localStore.lookupMutationDocuments(batchId);
+    const documents = await lookupMutationDocuments(this.localStore, batchId);
 
     if (documents === null) {
       // A throttled tab may not have seen the mutation before it was completed
@@ -1120,7 +1107,7 @@ class MultiTabSyncEngineImpl extends SyncEngineImpl {
       // NOTE: Both these methods are no-ops for batches that originated from
       // other clients.
       this.processUserCallback(batchId, error ? error : null);
-      this.localStore.removeCachedMutationBatchMetadata(batchId);
+      removeCachedMutationBatchMetadata(this.localStore, batchId);
     } else {
       fail(`Unknown batchState: ${batchState}`);
     }
@@ -1236,7 +1223,7 @@ class MultiTabSyncEngineImpl extends SyncEngineImpl {
         );
         // For queries that never executed on this client, we need to
         // allocate the target in LocalStore and initialize a new View.
-        const target = await this.localStore.getTarget(targetId);
+        const target = await getTarget(this.localStore, targetId);
         debugAssert(!!target, `Target for id ${targetId} not found`);
         targetData = await this.localStore.allocateTarget(target);
         await this.initializeViewAndComputeSnapshot(
@@ -1277,7 +1264,7 @@ class MultiTabSyncEngineImpl extends SyncEngineImpl {
   }
 
   getActiveClients(): Promise<ClientId[]> {
-    return this.localStore.getActiveClients();
+    return localStoreGetActiveClients(this.localStore);
   }
 
   async applyTargetState(
@@ -1296,7 +1283,7 @@ class MultiTabSyncEngineImpl extends SyncEngineImpl {
       switch (state) {
         case 'current':
         case 'not-current': {
-          const changes = await this.localStore.getNewDocumentChanges();
+          const changes = await getNewDocumentChanges(this.localStore);
           const synthesizedRemoteEvent = RemoteEvent.createSynthesizedRemoteEventForCurrentChange(
             targetId,
             state === 'current'
@@ -1336,7 +1323,7 @@ class MultiTabSyncEngineImpl extends SyncEngineImpl {
         continue;
       }
 
-      const target = await this.localStore.getTarget(targetId);
+      const target = await getTarget(this.localStore, targetId);
       debugAssert(
         !!target,
         `Query data for active target ${targetId} not found`
@@ -1370,7 +1357,7 @@ class MultiTabSyncEngineImpl extends SyncEngineImpl {
 }
 
 export function newMultiTabSyncEngine(
-  localStore: MultiTabLocalStore,
+  localStore: LocalStore,
   remoteStore: RemoteStore,
   datastore: Datastore,
   sharedClientState: SharedClientState,