scouterx.webapp.framework.exception.mapper.GenericExceptionMapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of scouter-webapp Show documentation
There is a newer version: 2.20.0
/*
 *  Copyright 2015 the original author or authors.
 *  @https://github.com/scouter-project/scouter
 *
 *  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 scouterx.webapp.framework.exception.mapper;

import lombok.extern.slf4j.Slf4j;
import scouterx.webapp.framework.exception.ErrorState;
import scouterx.webapp.framework.exception.ErrorStateBizException;
import scouterx.webapp.framework.exception.ErrorStateException;
import scouterx.webapp.view.CommonResultView;

import javax.ws.rs.WebApplicationException;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.UriInfo;
import javax.ws.rs.ext.ExceptionMapper;
import javax.ws.rs.ext.Provider;
import java.util.Arrays;
import java.util.stream.Collectors;

/**
 * Created by Gun Lee([email protected]) on 2017. 8. 25.
 */
@Provider
@Slf4j
public class GenericExceptionMapper implements ExceptionMapper {
	@Context
	UriInfo uriInfo;

	@Override
	public Response toResponse(Throwable throwable) {
		if (throwable instanceof ErrorStateBizException) {
            return handleErrorStateBizException((ErrorStateBizException) throwable);

		} else if (throwable instanceof ErrorStateException) {
            return handlerErrorStateException((ErrorStateException) throwable);

		} else if(throwable instanceof WebApplicationException) {
            return handlerWebApplicationException((WebApplicationException) throwable);

		} else {
            return handleThrowable(throwable);
		}
	}

    private Response handleThrowable(Throwable throwable) {
	    String errorMessage = getMessageWithCause(throwable);
	    CommonResultView resultView = CommonResultView.fail(500, 500, errorMessage, null);

        log.error("[WebApplicationException] {} - {}, [uri]{}", resultView.getStatus()
                , resultView.getMessage(), uriInfo.getPath(), throwable);

        return Response.status(resultView.getStatus())
                .entity(resultView)
                .type(MediaType.APPLICATION_JSON).build();
    }

	private Response handlerWebApplicationException(WebApplicationException throwable) {
        WebApplicationException ex = throwable;
        CommonResultView resultView = CommonResultView.fail(ex.getResponse().getStatus()
                , ex.getResponse().getStatus(), ex.getMessage(), null);


        log.error("[WebApplicationException] {} - {}, [uri]{}", resultView.getStatus()
                , resultView.getMessage(), uriInfo.getPath());

        return Response.status(resultView.getStatus())
                .entity(resultView)
                .type(MediaType.APPLICATION_JSON).build();
    }

    private Response handlerErrorStateException(ErrorStateException throwable) {
        ErrorStateException ex = throwable;
        ErrorState errorState = ex.getErrorState();
        CommonResultView resultView = CommonResultView.fail(errorState.getStatus().getStatusCode()
                , errorState.getErrorCode(), errorState.getErrorMessage(), null);

        StackTraceElement lastStack = Arrays.stream(ex.getStackTrace()).findFirst().orElse(null);
        log.error("[ErrorStateException] {} - {} - {}, [uri]{} at {}", resultView.getStatus(),
                resultView.getMessage(), ex.getMessage(), uriInfo.getPath(), lastStack, ex);

        return Response.status(resultView.getStatus())
                .entity(resultView)
                .type(MediaType.APPLICATION_JSON).build();
    }

    private Response handleErrorStateBizException(ErrorStateBizException throwable) {
        ErrorStateBizException ex = throwable;
        ErrorState errorState = ex.getErrorState();
        String errorMessage = (errorState == ErrorState.VALIDATE_ERROR) ?
                errorState.getErrorMessage() + ex.getMessage() : errorState.getErrorMessage();

        CommonResultView resultView = CommonResultView.fail(errorState.getStatus().getStatusCode()
                , errorState.getErrorCode(), errorMessage, null);

        String lastStack = Arrays.stream(ex.getStackTrace()).limit(3).map(StackTraceElement::toString).collect(Collectors.joining("\\n "));

        log.error("[ErrorStateBizException] {} - {} - {}, [uri]{} at {}", resultView.getStatus()
                , resultView.getMessage(), ex.getMessage(), uriInfo.getPath(), lastStack);

        return Response.status(resultView.getStatus())
                .entity(resultView)
                .type(MediaType.APPLICATION_JSON).build();
    }

	private String getMessageWithCause(Throwable throwable) {
		StringBuilder sb = new StringBuilder().append(throwable.getMessage());
		if (throwable.getCause() != null) {
			sb.append(" [caused by] ").append(throwable.getCause().getMessage());
		}
		return sb.toString();
	}
}