@firebase/firestore 3.4.10 → 3.4.11-canary.b60cf76e1

package/dist/index.node.mjs

@@ -8,7 +8,7 @@ import * as grpc from '@grpc/grpc-js';
 import * as protoLoader from '@grpc/proto-loader';
 
 const name = "@firebase/firestore";
-const version$1 = "3.4.10";
+const version$1 = "3.4.11-canary.b60cf76e1";
 
 /**
  * @license
@@ -61,7 +61,7 @@ User.GOOGLE_CREDENTIALS = new User('google-credentials-uid');
 User.FIRST_PARTY = new User('first-party-uid');
 User.MOCK_USER = new User('mock-user');
 
-const version = "9.8.3";
+const version = "9.8.4-canary.b60cf76e1";
 
 /**
  * @license
@@ -2225,7 +2225,8 @@ class SimpleDb {
                 await transaction.completionPromise;
                 return transactionFnResult;
             }
-            catch (error) {
+            catch (e) {
+                const error = e;
                 // TODO(schmidt-sebastian): We could probably be smarter about this and
                 // not retry exceptions that are likely unrecoverable (such as quota
                 // exceeded errors).
@@ -17236,7 +17237,7 @@ class GrpcConnection {
             });
         });
     }
-    invokeStreamingRPC(rpcName, path, request, authToken, appCheckToken) {
+    invokeStreamingRPC(rpcName, path, request, authToken, appCheckToken, expectedResponseCount) {
         const results = [];
         const responseDeferred = new Deferred();
         logDebug(LOG_TAG$9, `RPC '${rpcName}' invoked (streaming) with request:`, request);
@@ -17244,13 +17245,22 @@ class GrpcConnection {
         const metadata = createMetadata(this.databasePath, authToken, appCheckToken, this.databaseInfo.appId);
         const jsonRequest = Object.assign(Object.assign({}, request), { database: this.databasePath });
         const stream = stub[rpcName](jsonRequest, metadata);
+        let callbackFired = false;
         stream.on('data', (response) => {
             logDebug(LOG_TAG$9, `RPC ${rpcName} received result:`, response);
             results.push(response);
+            if (expectedResponseCount !== undefined &&
+                results.length === expectedResponseCount) {
+                callbackFired = true;
+                responseDeferred.resolve(results);
+            }
         });
         stream.on('end', () => {
             logDebug(LOG_TAG$9, `RPC '${rpcName}' completed.`);
-            responseDeferred.resolve(results);
+            if (!callbackFired) {
+                callbackFired = true;
+                responseDeferred.resolve(results);
+            }
         });
         stream.on('error', (grpcError) => {
             logDebug(LOG_TAG$9, `RPC '${rpcName}' failed with error:`, grpcError);
@@ -20803,14 +20813,14 @@ class DatastoreImpl extends Datastore {
         });
     }
     /** Invokes the provided RPC with streamed results with auth and AppCheck tokens. */
-    invokeStreamingRPC(rpcName, path, request) {
+    invokeStreamingRPC(rpcName, path, request, expectedResponseCount) {
         this.verifyInitialized();
         return Promise.all([
             this.authCredentials.getToken(),
             this.appCheckCredentials.getToken()
         ])
             .then(([authToken, appCheckToken]) => {
-            return this.connection.invokeStreamingRPC(rpcName, path, request, authToken, appCheckToken);
+            return this.connection.invokeStreamingRPC(rpcName, path, request, authToken, appCheckToken, expectedResponseCount);
         })
             .catch((error) => {
             if (error.name === 'FirebaseError') {
@@ -20848,7 +20858,7 @@ async function invokeBatchGetDocumentsRpc(datastore, keys) {
     const request = {
         documents: keys.map(k => toName(datastoreImpl.serializer, k))
     };
-    const response = await datastoreImpl.invokeStreamingRPC('BatchGetDocuments', path, request);
+    const response = await datastoreImpl.invokeStreamingRPC('BatchGetDocuments', path, request, keys.length);
     const docs = new Map();
     response.forEach(proto => {
         const doc = fromBatchGetDocumentsResponse(datastoreImpl.serializer, proto);
@@ -26190,13 +26200,14 @@ function setPersistenceProviders(client, onlineComponentProvider, offlineCompone
             persistenceResult.resolve();
         }
         catch (e) {
-            if (!canFallbackFromIndexedDbError(e)) {
-                throw e;
+            const error = e;
+            if (!canFallbackFromIndexedDbError(error)) {
+                throw error;
             }
             console.warn('Error enabling offline persistence. Falling back to ' +
                 'persistence disabled: ' +
-                e);
-            persistenceResult.reject(e);
+                error);
+            persistenceResult.reject(error);
         }
     })
         .then(() => persistenceResult.promise);
@@ -29141,11 +29152,12 @@ function setIndexConfiguration(firestore, jsonOrConfiguration) {
     return Promise.resolve();
 }
 function tryParseJson(json) {
+    var _a;
     try {
         return JSON.parse(json);
     }
     catch (e) {
-        throw new FirestoreError(Code.INVALID_ARGUMENT, 'Failed to parse JSON:' + e.message);
+        throw new FirestoreError(Code.INVALID_ARGUMENT, 'Failed to parse JSON:' + ((_a = e) === null || _a === void 0 ? void 0 : _a.message));
     }
 }
 function tryGetString(data, property) {