Skip to content

Commit 7c23d2b

Browse files
yoshi-automationPraful Makani
yoshi-automation
authored and
Praful Makani
committed
Regenerate firestore client (#4767)
1 parent a954d42 commit 7c23d2b

File tree

4 files changed

+293
-5
lines changed

4 files changed

+293
-5
lines changed

‎google-cloud-clients/google-cloud-firestore/src/main/java/com/google/cloud/firestore/v1/stub/GrpcFirestoreAdminStub.java‎

Lines changed: 84 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,9 @@
2424
import com.google.api.gax.grpc.GrpcCallSettings;
2525
import com.google.api.gax.grpc.GrpcStubCallableFactory;
2626
import com.google.api.gax.rpc.ClientContext;
27+
import com.google.api.gax.rpc.RequestParamsExtractor;
2728
import com.google.api.gax.rpc.UnaryCallable;
29+
import com.google.common.collect.ImmutableMap;
2830
import com.google.firestore.admin.v1.CreateIndexRequest;
2931
import com.google.firestore.admin.v1.DeleteIndexRequest;
3032
import com.google.firestore.admin.v1.ExportDocumentsRequest;
@@ -43,6 +45,7 @@
4345
import io.grpc.MethodDescriptor;
4446
import io.grpc.protobuf.ProtoUtils;
4547
import java.io.IOException;
48+
import java.util.Map;
4649
import java.util.concurrent.TimeUnit;
4750
import javax.annotation.Generated;
4851

@@ -186,38 +189,119 @@ protected GrpcFirestoreAdminStub(
186189
GrpcCallSettings<CreateIndexRequest, Operation> createIndexTransportSettings =
187190
GrpcCallSettings.<CreateIndexRequest, Operation>newBuilder()
188191
.setMethodDescriptor(createIndexMethodDescriptor)
192+
.setParamsExtractor(
193+
new RequestParamsExtractor<CreateIndexRequest>() {
194+
@Override
195+
public Map<String, String> extract(CreateIndexRequest request) {
196+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
197+
params.put("parent", String.valueOf(request.getParent()));
198+
return params.build();
199+
}
200+
})
189201
.build();
190202
GrpcCallSettings<ListIndexesRequest, ListIndexesResponse> listIndexesTransportSettings =
191203
GrpcCallSettings.<ListIndexesRequest, ListIndexesResponse>newBuilder()
192204
.setMethodDescriptor(listIndexesMethodDescriptor)
205+
.setParamsExtractor(
206+
new RequestParamsExtractor<ListIndexesRequest>() {
207+
@Override
208+
public Map<String, String> extract(ListIndexesRequest request) {
209+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
210+
params.put("parent", String.valueOf(request.getParent()));
211+
return params.build();
212+
}
213+
})
193214
.build();
194215
GrpcCallSettings<GetIndexRequest, Index> getIndexTransportSettings =
195216
GrpcCallSettings.<GetIndexRequest, Index>newBuilder()
196217
.setMethodDescriptor(getIndexMethodDescriptor)
218+
.setParamsExtractor(
219+
new RequestParamsExtractor<GetIndexRequest>() {
220+
@Override
221+
public Map<String, String> extract(GetIndexRequest request) {
222+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
223+
params.put("name", String.valueOf(request.getName()));
224+
return params.build();
225+
}
226+
})
197227
.build();
198228
GrpcCallSettings<DeleteIndexRequest, Empty> deleteIndexTransportSettings =
199229
GrpcCallSettings.<DeleteIndexRequest, Empty>newBuilder()
200230
.setMethodDescriptor(deleteIndexMethodDescriptor)
231+
.setParamsExtractor(
232+
new RequestParamsExtractor<DeleteIndexRequest>() {
233+
@Override
234+
public Map<String, String> extract(DeleteIndexRequest request) {
235+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
236+
params.put("name", String.valueOf(request.getName()));
237+
return params.build();
238+
}
239+
})
201240
.build();
202241
GrpcCallSettings<ImportDocumentsRequest, Operation> importDocumentsTransportSettings =
203242
GrpcCallSettings.<ImportDocumentsRequest, Operation>newBuilder()
204243
.setMethodDescriptor(importDocumentsMethodDescriptor)
244+
.setParamsExtractor(
245+
new RequestParamsExtractor<ImportDocumentsRequest>() {
246+
@Override
247+
public Map<String, String> extract(ImportDocumentsRequest request) {
248+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
249+
params.put("name", String.valueOf(request.getName()));
250+
return params.build();
251+
}
252+
})
205253
.build();
206254
GrpcCallSettings<ExportDocumentsRequest, Operation> exportDocumentsTransportSettings =
207255
GrpcCallSettings.<ExportDocumentsRequest, Operation>newBuilder()
208256
.setMethodDescriptor(exportDocumentsMethodDescriptor)
257+
.setParamsExtractor(
258+
new RequestParamsExtractor<ExportDocumentsRequest>() {
259+
@Override
260+
public Map<String, String> extract(ExportDocumentsRequest request) {
261+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
262+
params.put("name", String.valueOf(request.getName()));
263+
return params.build();
264+
}
265+
})
209266
.build();
210267
GrpcCallSettings<GetFieldRequest, Field> getFieldTransportSettings =
211268
GrpcCallSettings.<GetFieldRequest, Field>newBuilder()
212269
.setMethodDescriptor(getFieldMethodDescriptor)
270+
.setParamsExtractor(
271+
new RequestParamsExtractor<GetFieldRequest>() {
272+
@Override
273+
public Map<String, String> extract(GetFieldRequest request) {
274+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
275+
params.put("name", String.valueOf(request.getName()));
276+
return params.build();
277+
}
278+
})
213279
.build();
214280
GrpcCallSettings<ListFieldsRequest, ListFieldsResponse> listFieldsTransportSettings =
215281
GrpcCallSettings.<ListFieldsRequest, ListFieldsResponse>newBuilder()
216282
.setMethodDescriptor(listFieldsMethodDescriptor)
283+
.setParamsExtractor(
284+
new RequestParamsExtractor<ListFieldsRequest>() {
285+
@Override
286+
public Map<String, String> extract(ListFieldsRequest request) {
287+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
288+
params.put("parent", String.valueOf(request.getParent()));
289+
return params.build();
290+
}
291+
})
217292
.build();
218293
GrpcCallSettings<UpdateFieldRequest, Operation> updateFieldTransportSettings =
219294
GrpcCallSettings.<UpdateFieldRequest, Operation>newBuilder()
220295
.setMethodDescriptor(updateFieldMethodDescriptor)
296+
.setParamsExtractor(
297+
new RequestParamsExtractor<UpdateFieldRequest>() {
298+
@Override
299+
public Map<String, String> extract(UpdateFieldRequest request) {
300+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
301+
params.put("field.name", String.valueOf(request.getField().getName()));
302+
return params.build();
303+
}
304+
})
221305
.build();
222306

223307
this.createIndexCallable =

‎google-cloud-clients/google-cloud-firestore/src/main/java/com/google/cloud/firestore/v1/stub/GrpcFirestoreStub.java‎

Lines changed: 102 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,10 @@
2525
import com.google.api.gax.grpc.GrpcStubCallableFactory;
2626
import com.google.api.gax.rpc.BidiStreamingCallable;
2727
import com.google.api.gax.rpc.ClientContext;
28+
import com.google.api.gax.rpc.RequestParamsExtractor;
2829
import com.google.api.gax.rpc.ServerStreamingCallable;
2930
import com.google.api.gax.rpc.UnaryCallable;
31+
import com.google.common.collect.ImmutableMap;
3032
import com.google.firestore.v1.BatchGetDocumentsRequest;
3133
import com.google.firestore.v1.BatchGetDocumentsResponse;
3234
import com.google.firestore.v1.BeginTransactionRequest;
@@ -53,6 +55,7 @@
5355
import io.grpc.MethodDescriptor;
5456
import io.grpc.protobuf.ProtoUtils;
5557
import java.io.IOException;
58+
import java.util.Map;
5659
import java.util.concurrent.TimeUnit;
5760
import javax.annotation.Generated;
5861

@@ -241,6 +244,15 @@ protected GrpcFirestoreStub(
241244
GrpcCallSettings<GetDocumentRequest, Document> getDocumentTransportSettings =
242245
GrpcCallSettings.<GetDocumentRequest, Document>newBuilder()
243246
.setMethodDescriptor(getDocumentMethodDescriptor)
247+
.setParamsExtractor(
248+
new RequestParamsExtractor<GetDocumentRequest>() {
249+
@Override
250+
public Map<String, String> extract(GetDocumentRequest request) {
251+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
252+
params.put("name", String.valueOf(request.getName()));
253+
return params.build();
254+
}
255+
})
244256
.build();
245257
GrpcCallSettings<ListDocumentsRequest, ListDocumentsResponse> listDocumentsTransportSettings =
246258
GrpcCallSettings.<ListDocumentsRequest, ListDocumentsResponse>newBuilder()
@@ -253,45 +265,135 @@ protected GrpcFirestoreStub(
253265
GrpcCallSettings<UpdateDocumentRequest, Document> updateDocumentTransportSettings =
254266
GrpcCallSettings.<UpdateDocumentRequest, Document>newBuilder()
255267
.setMethodDescriptor(updateDocumentMethodDescriptor)
268+
.setParamsExtractor(
269+
new RequestParamsExtractor<UpdateDocumentRequest>() {
270+
@Override
271+
public Map<String, String> extract(UpdateDocumentRequest request) {
272+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
273+
params.put("document.name", String.valueOf(request.getDocument().getName()));
274+
return params.build();
275+
}
276+
})
256277
.build();
257278
GrpcCallSettings<DeleteDocumentRequest, Empty> deleteDocumentTransportSettings =
258279
GrpcCallSettings.<DeleteDocumentRequest, Empty>newBuilder()
259280
.setMethodDescriptor(deleteDocumentMethodDescriptor)
281+
.setParamsExtractor(
282+
new RequestParamsExtractor<DeleteDocumentRequest>() {
283+
@Override
284+
public Map<String, String> extract(DeleteDocumentRequest request) {
285+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
286+
params.put("name", String.valueOf(request.getName()));
287+
return params.build();
288+
}
289+
})
260290
.build();
261291
GrpcCallSettings<BatchGetDocumentsRequest, BatchGetDocumentsResponse>
262292
batchGetDocumentsTransportSettings =
263293
GrpcCallSettings.<BatchGetDocumentsRequest, BatchGetDocumentsResponse>newBuilder()
264294
.setMethodDescriptor(batchGetDocumentsMethodDescriptor)
295+
.setParamsExtractor(
296+
new RequestParamsExtractor<BatchGetDocumentsRequest>() {
297+
@Override
298+
public Map<String, String> extract(BatchGetDocumentsRequest request) {
299+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
300+
params.put("database", String.valueOf(request.getDatabase()));
301+
return params.build();
302+
}
303+
})
265304
.build();
266305
GrpcCallSettings<BeginTransactionRequest, BeginTransactionResponse>
267306
beginTransactionTransportSettings =
268307
GrpcCallSettings.<BeginTransactionRequest, BeginTransactionResponse>newBuilder()
269308
.setMethodDescriptor(beginTransactionMethodDescriptor)
309+
.setParamsExtractor(
310+
new RequestParamsExtractor<BeginTransactionRequest>() {
311+
@Override
312+
public Map<String, String> extract(BeginTransactionRequest request) {
313+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
314+
params.put("database", String.valueOf(request.getDatabase()));
315+
return params.build();
316+
}
317+
})
270318
.build();
271319
GrpcCallSettings<CommitRequest, CommitResponse> commitTransportSettings =
272320
GrpcCallSettings.<CommitRequest, CommitResponse>newBuilder()
273321
.setMethodDescriptor(commitMethodDescriptor)
322+
.setParamsExtractor(
323+
new RequestParamsExtractor<CommitRequest>() {
324+
@Override
325+
public Map<String, String> extract(CommitRequest request) {
326+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
327+
params.put("database", String.valueOf(request.getDatabase()));
328+
return params.build();
329+
}
330+
})
274331
.build();
275332
GrpcCallSettings<RollbackRequest, Empty> rollbackTransportSettings =
276333
GrpcCallSettings.<RollbackRequest, Empty>newBuilder()
277334
.setMethodDescriptor(rollbackMethodDescriptor)
335+
.setParamsExtractor(
336+
new RequestParamsExtractor<RollbackRequest>() {
337+
@Override
338+
public Map<String, String> extract(RollbackRequest request) {
339+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
340+
params.put("database", String.valueOf(request.getDatabase()));
341+
return params.build();
342+
}
343+
})
278344
.build();
279345
GrpcCallSettings<RunQueryRequest, RunQueryResponse> runQueryTransportSettings =
280346
GrpcCallSettings.<RunQueryRequest, RunQueryResponse>newBuilder()
281347
.setMethodDescriptor(runQueryMethodDescriptor)
348+
.setParamsExtractor(
349+
new RequestParamsExtractor<RunQueryRequest>() {
350+
@Override
351+
public Map<String, String> extract(RunQueryRequest request) {
352+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
353+
params.put("parent", String.valueOf(request.getParent()));
354+
return params.build();
355+
}
356+
})
282357
.build();
283358
GrpcCallSettings<WriteRequest, WriteResponse> writeTransportSettings =
284359
GrpcCallSettings.<WriteRequest, WriteResponse>newBuilder()
285360
.setMethodDescriptor(writeMethodDescriptor)
361+
.setParamsExtractor(
362+
new RequestParamsExtractor<WriteRequest>() {
363+
@Override
364+
public Map<String, String> extract(WriteRequest request) {
365+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
366+
params.put("database", String.valueOf(request.getDatabase()));
367+
return params.build();
368+
}
369+
})
286370
.build();
287371
GrpcCallSettings<ListenRequest, ListenResponse> listenTransportSettings =
288372
GrpcCallSettings.<ListenRequest, ListenResponse>newBuilder()
289373
.setMethodDescriptor(listenMethodDescriptor)
374+
.setParamsExtractor(
375+
new RequestParamsExtractor<ListenRequest>() {
376+
@Override
377+
public Map<String, String> extract(ListenRequest request) {
378+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
379+
params.put("database", String.valueOf(request.getDatabase()));
380+
return params.build();
381+
}
382+
})
290383
.build();
291384
GrpcCallSettings<ListCollectionIdsRequest, ListCollectionIdsResponse>
292385
listCollectionIdsTransportSettings =
293386
GrpcCallSettings.<ListCollectionIdsRequest, ListCollectionIdsResponse>newBuilder()
294387
.setMethodDescriptor(listCollectionIdsMethodDescriptor)
388+
.setParamsExtractor(
389+
new RequestParamsExtractor<ListCollectionIdsRequest>() {
390+
@Override
391+
public Map<String, String> extract(ListCollectionIdsRequest request) {
392+
ImmutableMap.Builder<String, String> params = ImmutableMap.builder();
393+
params.put("parent", String.valueOf(request.getParent()));
394+
return params.build();
395+
}
396+
})
295397
.build();
296398

297399
this.getDocumentCallable =

0 commit comments

Comments
 (0)