Skip to content

Commit 3060e86

Browse files
feat: add new types QueryMode, QueryPlan, ResultSetStats (#1516)
* feat: add new types QueryMode, QueryPlan, ResultSetStats feat: add QueryMode field to RunQueryRequest feat: add ResultSetStats field to RunQueryResponse feat: add QueryMode field to RunAggregationQueryRequest feat: add ResultSetStats field to RunAggregationQueryResponse PiperOrigin-RevId: 595771083 Source-Link: googleapis/googleapis@2027807 Source-Link: https://github.com/googleapis/googleapis-gen/commit/5407e2b6863928c26f52db9f347c6b5556e702f2 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNTQwN2UyYjY4NjM5MjhjMjZmNTJkYjlmMzQ3YzZiNTU1NmU3MDJmMiJ9 * 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md --------- Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
1 parent 866b2b4 commit 3060e86

20 files changed

+3847
-241
lines changed

google-cloud-firestore/src/main/java/com/google/cloud/firestore/v1/FirestoreClient.java

+9-2
Original file line numberDiff line numberDiff line change
@@ -875,7 +875,11 @@ public final UnaryCallable<RollbackRequest, Empty> rollbackCallable() {
875875
* // - It may require specifying regional endpoints when creating the service client as shown in
876876
* // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
877877
* try (FirestoreClient firestoreClient = FirestoreClient.create()) {
878-
* RunQueryRequest request = RunQueryRequest.newBuilder().setParent("parent-995424086").build();
878+
* RunQueryRequest request =
879+
* RunQueryRequest.newBuilder()
880+
* .setParent("parent-995424086")
881+
* .setMode(QueryMode.forNumber(0))
882+
* .build();
879883
* ServerStream<RunQueryResponse> stream = firestoreClient.runQueryCallable().call(request);
880884
* for (RunQueryResponse response : stream) {
881885
* // Do something when a response is received.
@@ -911,7 +915,10 @@ public final ServerStreamingCallable<RunQueryRequest, RunQueryResponse> runQuery
911915
* // https://cloud.google.com/java/docs/setup#configure_endpoints_for_the_client_library
912916
* try (FirestoreClient firestoreClient = FirestoreClient.create()) {
913917
* RunAggregationQueryRequest request =
914-
* RunAggregationQueryRequest.newBuilder().setParent("parent-995424086").build();
918+
* RunAggregationQueryRequest.newBuilder()
919+
* .setParent("parent-995424086")
920+
* .setMode(QueryMode.forNumber(0))
921+
* .build();
915922
* ServerStream<RunAggregationQueryResponse> stream =
916923
* firestoreClient.runAggregationQueryCallable().call(request);
917924
* for (RunAggregationQueryResponse response : stream) {

google-cloud-firestore/src/main/resources/META-INF/native-image/com.google.cloud.firestore.v1/reflect-config.json

+45
Original file line numberDiff line numberDiff line change
@@ -1070,6 +1070,51 @@
10701070
"allDeclaredClasses": true,
10711071
"allPublicClasses": true
10721072
},
1073+
{
1074+
"name": "com.google.firestore.v1.QueryMode",
1075+
"queryAllDeclaredConstructors": true,
1076+
"queryAllPublicConstructors": true,
1077+
"queryAllDeclaredMethods": true,
1078+
"allPublicMethods": true,
1079+
"allDeclaredClasses": true,
1080+
"allPublicClasses": true
1081+
},
1082+
{
1083+
"name": "com.google.firestore.v1.QueryPlan",
1084+
"queryAllDeclaredConstructors": true,
1085+
"queryAllPublicConstructors": true,
1086+
"queryAllDeclaredMethods": true,
1087+
"allPublicMethods": true,
1088+
"allDeclaredClasses": true,
1089+
"allPublicClasses": true
1090+
},
1091+
{
1092+
"name": "com.google.firestore.v1.QueryPlan$Builder",
1093+
"queryAllDeclaredConstructors": true,
1094+
"queryAllPublicConstructors": true,
1095+
"queryAllDeclaredMethods": true,
1096+
"allPublicMethods": true,
1097+
"allDeclaredClasses": true,
1098+
"allPublicClasses": true
1099+
},
1100+
{
1101+
"name": "com.google.firestore.v1.ResultSetStats",
1102+
"queryAllDeclaredConstructors": true,
1103+
"queryAllPublicConstructors": true,
1104+
"queryAllDeclaredMethods": true,
1105+
"allPublicMethods": true,
1106+
"allDeclaredClasses": true,
1107+
"allPublicClasses": true
1108+
},
1109+
{
1110+
"name": "com.google.firestore.v1.ResultSetStats$Builder",
1111+
"queryAllDeclaredConstructors": true,
1112+
"queryAllPublicConstructors": true,
1113+
"queryAllDeclaredMethods": true,
1114+
"allPublicMethods": true,
1115+
"allDeclaredClasses": true,
1116+
"allPublicClasses": true
1117+
},
10731118
{
10741119
"name": "com.google.firestore.v1.RollbackRequest",
10751120
"queryAllDeclaredConstructors": true,

google-cloud-firestore/src/test/java/com/google/cloud/firestore/v1/FirestoreClientTest.java

+22-4
Original file line numberDiff line numberDiff line change
@@ -56,6 +56,8 @@
5656
import com.google.firestore.v1.ListenResponse;
5757
import com.google.firestore.v1.PartitionQueryRequest;
5858
import com.google.firestore.v1.PartitionQueryResponse;
59+
import com.google.firestore.v1.QueryMode;
60+
import com.google.firestore.v1.ResultSetStats;
5961
import com.google.firestore.v1.RollbackRequest;
6062
import com.google.firestore.v1.RunAggregationQueryRequest;
6163
import com.google.firestore.v1.RunAggregationQueryResponse;
@@ -505,9 +507,14 @@ public void runQueryTest() throws Exception {
505507
.setDocument(Document.newBuilder().build())
506508
.setReadTime(Timestamp.newBuilder().build())
507509
.setSkippedResults(880286183)
510+
.setStats(ResultSetStats.newBuilder().build())
508511
.build();
509512
mockFirestore.addResponse(expectedResponse);
510-
RunQueryRequest request = RunQueryRequest.newBuilder().setParent("parent-995424086").build();
513+
RunQueryRequest request =
514+
RunQueryRequest.newBuilder()
515+
.setParent("parent-995424086")
516+
.setMode(QueryMode.forNumber(0))
517+
.build();
511518

512519
MockStreamObserver<RunQueryResponse> responseObserver = new MockStreamObserver<>();
513520

@@ -523,7 +530,11 @@ public void runQueryTest() throws Exception {
523530
public void runQueryExceptionTest() throws Exception {
524531
StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
525532
mockFirestore.addException(exception);
526-
RunQueryRequest request = RunQueryRequest.newBuilder().setParent("parent-995424086").build();
533+
RunQueryRequest request =
534+
RunQueryRequest.newBuilder()
535+
.setParent("parent-995424086")
536+
.setMode(QueryMode.forNumber(0))
537+
.build();
527538

528539
MockStreamObserver<RunQueryResponse> responseObserver = new MockStreamObserver<>();
529540

@@ -547,10 +558,14 @@ public void runAggregationQueryTest() throws Exception {
547558
.setResult(AggregationResult.newBuilder().build())
548559
.setTransaction(ByteString.EMPTY)
549560
.setReadTime(Timestamp.newBuilder().build())
561+
.setStats(ResultSetStats.newBuilder().build())
550562
.build();
551563
mockFirestore.addResponse(expectedResponse);
552564
RunAggregationQueryRequest request =
553-
RunAggregationQueryRequest.newBuilder().setParent("parent-995424086").build();
565+
RunAggregationQueryRequest.newBuilder()
566+
.setParent("parent-995424086")
567+
.setMode(QueryMode.forNumber(0))
568+
.build();
554569

555570
MockStreamObserver<RunAggregationQueryResponse> responseObserver = new MockStreamObserver<>();
556571

@@ -568,7 +583,10 @@ public void runAggregationQueryExceptionTest() throws Exception {
568583
StatusRuntimeException exception = new StatusRuntimeException(io.grpc.Status.INVALID_ARGUMENT);
569584
mockFirestore.addException(exception);
570585
RunAggregationQueryRequest request =
571-
RunAggregationQueryRequest.newBuilder().setParent("parent-995424086").build();
586+
RunAggregationQueryRequest.newBuilder()
587+
.setParent("parent-995424086")
588+
.setMode(QueryMode.forNumber(0))
589+
.build();
572590

573591
MockStreamObserver<RunAggregationQueryResponse> responseObserver = new MockStreamObserver<>();
574592

0 commit comments

Comments
 (0)