@firebase/firestore 3.4.10-canary.8e952e2ee → 3.4.10-canary.bf4959b60

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-canary.8e952e2ee";
+const version$1 = "3.4.10-canary.bf4959b60";
 
 /**
  * @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-canary.8e952e2ee";
+const version = "9.8.3-canary.bf4959b60";
 
 /**
  * @license
@@ -17237,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);
@@ -17245,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);
@@ -20804,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') {
@@ -20849,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);