001/*
002 * Licensed under the Apache License, Version 2.0 (the "License");
003 * you may not use this file except in compliance with the License.
004 * You may obtain a copy of the License at
005 *
006 *     http://www.apache.org/licenses/LICENSE-2.0
007 *
008 * Unless required by applicable law or agreed to in writing, software
009 * distributed under the License is distributed on an "AS IS" BASIS,
010 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
011 * See the License for the specific language governing permissions and
012 * limitations under the License.
013 */
014package org.gbif.ws.server.mapper;
015
016import org.gbif.ws.NotFoundException;
017import org.gbif.ws.WebApplicationException;
018
019import org.springframework.http.ResponseEntity;
020import org.springframework.web.bind.annotation.ControllerAdvice;
021import org.springframework.web.bind.annotation.ExceptionHandler;
022
023@ControllerAdvice
024public class WebApplicationExceptionMapper {
025
026  @SuppressWarnings("rawtypes")
027  @ExceptionHandler(WebApplicationException.class)
028  public ResponseEntity handleWebApplicationException(WebApplicationException e) {
029    return ResponseEntity.status(e.getStatus())
030        .contentType(e.getContentType())
031        .body(e.getMessage());
032  }
033
034  @SuppressWarnings("rawtypes")
035  @ExceptionHandler(NotFoundException.class)
036  public ResponseEntity handleNotException(NotFoundException e) {
037    return ResponseEntity.status(e.getStatus())
038        .contentType(e.getContentType())
039        .body(e.getMessage());
040  }
041}