|
26 | 26 | import com.google.api.gax.grpc.GrpcCallSettings; |
27 | 27 | import com.google.api.gax.grpc.GrpcStubCallableFactory; |
28 | 28 | import com.google.api.gax.rpc.ClientContext; |
29 | | -import com.google.api.gax.rpc.RequestParamsExtractor; |
30 | 29 | import com.google.api.gax.rpc.UnaryCallable; |
31 | 30 | import com.google.cloud.websecurityscanner.v1.CreateScanConfigRequest; |
32 | 31 | import com.google.cloud.websecurityscanner.v1.DeleteScanConfigRequest; |
|
55 | 54 | import io.grpc.MethodDescriptor; |
56 | 55 | import io.grpc.protobuf.ProtoUtils; |
57 | 56 | import java.io.IOException; |
58 | | -import java.util.Map; |
59 | 57 | import java.util.concurrent.TimeUnit; |
60 | 58 | import javax.annotation.Generated; |
61 | 59 |
|
@@ -274,173 +272,133 @@ protected GrpcWebSecurityScannerStub( |
274 | 272 | GrpcCallSettings.<CreateScanConfigRequest, ScanConfig>newBuilder() |
275 | 273 | .setMethodDescriptor(createScanConfigMethodDescriptor) |
276 | 274 | .setParamsExtractor( |
277 | | - new RequestParamsExtractor<CreateScanConfigRequest>() { |
278 | | - @Override |
279 | | - public Map<String, String> extract(CreateScanConfigRequest request) { |
280 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
281 | | - params.put("parent", String.valueOf(request.getParent())); |
282 | | - return params.build(); |
283 | | - } |
| 275 | + request -> { |
| 276 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 277 | + params.put("parent", String.valueOf(request.getParent())); |
| 278 | + return params.build(); |
284 | 279 | }) |
285 | 280 | .build(); |
286 | 281 | GrpcCallSettings<DeleteScanConfigRequest, Empty> deleteScanConfigTransportSettings = |
287 | 282 | GrpcCallSettings.<DeleteScanConfigRequest, Empty>newBuilder() |
288 | 283 | .setMethodDescriptor(deleteScanConfigMethodDescriptor) |
289 | 284 | .setParamsExtractor( |
290 | | - new RequestParamsExtractor<DeleteScanConfigRequest>() { |
291 | | - @Override |
292 | | - public Map<String, String> extract(DeleteScanConfigRequest request) { |
293 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
294 | | - params.put("name", String.valueOf(request.getName())); |
295 | | - return params.build(); |
296 | | - } |
| 285 | + request -> { |
| 286 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 287 | + params.put("name", String.valueOf(request.getName())); |
| 288 | + return params.build(); |
297 | 289 | }) |
298 | 290 | .build(); |
299 | 291 | GrpcCallSettings<GetScanConfigRequest, ScanConfig> getScanConfigTransportSettings = |
300 | 292 | GrpcCallSettings.<GetScanConfigRequest, ScanConfig>newBuilder() |
301 | 293 | .setMethodDescriptor(getScanConfigMethodDescriptor) |
302 | 294 | .setParamsExtractor( |
303 | | - new RequestParamsExtractor<GetScanConfigRequest>() { |
304 | | - @Override |
305 | | - public Map<String, String> extract(GetScanConfigRequest request) { |
306 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
307 | | - params.put("name", String.valueOf(request.getName())); |
308 | | - return params.build(); |
309 | | - } |
| 295 | + request -> { |
| 296 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 297 | + params.put("name", String.valueOf(request.getName())); |
| 298 | + return params.build(); |
310 | 299 | }) |
311 | 300 | .build(); |
312 | 301 | GrpcCallSettings<ListScanConfigsRequest, ListScanConfigsResponse> |
313 | 302 | listScanConfigsTransportSettings = |
314 | 303 | GrpcCallSettings.<ListScanConfigsRequest, ListScanConfigsResponse>newBuilder() |
315 | 304 | .setMethodDescriptor(listScanConfigsMethodDescriptor) |
316 | 305 | .setParamsExtractor( |
317 | | - new RequestParamsExtractor<ListScanConfigsRequest>() { |
318 | | - @Override |
319 | | - public Map<String, String> extract(ListScanConfigsRequest request) { |
320 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
321 | | - params.put("parent", String.valueOf(request.getParent())); |
322 | | - return params.build(); |
323 | | - } |
| 306 | + request -> { |
| 307 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 308 | + params.put("parent", String.valueOf(request.getParent())); |
| 309 | + return params.build(); |
324 | 310 | }) |
325 | 311 | .build(); |
326 | 312 | GrpcCallSettings<UpdateScanConfigRequest, ScanConfig> updateScanConfigTransportSettings = |
327 | 313 | GrpcCallSettings.<UpdateScanConfigRequest, ScanConfig>newBuilder() |
328 | 314 | .setMethodDescriptor(updateScanConfigMethodDescriptor) |
329 | 315 | .setParamsExtractor( |
330 | | - new RequestParamsExtractor<UpdateScanConfigRequest>() { |
331 | | - @Override |
332 | | - public Map<String, String> extract(UpdateScanConfigRequest request) { |
333 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
334 | | - params.put( |
335 | | - "scan_config.name", String.valueOf(request.getScanConfig().getName())); |
336 | | - return params.build(); |
337 | | - } |
| 316 | + request -> { |
| 317 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 318 | + params.put("scan_config.name", String.valueOf(request.getScanConfig().getName())); |
| 319 | + return params.build(); |
338 | 320 | }) |
339 | 321 | .build(); |
340 | 322 | GrpcCallSettings<StartScanRunRequest, ScanRun> startScanRunTransportSettings = |
341 | 323 | GrpcCallSettings.<StartScanRunRequest, ScanRun>newBuilder() |
342 | 324 | .setMethodDescriptor(startScanRunMethodDescriptor) |
343 | 325 | .setParamsExtractor( |
344 | | - new RequestParamsExtractor<StartScanRunRequest>() { |
345 | | - @Override |
346 | | - public Map<String, String> extract(StartScanRunRequest request) { |
347 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
348 | | - params.put("name", String.valueOf(request.getName())); |
349 | | - return params.build(); |
350 | | - } |
| 326 | + request -> { |
| 327 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 328 | + params.put("name", String.valueOf(request.getName())); |
| 329 | + return params.build(); |
351 | 330 | }) |
352 | 331 | .build(); |
353 | 332 | GrpcCallSettings<GetScanRunRequest, ScanRun> getScanRunTransportSettings = |
354 | 333 | GrpcCallSettings.<GetScanRunRequest, ScanRun>newBuilder() |
355 | 334 | .setMethodDescriptor(getScanRunMethodDescriptor) |
356 | 335 | .setParamsExtractor( |
357 | | - new RequestParamsExtractor<GetScanRunRequest>() { |
358 | | - @Override |
359 | | - public Map<String, String> extract(GetScanRunRequest request) { |
360 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
361 | | - params.put("name", String.valueOf(request.getName())); |
362 | | - return params.build(); |
363 | | - } |
| 336 | + request -> { |
| 337 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 338 | + params.put("name", String.valueOf(request.getName())); |
| 339 | + return params.build(); |
364 | 340 | }) |
365 | 341 | .build(); |
366 | 342 | GrpcCallSettings<ListScanRunsRequest, ListScanRunsResponse> listScanRunsTransportSettings = |
367 | 343 | GrpcCallSettings.<ListScanRunsRequest, ListScanRunsResponse>newBuilder() |
368 | 344 | .setMethodDescriptor(listScanRunsMethodDescriptor) |
369 | 345 | .setParamsExtractor( |
370 | | - new RequestParamsExtractor<ListScanRunsRequest>() { |
371 | | - @Override |
372 | | - public Map<String, String> extract(ListScanRunsRequest request) { |
373 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
374 | | - params.put("parent", String.valueOf(request.getParent())); |
375 | | - return params.build(); |
376 | | - } |
| 346 | + request -> { |
| 347 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 348 | + params.put("parent", String.valueOf(request.getParent())); |
| 349 | + return params.build(); |
377 | 350 | }) |
378 | 351 | .build(); |
379 | 352 | GrpcCallSettings<StopScanRunRequest, ScanRun> stopScanRunTransportSettings = |
380 | 353 | GrpcCallSettings.<StopScanRunRequest, ScanRun>newBuilder() |
381 | 354 | .setMethodDescriptor(stopScanRunMethodDescriptor) |
382 | 355 | .setParamsExtractor( |
383 | | - new RequestParamsExtractor<StopScanRunRequest>() { |
384 | | - @Override |
385 | | - public Map<String, String> extract(StopScanRunRequest request) { |
386 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
387 | | - params.put("name", String.valueOf(request.getName())); |
388 | | - return params.build(); |
389 | | - } |
| 356 | + request -> { |
| 357 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 358 | + params.put("name", String.valueOf(request.getName())); |
| 359 | + return params.build(); |
390 | 360 | }) |
391 | 361 | .build(); |
392 | 362 | GrpcCallSettings<ListCrawledUrlsRequest, ListCrawledUrlsResponse> |
393 | 363 | listCrawledUrlsTransportSettings = |
394 | 364 | GrpcCallSettings.<ListCrawledUrlsRequest, ListCrawledUrlsResponse>newBuilder() |
395 | 365 | .setMethodDescriptor(listCrawledUrlsMethodDescriptor) |
396 | 366 | .setParamsExtractor( |
397 | | - new RequestParamsExtractor<ListCrawledUrlsRequest>() { |
398 | | - @Override |
399 | | - public Map<String, String> extract(ListCrawledUrlsRequest request) { |
400 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
401 | | - params.put("parent", String.valueOf(request.getParent())); |
402 | | - return params.build(); |
403 | | - } |
| 367 | + request -> { |
| 368 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 369 | + params.put("parent", String.valueOf(request.getParent())); |
| 370 | + return params.build(); |
404 | 371 | }) |
405 | 372 | .build(); |
406 | 373 | GrpcCallSettings<GetFindingRequest, Finding> getFindingTransportSettings = |
407 | 374 | GrpcCallSettings.<GetFindingRequest, Finding>newBuilder() |
408 | 375 | .setMethodDescriptor(getFindingMethodDescriptor) |
409 | 376 | .setParamsExtractor( |
410 | | - new RequestParamsExtractor<GetFindingRequest>() { |
411 | | - @Override |
412 | | - public Map<String, String> extract(GetFindingRequest request) { |
413 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
414 | | - params.put("name", String.valueOf(request.getName())); |
415 | | - return params.build(); |
416 | | - } |
| 377 | + request -> { |
| 378 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 379 | + params.put("name", String.valueOf(request.getName())); |
| 380 | + return params.build(); |
417 | 381 | }) |
418 | 382 | .build(); |
419 | 383 | GrpcCallSettings<ListFindingsRequest, ListFindingsResponse> listFindingsTransportSettings = |
420 | 384 | GrpcCallSettings.<ListFindingsRequest, ListFindingsResponse>newBuilder() |
421 | 385 | .setMethodDescriptor(listFindingsMethodDescriptor) |
422 | 386 | .setParamsExtractor( |
423 | | - new RequestParamsExtractor<ListFindingsRequest>() { |
424 | | - @Override |
425 | | - public Map<String, String> extract(ListFindingsRequest request) { |
426 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
427 | | - params.put("parent", String.valueOf(request.getParent())); |
428 | | - return params.build(); |
429 | | - } |
| 387 | + request -> { |
| 388 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 389 | + params.put("parent", String.valueOf(request.getParent())); |
| 390 | + return params.build(); |
430 | 391 | }) |
431 | 392 | .build(); |
432 | 393 | GrpcCallSettings<ListFindingTypeStatsRequest, ListFindingTypeStatsResponse> |
433 | 394 | listFindingTypeStatsTransportSettings = |
434 | 395 | GrpcCallSettings.<ListFindingTypeStatsRequest, ListFindingTypeStatsResponse>newBuilder() |
435 | 396 | .setMethodDescriptor(listFindingTypeStatsMethodDescriptor) |
436 | 397 | .setParamsExtractor( |
437 | | - new RequestParamsExtractor<ListFindingTypeStatsRequest>() { |
438 | | - @Override |
439 | | - public Map<String, String> extract(ListFindingTypeStatsRequest request) { |
440 | | - ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
441 | | - params.put("parent", String.valueOf(request.getParent())); |
442 | | - return params.build(); |
443 | | - } |
| 398 | + request -> { |
| 399 | + ImmutableMap.Builder<String, String> params = ImmutableMap.builder(); |
| 400 | + params.put("parent", String.valueOf(request.getParent())); |
| 401 | + return params.build(); |
444 | 402 | }) |
445 | 403 | .build(); |
446 | 404 |
|
@@ -596,7 +554,13 @@ public UnaryCallable<ListFindingsRequest, ListFindingsPagedResponse> listFinding |
596 | 554 |
|
597 | 555 | @Override |
598 | 556 | public final void close() { |
599 | | - shutdown(); |
| 557 | + try { |
| 558 | + backgroundResources.close(); |
| 559 | + } catch (RuntimeException e) { |
| 560 | + throw e; |
| 561 | + } catch (Exception e) { |
| 562 | + throw new IllegalStateException("Failed to close resource", e); |
| 563 | + } |
600 | 564 | } |
601 | 565 |
|
602 | 566 | @Override |
|
0 commit comments