|
25 | 25 | import com.google.api.gax.grpc.GrpcCallSettings; |
26 | 26 | import com.google.api.gax.grpc.GrpcStubCallableFactory; |
27 | 27 | import com.google.api.gax.rpc.ClientContext; |
| 28 | +import com.google.api.gax.rpc.RequestParamsExtractor; |
28 | 29 | import com.google.api.gax.rpc.UnaryCallable; |
29 | 30 | import com.google.cloud.talent.v4beta1.BatchDeleteJobsRequest; |
30 | 31 | import com.google.cloud.talent.v4beta1.CreateJobRequest; |
|
36 | 37 | import com.google.cloud.talent.v4beta1.SearchJobsRequest; |
37 | 38 | import com.google.cloud.talent.v4beta1.SearchJobsResponse; |
38 | 39 | import com.google.cloud.talent.v4beta1.UpdateJobRequest; |
| 40 | +import com.google.common.collect.ImmutableMap; |
39 | 41 | import com.google.protobuf.Empty; |
40 | 42 | import io.grpc.MethodDescriptor; |
41 | 43 | import io.grpc.protobuf.ProtoUtils; |
42 | 44 | import java.io.IOException; |
| 45 | +import java.util.Map; |
43 | 46 | import java.util.concurrent.TimeUnit; |
44 | 47 | import javax.annotation.Generated; |
45 | 48 |
|
@@ -172,34 +175,106 @@ protected GrpcJobServiceStub( |
172 | 175 | GrpcCallSettings<CreateJobRequest, Job> createJobTransportSettings = |
173 | 176 | GrpcCallSettings.<CreateJobRequest, Job>newBuilder() |
174 | 177 | .setMethodDescriptor(createJobMethodDescriptor) |
| 178 | + .setParamsExtractor( |
| 179 | + new RequestParamsExtractor<CreateJobRequest>() { |
| 180 | + @Override |
| 181 | + public Map<String, String> extract(CreateJobRequest request) { |
| 182 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 183 | + params.put("parent", String.valueOf(request.getParent())); |
| 184 | + return params.build(); |
| 185 | + } |
| 186 | + }) |
175 | 187 | .build(); |
176 | 188 | GrpcCallSettings<GetJobRequest, Job> getJobTransportSettings = |
177 | 189 | GrpcCallSettings.<GetJobRequest, Job>newBuilder() |
178 | 190 | .setMethodDescriptor(getJobMethodDescriptor) |
| 191 | + .setParamsExtractor( |
| 192 | + new RequestParamsExtractor<GetJobRequest>() { |
| 193 | + @Override |
| 194 | + public Map<String, String> extract(GetJobRequest request) { |
| 195 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 196 | + params.put("name", String.valueOf(request.getName())); |
| 197 | + return params.build(); |
| 198 | + } |
| 199 | + }) |
179 | 200 | .build(); |
180 | 201 | GrpcCallSettings<UpdateJobRequest, Job> updateJobTransportSettings = |
181 | 202 | GrpcCallSettings.<UpdateJobRequest, Job>newBuilder() |
182 | 203 | .setMethodDescriptor(updateJobMethodDescriptor) |
| 204 | + .setParamsExtractor( |
| 205 | + new RequestParamsExtractor<UpdateJobRequest>() { |
| 206 | + @Override |
| 207 | + public Map<String, String> extract(UpdateJobRequest request) { |
| 208 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 209 | + params.put("job.name", String.valueOf(request.getJob().getName())); |
| 210 | + return params.build(); |
| 211 | + } |
| 212 | + }) |
183 | 213 | .build(); |
184 | 214 | GrpcCallSettings<DeleteJobRequest, Empty> deleteJobTransportSettings = |
185 | 215 | GrpcCallSettings.<DeleteJobRequest, Empty>newBuilder() |
186 | 216 | .setMethodDescriptor(deleteJobMethodDescriptor) |
| 217 | + .setParamsExtractor( |
| 218 | + new RequestParamsExtractor<DeleteJobRequest>() { |
| 219 | + @Override |
| 220 | + public Map<String, String> extract(DeleteJobRequest request) { |
| 221 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 222 | + params.put("name", String.valueOf(request.getName())); |
| 223 | + return params.build(); |
| 224 | + } |
| 225 | + }) |
187 | 226 | .build(); |
188 | 227 | GrpcCallSettings<ListJobsRequest, ListJobsResponse> listJobsTransportSettings = |
189 | 228 | GrpcCallSettings.<ListJobsRequest, ListJobsResponse>newBuilder() |
190 | 229 | .setMethodDescriptor(listJobsMethodDescriptor) |
| 230 | + .setParamsExtractor( |
| 231 | + new RequestParamsExtractor<ListJobsRequest>() { |
| 232 | + @Override |
| 233 | + public Map<String, String> extract(ListJobsRequest request) { |
| 234 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 235 | + params.put("parent", String.valueOf(request.getParent())); |
| 236 | + return params.build(); |
| 237 | + } |
| 238 | + }) |
191 | 239 | .build(); |
192 | 240 | GrpcCallSettings<BatchDeleteJobsRequest, Empty> batchDeleteJobsTransportSettings = |
193 | 241 | GrpcCallSettings.<BatchDeleteJobsRequest, Empty>newBuilder() |
194 | 242 | .setMethodDescriptor(batchDeleteJobsMethodDescriptor) |
| 243 | + .setParamsExtractor( |
| 244 | + new RequestParamsExtractor<BatchDeleteJobsRequest>() { |
| 245 | + @Override |
| 246 | + public Map<String, String> extract(BatchDeleteJobsRequest request) { |
| 247 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 248 | + params.put("parent", String.valueOf(request.getParent())); |
| 249 | + return params.build(); |
| 250 | + } |
| 251 | + }) |
195 | 252 | .build(); |
196 | 253 | GrpcCallSettings<SearchJobsRequest, SearchJobsResponse> searchJobsTransportSettings = |
197 | 254 | GrpcCallSettings.<SearchJobsRequest, SearchJobsResponse>newBuilder() |
198 | 255 | .setMethodDescriptor(searchJobsMethodDescriptor) |
| 256 | + .setParamsExtractor( |
| 257 | + new RequestParamsExtractor<SearchJobsRequest>() { |
| 258 | + @Override |
| 259 | + public Map<String, String> extract(SearchJobsRequest request) { |
| 260 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 261 | + params.put("parent", String.valueOf(request.getParent())); |
| 262 | + return params.build(); |
| 263 | + } |
| 264 | + }) |
199 | 265 | .build(); |
200 | 266 | GrpcCallSettings<SearchJobsRequest, SearchJobsResponse> searchJobsForAlertTransportSettings = |
201 | 267 | GrpcCallSettings.<SearchJobsRequest, SearchJobsResponse>newBuilder() |
202 | 268 | .setMethodDescriptor(searchJobsForAlertMethodDescriptor) |
| 269 | + .setParamsExtractor( |
| 270 | + new RequestParamsExtractor<SearchJobsRequest>() { |
| 271 | + @Override |
| 272 | + public Map<String, String> extract(SearchJobsRequest request) { |
| 273 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 274 | + params.put("parent", String.valueOf(request.getParent())); |
| 275 | + return params.build(); |
| 276 | + } |
| 277 | + }) |
203 | 278 | .build(); |
204 | 279 |
|
205 | 280 | this.createJobCallable = |
|
0 commit comments