org.openbaton.marketplace.api.exceptions.GlobalExceptionHandler.java Source code

Introduction

Here is the source code for org.openbaton.marketplace.api.exceptions.GlobalExceptionHandler.java

Source

/*
 * #
 * # Copyright (c) 2015 Fraunhofer FOKUS
 * #
 * # Licensed under the Apache License, Version 2.0 (the "License");
 * # you may not use this file except in compliance with the License.
 * # You may obtain a copy of the License at
 * #
 * #     http://www.apache.org/licenses/LICENSE-2.0
 * #
 * # Unless required by applicable law or agreed to in writing, software
 * # distributed under the License is distributed on an "AS IS" BASIS,
 * # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * # See the License for the specific language governing permissions and
 * # limitations under the License.
 * #
 *
 */

/*
 */

package org.openbaton.marketplace.api.exceptions;

import org.openbaton.exceptions.NotFoundException;
import org.openbaton.marketplace.exceptions.FailedToUploadException;
import org.openbaton.marketplace.exceptions.ImageRepositoryNotEnabled;
import org.openbaton.marketplace.exceptions.PackageIntegrityException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.http.HttpHeaders;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.ResponseStatus;
import org.springframework.web.context.request.WebRequest;

import java.io.IOException;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;

/**
 * Created by gca on 27/08/15.
 */
@ControllerAdvice
public class GlobalExceptionHandler {

    private Logger log = LoggerFactory.getLogger(this.getClass());

    @ExceptionHandler({ NotFoundException.class, ImageRepositoryNotEnabled.class })
    @ResponseStatus(value = HttpStatus.NOT_FOUND)
    protected ResponseEntity<Object> handleNotFoundException(HttpServletRequest req, Exception e) {
        log.error("Exception with message " + e.getMessage() + " was thrown");
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_JSON);
        Map body = new HashMap<>();
        body.put("error", "Not Found");
        body.put("exception", e.getClass().toString());
        body.put("message", e.getMessage());
        body.put("path", req.getRequestURI());
        body.put("status", HttpStatus.NOT_FOUND.value());
        body.put("timestamp", new Date().getTime());
        ResponseEntity responseEntity = new ResponseEntity(body, headers, HttpStatus.NOT_FOUND);
        return responseEntity;
    }

    @ExceptionHandler({ InterruptedException.class, IOException.class })
    @ResponseStatus(value = HttpStatus.NOT_FOUND)
    protected ResponseEntity<Object> handleException(HttpServletRequest req, Exception e) {
        log.error("Exception with message " + e.getMessage() + " was thrown");
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_JSON);
        Map body = new HashMap<>();
        body.put("error", "Upload Error");
        body.put("exception", e.getClass().toString());
        body.put("message", e.getMessage());
        body.put("path", req.getRequestURI());
        body.put("status", HttpStatus.I_AM_A_TEAPOT.value());
        body.put("timestamp", new Date().getTime());
        ResponseEntity responseEntity = new ResponseEntity(body, headers, HttpStatus.I_AM_A_TEAPOT);
        return responseEntity;
    }

    @ExceptionHandler({ PackageIntegrityException.class })
    @ResponseStatus(value = HttpStatus.BAD_REQUEST)
    protected @ResponseBody ResponseEntity<Object> handleInvalidRequest(Exception e, WebRequest request) {
        if (log.isDebugEnabled()) {
            log.error("Exception was thrown -> Return message: " + e.getMessage(), e);
        } else {
            log.error("Exception was thrown -> Return message: " + e.getMessage());
        }
        ExceptionResource exc = new ExceptionResource("Bad Request", e.getMessage());
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_JSON);
        Map body = new HashMap<>();
        body.put("error", "Upload Error");
        body.put("exception", e.getClass().toString());
        body.put("message", e.getMessage());
        body.put("path", request.getContextPath());
        body.put("status", HttpStatus.BAD_REQUEST.value());
        body.put("timestamp", new Date().getTime());
        ResponseEntity responseEntity = new ResponseEntity(body, headers, HttpStatus.BAD_REQUEST);
        return responseEntity;
    }

    @ExceptionHandler({ FailedToUploadException.class })
    @ResponseStatus(value = HttpStatus.I_AM_A_TEAPOT)
    protected @ResponseBody ResponseEntity<Object> handleErrorRequest(Exception e, WebRequest request) {
        if (log.isDebugEnabled()) {
            log.error("Exception was thrown -> Return message: " + e.getMessage(), e);
        } else {
            log.error("Exception was thrown -> Return message: " + e.getMessage());
        }
        ExceptionResource exc = new ExceptionResource("Bad Request", e.getMessage());
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_JSON);
        Map body = new HashMap<>();
        body.put("error", "Upload Error");
        body.put("exception", e.getClass().toString());
        body.put("message", e.getMessage());
        body.put("path", request.getContextPath());
        body.put("status", HttpStatus.I_AM_A_TEAPOT.value());
        body.put("timestamp", new Date().getTime());
        ResponseEntity responseEntity = new ResponseEntity(body, headers, HttpStatus.I_AM_A_TEAPOT);
        return responseEntity;
    }
}