gerrit.onap Code Review - policy/clamp.git/blob

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  *  Copyright (C) 2021 Nordix Foundation.
   4  * ================================================================================
   5  * Licensed under the Apache License, Version 2.0 (the "License");
   6  * you may not use this file except in compliance with the License.
   7  * You may obtain a copy of the License at
   8  *
   9  *      http://www.apache.org/licenses/LICENSE-2.0
  10  *
  11  * Unless required by applicable law or agreed to in writing, software
  12  * distributed under the License is distributed on an "AS IS" BASIS,
  13  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14  * See the License for the specific language governing permissions and
  15  * limitations under the License.
  16  *
  17  * SPDX-License-Identifier: Apache-2.0
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.onap.policy.clamp.controlloop.runtime.main.web;
  22
  23 import org.onap.policy.clamp.controlloop.common.exception.ControlLoopException;
  24 import org.onap.policy.clamp.controlloop.models.messages.rest.SimpleResponse;
  25 import org.onap.policy.models.base.PfModelException;
  26 import org.onap.policy.models.base.PfModelRuntimeException;
  27 import org.springframework.http.ResponseEntity;
  28 import org.springframework.web.bind.annotation.ExceptionHandler;
  29 import org.springframework.web.bind.annotation.RestControllerAdvice;
  30
  31 @RestControllerAdvice
  32 public class GlobalControllerExceptionHandler {
  33
  34     /**
  35      * Handle ControlLoopException.
  36      *
  37      * @param ex ControlLoopException
  38      * @return ResponseEntity
  39      */
  40     @ExceptionHandler(ControlLoopException.class)
  41     public ResponseEntity<SimpleResponse> handleBadRequest(ControlLoopException ex) {
  42         var resp = new SimpleResponse();
  43         resp.setErrorDetails(ex.getErrorResponse().getErrorMessage());
  44         return ResponseEntity.status(ex.getErrorResponse().getResponseCode().getStatusCode()).body(resp);
  45     }
  46
  47     /**
  48      * Handle PfModelRuntimeException.
  49      *
  50      * @param ex PfModelRuntimeException
  51      * @return ResponseEntity
  52      */
  53     @ExceptionHandler(PfModelRuntimeException.class)
  54     public ResponseEntity<SimpleResponse> handleBadRequest(PfModelRuntimeException ex) {
  55         var resp = new SimpleResponse();
  56         resp.setErrorDetails(ex.getErrorResponse().getErrorMessage());
  57         return ResponseEntity.status(ex.getErrorResponse().getResponseCode().getStatusCode()).body(resp);
  58     }
  59
  60     /**
  61      * Handle PfModelException.
  62      *
  63      * @param ex PfModelException
  64      * @return ResponseEntity
  65      */
  66     @ExceptionHandler(PfModelException.class)
  67     public ResponseEntity<SimpleResponse> handleBadRequest(PfModelException ex) {
  68         var resp = new SimpleResponse();
  69         resp.setErrorDetails(ex.getErrorResponse().getErrorMessage());
  70         return ResponseEntity.status(ex.getErrorResponse().getResponseCode().getStatusCode()).body(resp);
  71     }
  72
  73 }