|
22 | 22 | import com.google.api.gax.grpc.GrpcStubCallableFactory; |
23 | 23 | import com.google.api.gax.rpc.ClientContext; |
24 | 24 | import com.google.api.gax.rpc.OperationCallable; |
| 25 | +import com.google.api.gax.rpc.RequestParamsExtractor; |
25 | 26 | import com.google.api.gax.rpc.UnaryCallable; |
26 | 27 | import com.google.cloud.asset.v1beta1.BatchGetAssetsHistoryRequest; |
27 | 28 | import com.google.cloud.asset.v1beta1.BatchGetAssetsHistoryResponse; |
28 | 29 | import com.google.cloud.asset.v1beta1.ExportAssetsRequest; |
29 | 30 | import com.google.cloud.asset.v1beta1.ExportAssetsResponse; |
| 31 | +import com.google.common.collect.ImmutableMap; |
30 | 32 | import com.google.longrunning.Operation; |
31 | 33 | import com.google.longrunning.stub.GrpcOperationsStub; |
32 | 34 | import io.grpc.MethodDescriptor; |
33 | 35 | import io.grpc.protobuf.ProtoUtils; |
34 | 36 | import java.io.IOException; |
| 37 | +import java.util.Map; |
35 | 38 | import java.util.concurrent.TimeUnit; |
36 | 39 | import javax.annotation.Generated; |
37 | 40 |
|
@@ -116,12 +119,30 @@ protected GrpcAssetServiceStub( |
116 | 119 | GrpcCallSettings<ExportAssetsRequest, Operation> exportAssetsTransportSettings = |
117 | 120 | GrpcCallSettings.<ExportAssetsRequest, Operation>newBuilder() |
118 | 121 | .setMethodDescriptor(exportAssetsMethodDescriptor) |
| 122 | + .setParamsExtractor( |
| 123 | + new RequestParamsExtractor<ExportAssetsRequest>() { |
| 124 | + @Override |
| 125 | + public Map<String, String> extract(ExportAssetsRequest request) { |
| 126 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 127 | + params.put("parent", String.valueOf(request.getParent())); |
| 128 | + return params.build(); |
| 129 | + } |
| 130 | + }) |
119 | 131 | .build(); |
120 | 132 | GrpcCallSettings<BatchGetAssetsHistoryRequest, BatchGetAssetsHistoryResponse> |
121 | 133 | batchGetAssetsHistoryTransportSettings = |
122 | 134 | GrpcCallSettings |
123 | 135 | .<BatchGetAssetsHistoryRequest, BatchGetAssetsHistoryResponse>newBuilder() |
124 | 136 | .setMethodDescriptor(batchGetAssetsHistoryMethodDescriptor) |
| 137 | + .setParamsExtractor( |
| 138 | + new RequestParamsExtractor<BatchGetAssetsHistoryRequest>() { |
| 139 | + @Override |
| 140 | + public Map<String, String> extract(BatchGetAssetsHistoryRequest request) { |
| 141 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 142 | + params.put("parent", String.valueOf(request.getParent())); |
| 143 | + return params.build(); |
| 144 | + } |
| 145 | + }) |
125 | 146 | .build(); |
126 | 147 |
|
127 | 148 | this.exportAssetsCallable = |
|
0 commit comments