cloud.agileframework.mvc.exception.SpringExceptionHandler Maven / Gradle / Ivy

Go to download
package cloud.agileframework.mvc.exception;

import cloud.agileframework.common.constant.Constant;
import cloud.agileframework.common.util.pattern.PatternUtil;
import cloud.agileframework.mvc.base.AbstractResponseFormat;
import cloud.agileframework.mvc.base.RETURN;
import cloud.agileframework.mvc.util.ViewUtil;
import cloud.agileframework.spring.util.BeanUtil;
import cloud.agileframework.spring.util.MessageUtil;
import cloud.agileframework.spring.util.ServletUtil;
import org.apache.commons.lang3.StringUtils;
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.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.servlet.HandlerExceptionResolver;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * @author 佟盟 on 2018/6/25
 */
@Order(Ordered.HIGHEST_PRECEDENCE)
@ControllerAdvice
public class SpringExceptionHandler implements HandlerExceptionResolver {
    private static final Logger logger = LoggerFactory.getLogger(SpringExceptionHandler.class);
    private static final String MESSAGE_HEAD = "异常捕捉";

    @ExceptionHandler(Throwable.class)
    public ModelAndView allExceptionHandler(HttpServletRequest request, HttpServletResponse response, Throwable e) {
        return createModelAndView(request, response, e);
    }

    public static ModelAndView createModelAndView(HttpServletRequest request, HttpServletResponse response, Throwable e) {
        //在请求中记录
        request.setAttribute(Constant.RequestAttributeAbout.ERROR_EXCEPTION, e);

        ModelAndView modelAndView;

        if (e instanceof AgileArgumentException) {
            Object attributeErrors = ServletUtil.getCurrentRequest().getAttribute(Constant.RequestAttributeAbout.ATTRIBUTE_ERROR);
            if (attributeErrors != null) {
                return ViewUtil.getResponseFormatData(RETURN.PARAMETER_ERROR, attributeErrors);
            }
        }

        RETURN r = to(message(e));
        if (r.getStatus() == HttpStatus.INTERNAL_SERVER_ERROR) {
            logger.error(MESSAGE_HEAD, e);
        } else {
            logger.warn(MESSAGE_HEAD, e);
        }

        AbstractResponseFormat abstractResponseFormat = BeanUtil.getBean(AbstractResponseFormat.class);
        if (abstractResponseFormat != null) {
            modelAndView = abstractResponseFormat.buildResponse(r, null);
        } else {
            modelAndView = new ModelAndView();
            modelAndView.addObject(Constant.ResponseAbout.HEAD, r);
        }
        modelAndView.setStatus(r.getStatus());

        return modelAndView;
    }

    /**
     * 提取国际化响应文
     *
     * @param e 异常
     * @return RETURN
     */
    private static String message(Throwable e) {
        Throwable cause = e.getCause();
        if (cause != null && cause.getClass() != e.getClass()) {
            String causeMessage = message(cause);
            if (!StringUtils.isEmpty(causeMessage)) {
                return causeMessage;
            }
        }
        String message;
        if (e instanceof AbstractCustomException) {
            message = MessageUtil.message(e.getClass().getName(), null, ((AbstractCustomException) e).getParams());
        } else {
            message = MessageUtil.message(e.getClass().getName(), null, e.getMessage());
        }
        if (message == null) {
            message = e.getMessage();
        }
        return message;
    }

    /**
     * 提取国际化响应文
     *
     * @param message 异常信息
     * @return RETURN
     */
    private static RETURN to(String message) {

        if (StringUtils.isEmpty(message)) {
            return RETURN.of(RETURN.FAIL.getCode(), RETURN.FAIL.getMsg(), HttpStatus.INTERNAL_SERVER_ERROR);
        } else {
            final String codePrefix = "^[\\d]{6}";
            if (PatternUtil.find(codePrefix, message) && !message.startsWith(Constant.NumberAbout.TWO + "")) {
                return RETURN.byMessage(HttpStatus.OK, message);
            }
            return RETURN.byMessage(HttpStatus.INTERNAL_SERVER_ERROR, message);
        }
    }

    @Override
    public ModelAndView resolveException(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {
        return createModelAndView(request, response, ex);
    }
}