1 /* 2 * Copyright 2022 Google LLC 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * https://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17 package com.google.cloud.dlp.v2; 18 19 import com.google.api.core.BetaApi; 20 import com.google.privacy.dlp.v2.ActivateJobTriggerRequest; 21 import com.google.privacy.dlp.v2.CancelDlpJobRequest; 22 import com.google.privacy.dlp.v2.CreateDeidentifyTemplateRequest; 23 import com.google.privacy.dlp.v2.CreateDlpJobRequest; 24 import com.google.privacy.dlp.v2.CreateInspectTemplateRequest; 25 import com.google.privacy.dlp.v2.CreateJobTriggerRequest; 26 import com.google.privacy.dlp.v2.CreateStoredInfoTypeRequest; 27 import com.google.privacy.dlp.v2.DeidentifyContentRequest; 28 import com.google.privacy.dlp.v2.DeidentifyContentResponse; 29 import com.google.privacy.dlp.v2.DeidentifyTemplate; 30 import com.google.privacy.dlp.v2.DeleteDeidentifyTemplateRequest; 31 import com.google.privacy.dlp.v2.DeleteDlpJobRequest; 32 import com.google.privacy.dlp.v2.DeleteInspectTemplateRequest; 33 import com.google.privacy.dlp.v2.DeleteJobTriggerRequest; 34 import com.google.privacy.dlp.v2.DeleteStoredInfoTypeRequest; 35 import com.google.privacy.dlp.v2.DlpJob; 36 import com.google.privacy.dlp.v2.DlpServiceGrpc.DlpServiceImplBase; 37 import com.google.privacy.dlp.v2.FinishDlpJobRequest; 38 import com.google.privacy.dlp.v2.GetDeidentifyTemplateRequest; 39 import com.google.privacy.dlp.v2.GetDlpJobRequest; 40 import com.google.privacy.dlp.v2.GetInspectTemplateRequest; 41 import com.google.privacy.dlp.v2.GetJobTriggerRequest; 42 import com.google.privacy.dlp.v2.GetStoredInfoTypeRequest; 43 import com.google.privacy.dlp.v2.HybridInspectDlpJobRequest; 44 import com.google.privacy.dlp.v2.HybridInspectJobTriggerRequest; 45 import com.google.privacy.dlp.v2.HybridInspectResponse; 46 import com.google.privacy.dlp.v2.InspectContentRequest; 47 import com.google.privacy.dlp.v2.InspectContentResponse; 48 import com.google.privacy.dlp.v2.InspectTemplate; 49 import com.google.privacy.dlp.v2.JobTrigger; 50 import com.google.privacy.dlp.v2.ListDeidentifyTemplatesRequest; 51 import com.google.privacy.dlp.v2.ListDeidentifyTemplatesResponse; 52 import com.google.privacy.dlp.v2.ListDlpJobsRequest; 53 import com.google.privacy.dlp.v2.ListDlpJobsResponse; 54 import com.google.privacy.dlp.v2.ListInfoTypesRequest; 55 import com.google.privacy.dlp.v2.ListInfoTypesResponse; 56 import com.google.privacy.dlp.v2.ListInspectTemplatesRequest; 57 import com.google.privacy.dlp.v2.ListInspectTemplatesResponse; 58 import com.google.privacy.dlp.v2.ListJobTriggersRequest; 59 import com.google.privacy.dlp.v2.ListJobTriggersResponse; 60 import com.google.privacy.dlp.v2.ListStoredInfoTypesRequest; 61 import com.google.privacy.dlp.v2.ListStoredInfoTypesResponse; 62 import com.google.privacy.dlp.v2.RedactImageRequest; 63 import com.google.privacy.dlp.v2.RedactImageResponse; 64 import com.google.privacy.dlp.v2.ReidentifyContentRequest; 65 import com.google.privacy.dlp.v2.ReidentifyContentResponse; 66 import com.google.privacy.dlp.v2.StoredInfoType; 67 import com.google.privacy.dlp.v2.UpdateDeidentifyTemplateRequest; 68 import com.google.privacy.dlp.v2.UpdateInspectTemplateRequest; 69 import com.google.privacy.dlp.v2.UpdateJobTriggerRequest; 70 import com.google.privacy.dlp.v2.UpdateStoredInfoTypeRequest; 71 import com.google.protobuf.AbstractMessage; 72 import com.google.protobuf.Empty; 73 import io.grpc.stub.StreamObserver; 74 import java.util.ArrayList; 75 import java.util.LinkedList; 76 import java.util.List; 77 import java.util.Queue; 78 import javax.annotation.Generated; 79 80 @BetaApi 81 @Generated("by gapic-generator-java") 82 public class MockDlpServiceImpl extends DlpServiceImplBase { 83 private List<AbstractMessage> requests; 84 private Queue<Object> responses; 85 MockDlpServiceImpl()86 public MockDlpServiceImpl() { 87 requests = new ArrayList<>(); 88 responses = new LinkedList<>(); 89 } 90 getRequests()91 public List<AbstractMessage> getRequests() { 92 return requests; 93 } 94 addResponse(AbstractMessage response)95 public void addResponse(AbstractMessage response) { 96 responses.add(response); 97 } 98 setResponses(List<AbstractMessage> responses)99 public void setResponses(List<AbstractMessage> responses) { 100 this.responses = new LinkedList<Object>(responses); 101 } 102 addException(Exception exception)103 public void addException(Exception exception) { 104 responses.add(exception); 105 } 106 reset()107 public void reset() { 108 requests = new ArrayList<>(); 109 responses = new LinkedList<>(); 110 } 111 112 @Override inspectContent( InspectContentRequest request, StreamObserver<InspectContentResponse> responseObserver)113 public void inspectContent( 114 InspectContentRequest request, StreamObserver<InspectContentResponse> responseObserver) { 115 Object response = responses.poll(); 116 if (response instanceof InspectContentResponse) { 117 requests.add(request); 118 responseObserver.onNext(((InspectContentResponse) response)); 119 responseObserver.onCompleted(); 120 } else if (response instanceof Exception) { 121 responseObserver.onError(((Exception) response)); 122 } else { 123 responseObserver.onError( 124 new IllegalArgumentException( 125 String.format( 126 "Unrecognized response type %s for method InspectContent, expected %s or %s", 127 response == null ? "null" : response.getClass().getName(), 128 InspectContentResponse.class.getName(), 129 Exception.class.getName()))); 130 } 131 } 132 133 @Override redactImage( RedactImageRequest request, StreamObserver<RedactImageResponse> responseObserver)134 public void redactImage( 135 RedactImageRequest request, StreamObserver<RedactImageResponse> responseObserver) { 136 Object response = responses.poll(); 137 if (response instanceof RedactImageResponse) { 138 requests.add(request); 139 responseObserver.onNext(((RedactImageResponse) response)); 140 responseObserver.onCompleted(); 141 } else if (response instanceof Exception) { 142 responseObserver.onError(((Exception) response)); 143 } else { 144 responseObserver.onError( 145 new IllegalArgumentException( 146 String.format( 147 "Unrecognized response type %s for method RedactImage, expected %s or %s", 148 response == null ? "null" : response.getClass().getName(), 149 RedactImageResponse.class.getName(), 150 Exception.class.getName()))); 151 } 152 } 153 154 @Override deidentifyContent( DeidentifyContentRequest request, StreamObserver<DeidentifyContentResponse> responseObserver)155 public void deidentifyContent( 156 DeidentifyContentRequest request, 157 StreamObserver<DeidentifyContentResponse> responseObserver) { 158 Object response = responses.poll(); 159 if (response instanceof DeidentifyContentResponse) { 160 requests.add(request); 161 responseObserver.onNext(((DeidentifyContentResponse) response)); 162 responseObserver.onCompleted(); 163 } else if (response instanceof Exception) { 164 responseObserver.onError(((Exception) response)); 165 } else { 166 responseObserver.onError( 167 new IllegalArgumentException( 168 String.format( 169 "Unrecognized response type %s for method DeidentifyContent, expected %s or %s", 170 response == null ? "null" : response.getClass().getName(), 171 DeidentifyContentResponse.class.getName(), 172 Exception.class.getName()))); 173 } 174 } 175 176 @Override reidentifyContent( ReidentifyContentRequest request, StreamObserver<ReidentifyContentResponse> responseObserver)177 public void reidentifyContent( 178 ReidentifyContentRequest request, 179 StreamObserver<ReidentifyContentResponse> responseObserver) { 180 Object response = responses.poll(); 181 if (response instanceof ReidentifyContentResponse) { 182 requests.add(request); 183 responseObserver.onNext(((ReidentifyContentResponse) response)); 184 responseObserver.onCompleted(); 185 } else if (response instanceof Exception) { 186 responseObserver.onError(((Exception) response)); 187 } else { 188 responseObserver.onError( 189 new IllegalArgumentException( 190 String.format( 191 "Unrecognized response type %s for method ReidentifyContent, expected %s or %s", 192 response == null ? "null" : response.getClass().getName(), 193 ReidentifyContentResponse.class.getName(), 194 Exception.class.getName()))); 195 } 196 } 197 198 @Override listInfoTypes( ListInfoTypesRequest request, StreamObserver<ListInfoTypesResponse> responseObserver)199 public void listInfoTypes( 200 ListInfoTypesRequest request, StreamObserver<ListInfoTypesResponse> responseObserver) { 201 Object response = responses.poll(); 202 if (response instanceof ListInfoTypesResponse) { 203 requests.add(request); 204 responseObserver.onNext(((ListInfoTypesResponse) response)); 205 responseObserver.onCompleted(); 206 } else if (response instanceof Exception) { 207 responseObserver.onError(((Exception) response)); 208 } else { 209 responseObserver.onError( 210 new IllegalArgumentException( 211 String.format( 212 "Unrecognized response type %s for method ListInfoTypes, expected %s or %s", 213 response == null ? "null" : response.getClass().getName(), 214 ListInfoTypesResponse.class.getName(), 215 Exception.class.getName()))); 216 } 217 } 218 219 @Override createInspectTemplate( CreateInspectTemplateRequest request, StreamObserver<InspectTemplate> responseObserver)220 public void createInspectTemplate( 221 CreateInspectTemplateRequest request, StreamObserver<InspectTemplate> responseObserver) { 222 Object response = responses.poll(); 223 if (response instanceof InspectTemplate) { 224 requests.add(request); 225 responseObserver.onNext(((InspectTemplate) response)); 226 responseObserver.onCompleted(); 227 } else if (response instanceof Exception) { 228 responseObserver.onError(((Exception) response)); 229 } else { 230 responseObserver.onError( 231 new IllegalArgumentException( 232 String.format( 233 "Unrecognized response type %s for method CreateInspectTemplate, expected %s or %s", 234 response == null ? "null" : response.getClass().getName(), 235 InspectTemplate.class.getName(), 236 Exception.class.getName()))); 237 } 238 } 239 240 @Override updateInspectTemplate( UpdateInspectTemplateRequest request, StreamObserver<InspectTemplate> responseObserver)241 public void updateInspectTemplate( 242 UpdateInspectTemplateRequest request, StreamObserver<InspectTemplate> responseObserver) { 243 Object response = responses.poll(); 244 if (response instanceof InspectTemplate) { 245 requests.add(request); 246 responseObserver.onNext(((InspectTemplate) response)); 247 responseObserver.onCompleted(); 248 } else if (response instanceof Exception) { 249 responseObserver.onError(((Exception) response)); 250 } else { 251 responseObserver.onError( 252 new IllegalArgumentException( 253 String.format( 254 "Unrecognized response type %s for method UpdateInspectTemplate, expected %s or %s", 255 response == null ? "null" : response.getClass().getName(), 256 InspectTemplate.class.getName(), 257 Exception.class.getName()))); 258 } 259 } 260 261 @Override getInspectTemplate( GetInspectTemplateRequest request, StreamObserver<InspectTemplate> responseObserver)262 public void getInspectTemplate( 263 GetInspectTemplateRequest request, StreamObserver<InspectTemplate> responseObserver) { 264 Object response = responses.poll(); 265 if (response instanceof InspectTemplate) { 266 requests.add(request); 267 responseObserver.onNext(((InspectTemplate) response)); 268 responseObserver.onCompleted(); 269 } else if (response instanceof Exception) { 270 responseObserver.onError(((Exception) response)); 271 } else { 272 responseObserver.onError( 273 new IllegalArgumentException( 274 String.format( 275 "Unrecognized response type %s for method GetInspectTemplate, expected %s or %s", 276 response == null ? "null" : response.getClass().getName(), 277 InspectTemplate.class.getName(), 278 Exception.class.getName()))); 279 } 280 } 281 282 @Override listInspectTemplates( ListInspectTemplatesRequest request, StreamObserver<ListInspectTemplatesResponse> responseObserver)283 public void listInspectTemplates( 284 ListInspectTemplatesRequest request, 285 StreamObserver<ListInspectTemplatesResponse> responseObserver) { 286 Object response = responses.poll(); 287 if (response instanceof ListInspectTemplatesResponse) { 288 requests.add(request); 289 responseObserver.onNext(((ListInspectTemplatesResponse) response)); 290 responseObserver.onCompleted(); 291 } else if (response instanceof Exception) { 292 responseObserver.onError(((Exception) response)); 293 } else { 294 responseObserver.onError( 295 new IllegalArgumentException( 296 String.format( 297 "Unrecognized response type %s for method ListInspectTemplates, expected %s or %s", 298 response == null ? "null" : response.getClass().getName(), 299 ListInspectTemplatesResponse.class.getName(), 300 Exception.class.getName()))); 301 } 302 } 303 304 @Override deleteInspectTemplate( DeleteInspectTemplateRequest request, StreamObserver<Empty> responseObserver)305 public void deleteInspectTemplate( 306 DeleteInspectTemplateRequest request, StreamObserver<Empty> responseObserver) { 307 Object response = responses.poll(); 308 if (response instanceof Empty) { 309 requests.add(request); 310 responseObserver.onNext(((Empty) response)); 311 responseObserver.onCompleted(); 312 } else if (response instanceof Exception) { 313 responseObserver.onError(((Exception) response)); 314 } else { 315 responseObserver.onError( 316 new IllegalArgumentException( 317 String.format( 318 "Unrecognized response type %s for method DeleteInspectTemplate, expected %s or %s", 319 response == null ? "null" : response.getClass().getName(), 320 Empty.class.getName(), 321 Exception.class.getName()))); 322 } 323 } 324 325 @Override createDeidentifyTemplate( CreateDeidentifyTemplateRequest request, StreamObserver<DeidentifyTemplate> responseObserver)326 public void createDeidentifyTemplate( 327 CreateDeidentifyTemplateRequest request, 328 StreamObserver<DeidentifyTemplate> responseObserver) { 329 Object response = responses.poll(); 330 if (response instanceof DeidentifyTemplate) { 331 requests.add(request); 332 responseObserver.onNext(((DeidentifyTemplate) response)); 333 responseObserver.onCompleted(); 334 } else if (response instanceof Exception) { 335 responseObserver.onError(((Exception) response)); 336 } else { 337 responseObserver.onError( 338 new IllegalArgumentException( 339 String.format( 340 "Unrecognized response type %s for method CreateDeidentifyTemplate, expected %s or %s", 341 response == null ? "null" : response.getClass().getName(), 342 DeidentifyTemplate.class.getName(), 343 Exception.class.getName()))); 344 } 345 } 346 347 @Override updateDeidentifyTemplate( UpdateDeidentifyTemplateRequest request, StreamObserver<DeidentifyTemplate> responseObserver)348 public void updateDeidentifyTemplate( 349 UpdateDeidentifyTemplateRequest request, 350 StreamObserver<DeidentifyTemplate> responseObserver) { 351 Object response = responses.poll(); 352 if (response instanceof DeidentifyTemplate) { 353 requests.add(request); 354 responseObserver.onNext(((DeidentifyTemplate) response)); 355 responseObserver.onCompleted(); 356 } else if (response instanceof Exception) { 357 responseObserver.onError(((Exception) response)); 358 } else { 359 responseObserver.onError( 360 new IllegalArgumentException( 361 String.format( 362 "Unrecognized response type %s for method UpdateDeidentifyTemplate, expected %s or %s", 363 response == null ? "null" : response.getClass().getName(), 364 DeidentifyTemplate.class.getName(), 365 Exception.class.getName()))); 366 } 367 } 368 369 @Override getDeidentifyTemplate( GetDeidentifyTemplateRequest request, StreamObserver<DeidentifyTemplate> responseObserver)370 public void getDeidentifyTemplate( 371 GetDeidentifyTemplateRequest request, StreamObserver<DeidentifyTemplate> responseObserver) { 372 Object response = responses.poll(); 373 if (response instanceof DeidentifyTemplate) { 374 requests.add(request); 375 responseObserver.onNext(((DeidentifyTemplate) response)); 376 responseObserver.onCompleted(); 377 } else if (response instanceof Exception) { 378 responseObserver.onError(((Exception) response)); 379 } else { 380 responseObserver.onError( 381 new IllegalArgumentException( 382 String.format( 383 "Unrecognized response type %s for method GetDeidentifyTemplate, expected %s or %s", 384 response == null ? "null" : response.getClass().getName(), 385 DeidentifyTemplate.class.getName(), 386 Exception.class.getName()))); 387 } 388 } 389 390 @Override listDeidentifyTemplates( ListDeidentifyTemplatesRequest request, StreamObserver<ListDeidentifyTemplatesResponse> responseObserver)391 public void listDeidentifyTemplates( 392 ListDeidentifyTemplatesRequest request, 393 StreamObserver<ListDeidentifyTemplatesResponse> responseObserver) { 394 Object response = responses.poll(); 395 if (response instanceof ListDeidentifyTemplatesResponse) { 396 requests.add(request); 397 responseObserver.onNext(((ListDeidentifyTemplatesResponse) response)); 398 responseObserver.onCompleted(); 399 } else if (response instanceof Exception) { 400 responseObserver.onError(((Exception) response)); 401 } else { 402 responseObserver.onError( 403 new IllegalArgumentException( 404 String.format( 405 "Unrecognized response type %s for method ListDeidentifyTemplates, expected %s or %s", 406 response == null ? "null" : response.getClass().getName(), 407 ListDeidentifyTemplatesResponse.class.getName(), 408 Exception.class.getName()))); 409 } 410 } 411 412 @Override deleteDeidentifyTemplate( DeleteDeidentifyTemplateRequest request, StreamObserver<Empty> responseObserver)413 public void deleteDeidentifyTemplate( 414 DeleteDeidentifyTemplateRequest request, StreamObserver<Empty> responseObserver) { 415 Object response = responses.poll(); 416 if (response instanceof Empty) { 417 requests.add(request); 418 responseObserver.onNext(((Empty) response)); 419 responseObserver.onCompleted(); 420 } else if (response instanceof Exception) { 421 responseObserver.onError(((Exception) response)); 422 } else { 423 responseObserver.onError( 424 new IllegalArgumentException( 425 String.format( 426 "Unrecognized response type %s for method DeleteDeidentifyTemplate, expected %s or %s", 427 response == null ? "null" : response.getClass().getName(), 428 Empty.class.getName(), 429 Exception.class.getName()))); 430 } 431 } 432 433 @Override createJobTrigger( CreateJobTriggerRequest request, StreamObserver<JobTrigger> responseObserver)434 public void createJobTrigger( 435 CreateJobTriggerRequest request, StreamObserver<JobTrigger> responseObserver) { 436 Object response = responses.poll(); 437 if (response instanceof JobTrigger) { 438 requests.add(request); 439 responseObserver.onNext(((JobTrigger) response)); 440 responseObserver.onCompleted(); 441 } else if (response instanceof Exception) { 442 responseObserver.onError(((Exception) response)); 443 } else { 444 responseObserver.onError( 445 new IllegalArgumentException( 446 String.format( 447 "Unrecognized response type %s for method CreateJobTrigger, expected %s or %s", 448 response == null ? "null" : response.getClass().getName(), 449 JobTrigger.class.getName(), 450 Exception.class.getName()))); 451 } 452 } 453 454 @Override updateJobTrigger( UpdateJobTriggerRequest request, StreamObserver<JobTrigger> responseObserver)455 public void updateJobTrigger( 456 UpdateJobTriggerRequest request, StreamObserver<JobTrigger> responseObserver) { 457 Object response = responses.poll(); 458 if (response instanceof JobTrigger) { 459 requests.add(request); 460 responseObserver.onNext(((JobTrigger) response)); 461 responseObserver.onCompleted(); 462 } else if (response instanceof Exception) { 463 responseObserver.onError(((Exception) response)); 464 } else { 465 responseObserver.onError( 466 new IllegalArgumentException( 467 String.format( 468 "Unrecognized response type %s for method UpdateJobTrigger, expected %s or %s", 469 response == null ? "null" : response.getClass().getName(), 470 JobTrigger.class.getName(), 471 Exception.class.getName()))); 472 } 473 } 474 475 @Override hybridInspectJobTrigger( HybridInspectJobTriggerRequest request, StreamObserver<HybridInspectResponse> responseObserver)476 public void hybridInspectJobTrigger( 477 HybridInspectJobTriggerRequest request, 478 StreamObserver<HybridInspectResponse> responseObserver) { 479 Object response = responses.poll(); 480 if (response instanceof HybridInspectResponse) { 481 requests.add(request); 482 responseObserver.onNext(((HybridInspectResponse) response)); 483 responseObserver.onCompleted(); 484 } else if (response instanceof Exception) { 485 responseObserver.onError(((Exception) response)); 486 } else { 487 responseObserver.onError( 488 new IllegalArgumentException( 489 String.format( 490 "Unrecognized response type %s for method HybridInspectJobTrigger, expected %s or %s", 491 response == null ? "null" : response.getClass().getName(), 492 HybridInspectResponse.class.getName(), 493 Exception.class.getName()))); 494 } 495 } 496 497 @Override getJobTrigger( GetJobTriggerRequest request, StreamObserver<JobTrigger> responseObserver)498 public void getJobTrigger( 499 GetJobTriggerRequest request, StreamObserver<JobTrigger> responseObserver) { 500 Object response = responses.poll(); 501 if (response instanceof JobTrigger) { 502 requests.add(request); 503 responseObserver.onNext(((JobTrigger) response)); 504 responseObserver.onCompleted(); 505 } else if (response instanceof Exception) { 506 responseObserver.onError(((Exception) response)); 507 } else { 508 responseObserver.onError( 509 new IllegalArgumentException( 510 String.format( 511 "Unrecognized response type %s for method GetJobTrigger, expected %s or %s", 512 response == null ? "null" : response.getClass().getName(), 513 JobTrigger.class.getName(), 514 Exception.class.getName()))); 515 } 516 } 517 518 @Override listJobTriggers( ListJobTriggersRequest request, StreamObserver<ListJobTriggersResponse> responseObserver)519 public void listJobTriggers( 520 ListJobTriggersRequest request, StreamObserver<ListJobTriggersResponse> responseObserver) { 521 Object response = responses.poll(); 522 if (response instanceof ListJobTriggersResponse) { 523 requests.add(request); 524 responseObserver.onNext(((ListJobTriggersResponse) response)); 525 responseObserver.onCompleted(); 526 } else if (response instanceof Exception) { 527 responseObserver.onError(((Exception) response)); 528 } else { 529 responseObserver.onError( 530 new IllegalArgumentException( 531 String.format( 532 "Unrecognized response type %s for method ListJobTriggers, expected %s or %s", 533 response == null ? "null" : response.getClass().getName(), 534 ListJobTriggersResponse.class.getName(), 535 Exception.class.getName()))); 536 } 537 } 538 539 @Override deleteJobTrigger( DeleteJobTriggerRequest request, StreamObserver<Empty> responseObserver)540 public void deleteJobTrigger( 541 DeleteJobTriggerRequest request, StreamObserver<Empty> responseObserver) { 542 Object response = responses.poll(); 543 if (response instanceof Empty) { 544 requests.add(request); 545 responseObserver.onNext(((Empty) response)); 546 responseObserver.onCompleted(); 547 } else if (response instanceof Exception) { 548 responseObserver.onError(((Exception) response)); 549 } else { 550 responseObserver.onError( 551 new IllegalArgumentException( 552 String.format( 553 "Unrecognized response type %s for method DeleteJobTrigger, expected %s or %s", 554 response == null ? "null" : response.getClass().getName(), 555 Empty.class.getName(), 556 Exception.class.getName()))); 557 } 558 } 559 560 @Override activateJobTrigger( ActivateJobTriggerRequest request, StreamObserver<DlpJob> responseObserver)561 public void activateJobTrigger( 562 ActivateJobTriggerRequest request, StreamObserver<DlpJob> responseObserver) { 563 Object response = responses.poll(); 564 if (response instanceof DlpJob) { 565 requests.add(request); 566 responseObserver.onNext(((DlpJob) response)); 567 responseObserver.onCompleted(); 568 } else if (response instanceof Exception) { 569 responseObserver.onError(((Exception) response)); 570 } else { 571 responseObserver.onError( 572 new IllegalArgumentException( 573 String.format( 574 "Unrecognized response type %s for method ActivateJobTrigger, expected %s or %s", 575 response == null ? "null" : response.getClass().getName(), 576 DlpJob.class.getName(), 577 Exception.class.getName()))); 578 } 579 } 580 581 @Override createDlpJob(CreateDlpJobRequest request, StreamObserver<DlpJob> responseObserver)582 public void createDlpJob(CreateDlpJobRequest request, StreamObserver<DlpJob> responseObserver) { 583 Object response = responses.poll(); 584 if (response instanceof DlpJob) { 585 requests.add(request); 586 responseObserver.onNext(((DlpJob) response)); 587 responseObserver.onCompleted(); 588 } else if (response instanceof Exception) { 589 responseObserver.onError(((Exception) response)); 590 } else { 591 responseObserver.onError( 592 new IllegalArgumentException( 593 String.format( 594 "Unrecognized response type %s for method CreateDlpJob, expected %s or %s", 595 response == null ? "null" : response.getClass().getName(), 596 DlpJob.class.getName(), 597 Exception.class.getName()))); 598 } 599 } 600 601 @Override listDlpJobs( ListDlpJobsRequest request, StreamObserver<ListDlpJobsResponse> responseObserver)602 public void listDlpJobs( 603 ListDlpJobsRequest request, StreamObserver<ListDlpJobsResponse> responseObserver) { 604 Object response = responses.poll(); 605 if (response instanceof ListDlpJobsResponse) { 606 requests.add(request); 607 responseObserver.onNext(((ListDlpJobsResponse) response)); 608 responseObserver.onCompleted(); 609 } else if (response instanceof Exception) { 610 responseObserver.onError(((Exception) response)); 611 } else { 612 responseObserver.onError( 613 new IllegalArgumentException( 614 String.format( 615 "Unrecognized response type %s for method ListDlpJobs, expected %s or %s", 616 response == null ? "null" : response.getClass().getName(), 617 ListDlpJobsResponse.class.getName(), 618 Exception.class.getName()))); 619 } 620 } 621 622 @Override getDlpJob(GetDlpJobRequest request, StreamObserver<DlpJob> responseObserver)623 public void getDlpJob(GetDlpJobRequest request, StreamObserver<DlpJob> responseObserver) { 624 Object response = responses.poll(); 625 if (response instanceof DlpJob) { 626 requests.add(request); 627 responseObserver.onNext(((DlpJob) response)); 628 responseObserver.onCompleted(); 629 } else if (response instanceof Exception) { 630 responseObserver.onError(((Exception) response)); 631 } else { 632 responseObserver.onError( 633 new IllegalArgumentException( 634 String.format( 635 "Unrecognized response type %s for method GetDlpJob, expected %s or %s", 636 response == null ? "null" : response.getClass().getName(), 637 DlpJob.class.getName(), 638 Exception.class.getName()))); 639 } 640 } 641 642 @Override deleteDlpJob(DeleteDlpJobRequest request, StreamObserver<Empty> responseObserver)643 public void deleteDlpJob(DeleteDlpJobRequest request, StreamObserver<Empty> responseObserver) { 644 Object response = responses.poll(); 645 if (response instanceof Empty) { 646 requests.add(request); 647 responseObserver.onNext(((Empty) response)); 648 responseObserver.onCompleted(); 649 } else if (response instanceof Exception) { 650 responseObserver.onError(((Exception) response)); 651 } else { 652 responseObserver.onError( 653 new IllegalArgumentException( 654 String.format( 655 "Unrecognized response type %s for method DeleteDlpJob, expected %s or %s", 656 response == null ? "null" : response.getClass().getName(), 657 Empty.class.getName(), 658 Exception.class.getName()))); 659 } 660 } 661 662 @Override cancelDlpJob(CancelDlpJobRequest request, StreamObserver<Empty> responseObserver)663 public void cancelDlpJob(CancelDlpJobRequest request, StreamObserver<Empty> responseObserver) { 664 Object response = responses.poll(); 665 if (response instanceof Empty) { 666 requests.add(request); 667 responseObserver.onNext(((Empty) response)); 668 responseObserver.onCompleted(); 669 } else if (response instanceof Exception) { 670 responseObserver.onError(((Exception) response)); 671 } else { 672 responseObserver.onError( 673 new IllegalArgumentException( 674 String.format( 675 "Unrecognized response type %s for method CancelDlpJob, expected %s or %s", 676 response == null ? "null" : response.getClass().getName(), 677 Empty.class.getName(), 678 Exception.class.getName()))); 679 } 680 } 681 682 @Override createStoredInfoType( CreateStoredInfoTypeRequest request, StreamObserver<StoredInfoType> responseObserver)683 public void createStoredInfoType( 684 CreateStoredInfoTypeRequest request, StreamObserver<StoredInfoType> responseObserver) { 685 Object response = responses.poll(); 686 if (response instanceof StoredInfoType) { 687 requests.add(request); 688 responseObserver.onNext(((StoredInfoType) response)); 689 responseObserver.onCompleted(); 690 } else if (response instanceof Exception) { 691 responseObserver.onError(((Exception) response)); 692 } else { 693 responseObserver.onError( 694 new IllegalArgumentException( 695 String.format( 696 "Unrecognized response type %s for method CreateStoredInfoType, expected %s or %s", 697 response == null ? "null" : response.getClass().getName(), 698 StoredInfoType.class.getName(), 699 Exception.class.getName()))); 700 } 701 } 702 703 @Override updateStoredInfoType( UpdateStoredInfoTypeRequest request, StreamObserver<StoredInfoType> responseObserver)704 public void updateStoredInfoType( 705 UpdateStoredInfoTypeRequest request, StreamObserver<StoredInfoType> responseObserver) { 706 Object response = responses.poll(); 707 if (response instanceof StoredInfoType) { 708 requests.add(request); 709 responseObserver.onNext(((StoredInfoType) response)); 710 responseObserver.onCompleted(); 711 } else if (response instanceof Exception) { 712 responseObserver.onError(((Exception) response)); 713 } else { 714 responseObserver.onError( 715 new IllegalArgumentException( 716 String.format( 717 "Unrecognized response type %s for method UpdateStoredInfoType, expected %s or %s", 718 response == null ? "null" : response.getClass().getName(), 719 StoredInfoType.class.getName(), 720 Exception.class.getName()))); 721 } 722 } 723 724 @Override getStoredInfoType( GetStoredInfoTypeRequest request, StreamObserver<StoredInfoType> responseObserver)725 public void getStoredInfoType( 726 GetStoredInfoTypeRequest request, StreamObserver<StoredInfoType> responseObserver) { 727 Object response = responses.poll(); 728 if (response instanceof StoredInfoType) { 729 requests.add(request); 730 responseObserver.onNext(((StoredInfoType) response)); 731 responseObserver.onCompleted(); 732 } else if (response instanceof Exception) { 733 responseObserver.onError(((Exception) response)); 734 } else { 735 responseObserver.onError( 736 new IllegalArgumentException( 737 String.format( 738 "Unrecognized response type %s for method GetStoredInfoType, expected %s or %s", 739 response == null ? "null" : response.getClass().getName(), 740 StoredInfoType.class.getName(), 741 Exception.class.getName()))); 742 } 743 } 744 745 @Override listStoredInfoTypes( ListStoredInfoTypesRequest request, StreamObserver<ListStoredInfoTypesResponse> responseObserver)746 public void listStoredInfoTypes( 747 ListStoredInfoTypesRequest request, 748 StreamObserver<ListStoredInfoTypesResponse> responseObserver) { 749 Object response = responses.poll(); 750 if (response instanceof ListStoredInfoTypesResponse) { 751 requests.add(request); 752 responseObserver.onNext(((ListStoredInfoTypesResponse) response)); 753 responseObserver.onCompleted(); 754 } else if (response instanceof Exception) { 755 responseObserver.onError(((Exception) response)); 756 } else { 757 responseObserver.onError( 758 new IllegalArgumentException( 759 String.format( 760 "Unrecognized response type %s for method ListStoredInfoTypes, expected %s or %s", 761 response == null ? "null" : response.getClass().getName(), 762 ListStoredInfoTypesResponse.class.getName(), 763 Exception.class.getName()))); 764 } 765 } 766 767 @Override deleteStoredInfoType( DeleteStoredInfoTypeRequest request, StreamObserver<Empty> responseObserver)768 public void deleteStoredInfoType( 769 DeleteStoredInfoTypeRequest request, StreamObserver<Empty> responseObserver) { 770 Object response = responses.poll(); 771 if (response instanceof Empty) { 772 requests.add(request); 773 responseObserver.onNext(((Empty) response)); 774 responseObserver.onCompleted(); 775 } else if (response instanceof Exception) { 776 responseObserver.onError(((Exception) response)); 777 } else { 778 responseObserver.onError( 779 new IllegalArgumentException( 780 String.format( 781 "Unrecognized response type %s for method DeleteStoredInfoType, expected %s or %s", 782 response == null ? "null" : response.getClass().getName(), 783 Empty.class.getName(), 784 Exception.class.getName()))); 785 } 786 } 787 788 @Override hybridInspectDlpJob( HybridInspectDlpJobRequest request, StreamObserver<HybridInspectResponse> responseObserver)789 public void hybridInspectDlpJob( 790 HybridInspectDlpJobRequest request, StreamObserver<HybridInspectResponse> responseObserver) { 791 Object response = responses.poll(); 792 if (response instanceof HybridInspectResponse) { 793 requests.add(request); 794 responseObserver.onNext(((HybridInspectResponse) response)); 795 responseObserver.onCompleted(); 796 } else if (response instanceof Exception) { 797 responseObserver.onError(((Exception) response)); 798 } else { 799 responseObserver.onError( 800 new IllegalArgumentException( 801 String.format( 802 "Unrecognized response type %s for method HybridInspectDlpJob, expected %s or %s", 803 response == null ? "null" : response.getClass().getName(), 804 HybridInspectResponse.class.getName(), 805 Exception.class.getName()))); 806 } 807 } 808 809 @Override finishDlpJob(FinishDlpJobRequest request, StreamObserver<Empty> responseObserver)810 public void finishDlpJob(FinishDlpJobRequest request, StreamObserver<Empty> responseObserver) { 811 Object response = responses.poll(); 812 if (response instanceof Empty) { 813 requests.add(request); 814 responseObserver.onNext(((Empty) response)); 815 responseObserver.onCompleted(); 816 } else if (response instanceof Exception) { 817 responseObserver.onError(((Exception) response)); 818 } else { 819 responseObserver.onError( 820 new IllegalArgumentException( 821 String.format( 822 "Unrecognized response type %s for method FinishDlpJob, expected %s or %s", 823 response == null ? "null" : response.getClass().getName(), 824 Empty.class.getName(), 825 Exception.class.getName()))); 826 } 827 } 828 } 829