org.trustedanalytics.user.common.WebErrorHandlers.java Source code

Introduction

Here is the source code for org.trustedanalytics.user.common.WebErrorHandlers.java

Source

/**
 *  Copyright(c)2015 IntelCorporation
 *
 *  LicensedundertheApacheLicense,Version2.0(the"License");
 *  youmaynotusethisfileexceptincompliancewiththeLicense.
 *  YoumayobtainacopyoftheLicenseat
 *
 *  http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unlessrequiredbyapplicablelaworagreedtoinwriting,software
 *  distributedundertheLicenseisdistributedonan"ASIS"BASIS,
 *  WITHOUTWARRANTIESORCONDITIONSOFANYKIND,eitherexpressorimplied.
 *  SeetheLicenseforthespecificlanguagegoverningpermissionsand
 *  limitationsundertheLicense.
 */
package org.trustedanalytics.user.common;

import static org.springframework.http.HttpStatus.BAD_REQUEST;

import org.trustedanalytics.cloud.cc.api.customizations.FeignResponseException;
import org.trustedanalytics.utils.errorhandling.ErrorFormatter;
import org.trustedanalytics.utils.errorhandling.RestErrorHandler;

import org.apache.commons.lang.StringUtils;
import org.codehaus.jackson.map.ObjectMapper;
import org.codehaus.jackson.type.TypeReference;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.core.Ordered;
import org.springframework.core.annotation.Order;
import org.springframework.http.HttpStatus;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.client.HttpStatusCodeException;

import java.io.IOException;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import javax.servlet.http.HttpServletResponse;

@ControllerAdvice
@Order(Ordered.LOWEST_PRECEDENCE)
public class WebErrorHandlers {

    private static final Logger LOGGER = LoggerFactory.getLogger(WebErrorHandlers.class);

    @ExceptionHandler(HttpMessageNotReadableException.class)
    public void badRequest(Exception e, HttpServletResponse response) throws IOException {
        logAndSendErrorResponse(response, BAD_REQUEST, e);
    }

    @ExceptionHandler(HttpStatusCodeException.class)
    public void handleHttpStatusCodeException(HttpStatusCodeException e, HttpServletResponse response)
            throws IOException {
        String message = extractErrorFromJSON(e.getResponseBodyAsString());
        message = StringUtils.isNotBlank(message) ? message : e.getMessage();
        logAndSendErrorResponse(response, e.getStatusCode(), message, e);
    }

    @ExceptionHandler(FeignResponseException.class)
    public void handleFeignException(FeignResponseException e, HttpServletResponse response) throws IOException {
        logAndSendErrorResponse(response, e.getStatusCode(), e.getMessage(), e);
    }

    @ExceptionHandler(Exception.class)
    public void handleException(Exception e, HttpServletResponse response) throws Exception {
        RestErrorHandler defaultErrorHandler = new RestErrorHandler();
        defaultErrorHandler.handleException(e, response);
    }

    private static long generateErrorId() {
        return new Date().getTime();
    }

    private static void logAndSendErrorResponse(HttpServletResponse response, HttpStatus status, Exception ex)
            throws IOException {
        logAndSendErrorResponse(response, status, status.getReasonPhrase(), ex);
    }

    private static void logAndSendErrorResponse(HttpServletResponse response, HttpStatus status, String reason,
            Exception ex) throws IOException {
        long errorId = generateErrorId();
        String errorMessage = ErrorFormatter.formatErrorMessage(reason, errorId);

        LOGGER.error(errorMessage, ex);
        response.sendError(status.value(), errorMessage);
    }

    private static String extractErrorFromJSON(String json) {
        Map<String, String> map = new HashMap<>();
        try {
            ObjectMapper mapper = new ObjectMapper();
            TypeReference<HashMap<String, String>> typeReference = new TypeReference<HashMap<String, String>>() {
            };
            map = mapper.readValue(json, typeReference);
        } catch (Exception e) {
            String msg = "Exception raised while extracting error from JSON ";
            LOGGER.error(msg, e);
        }
        return map.get("description");
    }
}