|
27 | 27 | import com.google.api.gax.grpc.GrpcStubCallableFactory; |
28 | 28 | import com.google.api.gax.rpc.ClientContext; |
29 | 29 | import com.google.api.gax.rpc.OperationCallable; |
| 30 | +import com.google.api.gax.rpc.RequestParamsExtractor; |
30 | 31 | import com.google.api.gax.rpc.UnaryCallable; |
31 | 32 | import com.google.cloud.vision.v1.AddProductToProductSetRequest; |
32 | 33 | import com.google.cloud.vision.v1.BatchOperationMetadata; |
|
55 | 56 | import com.google.cloud.vision.v1.RemoveProductFromProductSetRequest; |
56 | 57 | import com.google.cloud.vision.v1.UpdateProductRequest; |
57 | 58 | import com.google.cloud.vision.v1.UpdateProductSetRequest; |
| 59 | +import com.google.common.collect.ImmutableMap; |
58 | 60 | import com.google.longrunning.Operation; |
59 | 61 | import com.google.longrunning.stub.GrpcOperationsStub; |
60 | 62 | import com.google.protobuf.Empty; |
61 | 63 | import io.grpc.MethodDescriptor; |
62 | 64 | import io.grpc.protobuf.ProtoUtils; |
63 | 65 | import java.io.IOException; |
| 66 | +import java.util.Map; |
64 | 67 | import java.util.concurrent.TimeUnit; |
65 | 68 | import javax.annotation.Generated; |
66 | 69 |
|
@@ -320,80 +323,244 @@ protected GrpcProductSearchStub( |
320 | 323 | GrpcCallSettings<CreateProductRequest, Product> createProductTransportSettings = |
321 | 324 | GrpcCallSettings.<CreateProductRequest, Product>newBuilder() |
322 | 325 | .setMethodDescriptor(createProductMethodDescriptor) |
| 326 | + .setParamsExtractor( |
| 327 | + new RequestParamsExtractor<CreateProductRequest>() { |
| 328 | + @Override |
| 329 | + public Map<String, String> extract(CreateProductRequest request) { |
| 330 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 331 | + params.put("parent", String.valueOf(request.getParent())); |
| 332 | + return params.build(); |
| 333 | + } |
| 334 | + }) |
323 | 335 | .build(); |
324 | 336 | GrpcCallSettings<ListProductsRequest, ListProductsResponse> listProductsTransportSettings = |
325 | 337 | GrpcCallSettings.<ListProductsRequest, ListProductsResponse>newBuilder() |
326 | 338 | .setMethodDescriptor(listProductsMethodDescriptor) |
| 339 | + .setParamsExtractor( |
| 340 | + new RequestParamsExtractor<ListProductsRequest>() { |
| 341 | + @Override |
| 342 | + public Map<String, String> extract(ListProductsRequest request) { |
| 343 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 344 | + params.put("parent", String.valueOf(request.getParent())); |
| 345 | + return params.build(); |
| 346 | + } |
| 347 | + }) |
327 | 348 | .build(); |
328 | 349 | GrpcCallSettings<GetProductRequest, Product> getProductTransportSettings = |
329 | 350 | GrpcCallSettings.<GetProductRequest, Product>newBuilder() |
330 | 351 | .setMethodDescriptor(getProductMethodDescriptor) |
| 352 | + .setParamsExtractor( |
| 353 | + new RequestParamsExtractor<GetProductRequest>() { |
| 354 | + @Override |
| 355 | + public Map<String, String> extract(GetProductRequest request) { |
| 356 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 357 | + params.put("name", String.valueOf(request.getName())); |
| 358 | + return params.build(); |
| 359 | + } |
| 360 | + }) |
331 | 361 | .build(); |
332 | 362 | GrpcCallSettings<UpdateProductRequest, Product> updateProductTransportSettings = |
333 | 363 | GrpcCallSettings.<UpdateProductRequest, Product>newBuilder() |
334 | 364 | .setMethodDescriptor(updateProductMethodDescriptor) |
| 365 | + .setParamsExtractor( |
| 366 | + new RequestParamsExtractor<UpdateProductRequest>() { |
| 367 | + @Override |
| 368 | + public Map<String, String> extract(UpdateProductRequest request) { |
| 369 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 370 | + params.put("product.name", String.valueOf(request.getProduct().getName())); |
| 371 | + return params.build(); |
| 372 | + } |
| 373 | + }) |
335 | 374 | .build(); |
336 | 375 | GrpcCallSettings<DeleteProductRequest, Empty> deleteProductTransportSettings = |
337 | 376 | GrpcCallSettings.<DeleteProductRequest, Empty>newBuilder() |
338 | 377 | .setMethodDescriptor(deleteProductMethodDescriptor) |
| 378 | + .setParamsExtractor( |
| 379 | + new RequestParamsExtractor<DeleteProductRequest>() { |
| 380 | + @Override |
| 381 | + public Map<String, String> extract(DeleteProductRequest request) { |
| 382 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 383 | + params.put("name", String.valueOf(request.getName())); |
| 384 | + return params.build(); |
| 385 | + } |
| 386 | + }) |
339 | 387 | .build(); |
340 | 388 | GrpcCallSettings<ListReferenceImagesRequest, ListReferenceImagesResponse> |
341 | 389 | listReferenceImagesTransportSettings = |
342 | 390 | GrpcCallSettings.<ListReferenceImagesRequest, ListReferenceImagesResponse>newBuilder() |
343 | 391 | .setMethodDescriptor(listReferenceImagesMethodDescriptor) |
| 392 | + .setParamsExtractor( |
| 393 | + new RequestParamsExtractor<ListReferenceImagesRequest>() { |
| 394 | + @Override |
| 395 | + public Map<String, String> extract(ListReferenceImagesRequest request) { |
| 396 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 397 | + params.put("parent", String.valueOf(request.getParent())); |
| 398 | + return params.build(); |
| 399 | + } |
| 400 | + }) |
344 | 401 | .build(); |
345 | 402 | GrpcCallSettings<GetReferenceImageRequest, ReferenceImage> getReferenceImageTransportSettings = |
346 | 403 | GrpcCallSettings.<GetReferenceImageRequest, ReferenceImage>newBuilder() |
347 | 404 | .setMethodDescriptor(getReferenceImageMethodDescriptor) |
| 405 | + .setParamsExtractor( |
| 406 | + new RequestParamsExtractor<GetReferenceImageRequest>() { |
| 407 | + @Override |
| 408 | + public Map<String, String> extract(GetReferenceImageRequest request) { |
| 409 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 410 | + params.put("name", String.valueOf(request.getName())); |
| 411 | + return params.build(); |
| 412 | + } |
| 413 | + }) |
348 | 414 | .build(); |
349 | 415 | GrpcCallSettings<DeleteReferenceImageRequest, Empty> deleteReferenceImageTransportSettings = |
350 | 416 | GrpcCallSettings.<DeleteReferenceImageRequest, Empty>newBuilder() |
351 | 417 | .setMethodDescriptor(deleteReferenceImageMethodDescriptor) |
| 418 | + .setParamsExtractor( |
| 419 | + new RequestParamsExtractor<DeleteReferenceImageRequest>() { |
| 420 | + @Override |
| 421 | + public Map<String, String> extract(DeleteReferenceImageRequest request) { |
| 422 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 423 | + params.put("name", String.valueOf(request.getName())); |
| 424 | + return params.build(); |
| 425 | + } |
| 426 | + }) |
352 | 427 | .build(); |
353 | 428 | GrpcCallSettings<CreateReferenceImageRequest, ReferenceImage> |
354 | 429 | createReferenceImageTransportSettings = |
355 | 430 | GrpcCallSettings.<CreateReferenceImageRequest, ReferenceImage>newBuilder() |
356 | 431 | .setMethodDescriptor(createReferenceImageMethodDescriptor) |
| 432 | + .setParamsExtractor( |
| 433 | + new RequestParamsExtractor<CreateReferenceImageRequest>() { |
| 434 | + @Override |
| 435 | + public Map<String, String> extract(CreateReferenceImageRequest request) { |
| 436 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 437 | + params.put("parent", String.valueOf(request.getParent())); |
| 438 | + return params.build(); |
| 439 | + } |
| 440 | + }) |
357 | 441 | .build(); |
358 | 442 | GrpcCallSettings<CreateProductSetRequest, ProductSet> createProductSetTransportSettings = |
359 | 443 | GrpcCallSettings.<CreateProductSetRequest, ProductSet>newBuilder() |
360 | 444 | .setMethodDescriptor(createProductSetMethodDescriptor) |
| 445 | + .setParamsExtractor( |
| 446 | + new RequestParamsExtractor<CreateProductSetRequest>() { |
| 447 | + @Override |
| 448 | + public Map<String, String> extract(CreateProductSetRequest request) { |
| 449 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 450 | + params.put("parent", String.valueOf(request.getParent())); |
| 451 | + return params.build(); |
| 452 | + } |
| 453 | + }) |
361 | 454 | .build(); |
362 | 455 | GrpcCallSettings<ListProductSetsRequest, ListProductSetsResponse> |
363 | 456 | listProductSetsTransportSettings = |
364 | 457 | GrpcCallSettings.<ListProductSetsRequest, ListProductSetsResponse>newBuilder() |
365 | 458 | .setMethodDescriptor(listProductSetsMethodDescriptor) |
| 459 | + .setParamsExtractor( |
| 460 | + new RequestParamsExtractor<ListProductSetsRequest>() { |
| 461 | + @Override |
| 462 | + public Map<String, String> extract(ListProductSetsRequest request) { |
| 463 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 464 | + params.put("parent", String.valueOf(request.getParent())); |
| 465 | + return params.build(); |
| 466 | + } |
| 467 | + }) |
366 | 468 | .build(); |
367 | 469 | GrpcCallSettings<GetProductSetRequest, ProductSet> getProductSetTransportSettings = |
368 | 470 | GrpcCallSettings.<GetProductSetRequest, ProductSet>newBuilder() |
369 | 471 | .setMethodDescriptor(getProductSetMethodDescriptor) |
| 472 | + .setParamsExtractor( |
| 473 | + new RequestParamsExtractor<GetProductSetRequest>() { |
| 474 | + @Override |
| 475 | + public Map<String, String> extract(GetProductSetRequest request) { |
| 476 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 477 | + params.put("name", String.valueOf(request.getName())); |
| 478 | + return params.build(); |
| 479 | + } |
| 480 | + }) |
370 | 481 | .build(); |
371 | 482 | GrpcCallSettings<UpdateProductSetRequest, ProductSet> updateProductSetTransportSettings = |
372 | 483 | GrpcCallSettings.<UpdateProductSetRequest, ProductSet>newBuilder() |
373 | 484 | .setMethodDescriptor(updateProductSetMethodDescriptor) |
| 485 | + .setParamsExtractor( |
| 486 | + new RequestParamsExtractor<UpdateProductSetRequest>() { |
| 487 | + @Override |
| 488 | + public Map<String, String> extract(UpdateProductSetRequest request) { |
| 489 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 490 | + params.put( |
| 491 | + "product_set.name", String.valueOf(request.getProductSet().getName())); |
| 492 | + return params.build(); |
| 493 | + } |
| 494 | + }) |
374 | 495 | .build(); |
375 | 496 | GrpcCallSettings<DeleteProductSetRequest, Empty> deleteProductSetTransportSettings = |
376 | 497 | GrpcCallSettings.<DeleteProductSetRequest, Empty>newBuilder() |
377 | 498 | .setMethodDescriptor(deleteProductSetMethodDescriptor) |
| 499 | + .setParamsExtractor( |
| 500 | + new RequestParamsExtractor<DeleteProductSetRequest>() { |
| 501 | + @Override |
| 502 | + public Map<String, String> extract(DeleteProductSetRequest request) { |
| 503 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 504 | + params.put("name", String.valueOf(request.getName())); |
| 505 | + return params.build(); |
| 506 | + } |
| 507 | + }) |
378 | 508 | .build(); |
379 | 509 | GrpcCallSettings<AddProductToProductSetRequest, Empty> addProductToProductSetTransportSettings = |
380 | 510 | GrpcCallSettings.<AddProductToProductSetRequest, Empty>newBuilder() |
381 | 511 | .setMethodDescriptor(addProductToProductSetMethodDescriptor) |
| 512 | + .setParamsExtractor( |
| 513 | + new RequestParamsExtractor<AddProductToProductSetRequest>() { |
| 514 | + @Override |
| 515 | + public Map<String, String> extract(AddProductToProductSetRequest request) { |
| 516 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 517 | + params.put("name", String.valueOf(request.getName())); |
| 518 | + return params.build(); |
| 519 | + } |
| 520 | + }) |
382 | 521 | .build(); |
383 | 522 | GrpcCallSettings<RemoveProductFromProductSetRequest, Empty> |
384 | 523 | removeProductFromProductSetTransportSettings = |
385 | 524 | GrpcCallSettings.<RemoveProductFromProductSetRequest, Empty>newBuilder() |
386 | 525 | .setMethodDescriptor(removeProductFromProductSetMethodDescriptor) |
| 526 | + .setParamsExtractor( |
| 527 | + new RequestParamsExtractor<RemoveProductFromProductSetRequest>() { |
| 528 | + @Override |
| 529 | + public Map<String, String> extract( |
| 530 | + RemoveProductFromProductSetRequest request) { |
| 531 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 532 | + params.put("name", String.valueOf(request.getName())); |
| 533 | + return params.build(); |
| 534 | + } |
| 535 | + }) |
387 | 536 | .build(); |
388 | 537 | GrpcCallSettings<ListProductsInProductSetRequest, ListProductsInProductSetResponse> |
389 | 538 | listProductsInProductSetTransportSettings = |
390 | 539 | GrpcCallSettings |
391 | 540 | .<ListProductsInProductSetRequest, ListProductsInProductSetResponse>newBuilder() |
392 | 541 | .setMethodDescriptor(listProductsInProductSetMethodDescriptor) |
| 542 | + .setParamsExtractor( |
| 543 | + new RequestParamsExtractor<ListProductsInProductSetRequest>() { |
| 544 | + @Override |
| 545 | + public Map<String, String> extract(ListProductsInProductSetRequest request) { |
| 546 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 547 | + params.put("name", String.valueOf(request.getName())); |
| 548 | + return params.build(); |
| 549 | + } |
| 550 | + }) |
393 | 551 | .build(); |
394 | 552 | GrpcCallSettings<ImportProductSetsRequest, Operation> importProductSetsTransportSettings = |
395 | 553 | GrpcCallSettings.<ImportProductSetsRequest, Operation>newBuilder() |
396 | 554 | .setMethodDescriptor(importProductSetsMethodDescriptor) |
| 555 | + .setParamsExtractor( |
| 556 | + new RequestParamsExtractor<ImportProductSetsRequest>() { |
| 557 | + @Override |
| 558 | + public Map<String, String> extract(ImportProductSetsRequest request) { |
| 559 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 560 | + params.put("parent", String.valueOf(request.getParent())); |
| 561 | + return params.build(); |
| 562 | + } |
| 563 | + }) |
397 | 564 | .build(); |
398 | 565 |
|
399 | 566 | this.createProductCallable = |
|
0 commit comments