Add invokeRunQueryRpc() (#3025)

Author: schmidt-sebastian

packages/firestore/src/protos/firestore_proto_api.d.ts

@@ -314,6 +314,7 @@ export declare namespace firestoreV1ApiClientInterfaces {
     transaction?: string;
   }
   interface RunQueryRequest {
+    parent?: string;
     structuredQuery?: StructuredQuery;
     transaction?: string;
     newTransaction?: TransactionOptions;

packages/firestore/src/remote/datastore.ts

@@ -16,7 +16,7 @@
  */
 
 import { CredentialsProvider } from '../api/credentials';
-import { MaybeDocument } from '../model/document';
+import { MaybeDocument, Document } from '../model/document';
 import { DocumentKey } from '../model/document_key';
 import { Mutation, MutationResult } from '../model/mutation';
 import * as api from '../protos/firestore_proto_api';
@@ -31,6 +31,7 @@ import {
   WriteStreamListener
 } from './persistent_stream';
 import { AsyncQueue } from '../util/async_queue';
+import { Query } from '../core/query';
 
 /**
  * Datastore and its related methods are a wrapper around the external Google
@@ -149,6 +150,33 @@ export async function invokeBatchGetDocumentsRpc(
   return result;
 }
 
+export async function invokeRunQueryRpc(
+  datastore: Datastore,
+  query: Query
+): Promise<Document[]> {
+  const datastoreImpl = debugCast(datastore, DatastoreImpl);
+  const { structuredQuery, parent } = datastoreImpl.serializer.toQueryTarget(
+    query.toTarget()
+  );
+  const params = {
+    database: datastoreImpl.serializer.encodedDatabaseId,
+    parent,
+    structuredQuery
+  };
+
+  const response = await datastoreImpl.invokeStreamingRPC<
+    api.RunQueryRequest,
+    api.RunQueryResponse
+  >('RunQuery', params);
+
+  return (
+    response
+      // Omit RunQueryResponses that only contain readTimes.
+      .filter(proto => !!proto.document)
+      .map(proto => datastoreImpl.serializer.fromDocument(proto.document!))
+  );
+}
+
 export function newPersistentWriteStream(
   datastore: Datastore,
   queue: AsyncQueue,