spring boot项目使用@ControllerAdvice做全局异常处理

最新推荐文章于 2024-08-22 11:27:51 发布

bird_tp

最新推荐文章于 2024-08-22 11:27:51 发布

阅读量436

点赞数

分类专栏：工作总结文章标签：全局异常处理

本文链接：https://blog.csdn.net/bird_tp/article/details/117263981

版权

工作总结专栏收录该内容

43 篇文章 1 订阅

订阅专栏

一、博客背景

在项目中有遇到一个需求，对于项目中的特殊异常信息信息需要做一个处理，统一返回到前端。

二、代码

统一异常处理类

import com.tp.common.constant.CommonConstant;
import com.tp.common.exception.BusinessException;
import com.tp.common.exception.TokenException;
import com.tp.common.model.Resp;
import com.tp.common.utils.IpUtils;
import com.tp.common.utils.RespUtil;
import com.tp.service.threadlocal.UserContext;
import com.tp.web.config.EsLogConfig;
import lombok.extern.slf4j.Slf4j;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.validation.BindException;
import org.springframework.validation.FieldError;
import org.springframework.validation.ObjectError;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletRequest;
import java.util.List;


@ControllerAdvice
@Slf4j
public class ExceptionAdvice {

    private static final Logger logger = LoggerFactory.getLogger(ExceptionAdvice.class);

    @Resource
    HttpServletRequest httpServletRequest;

    @Resource
    EsLogConfig esLogConfig;



    @ExceptionHandler(Exception.class)
    @ResponseBody
    public Resp exception(Exception e) {
        logger.error("请求出现未知异常", e);
        saveLogToEsQueue(e, httpServletRequest, "未知异常，请联系客服！", CommonConstant.ErrorEnum.EXCEPTION_CODE.getCode(), getPhone());
        return RespUtil.fail("未知异常，请联系客服！", CommonConstant.ErrorEnum.EXCEPTION_CODE.getCode());

    }

    @ExceptionHandler(BusinessException.class)
    @ResponseBody
    public Resp businessException(BusinessException e) {
        log.error(e.getMessage(), e);
        saveLogToEsQueue(e, httpServletRequest, e.getMessage(), e.getCode(), getPhone());
        return RespUtil.fail(e.getMessage(), e.getCode());
    }

    @ExceptionHandler(IllegalArgumentException.class)
    @ResponseBody
    public Resp IllegalArgumentException(IllegalArgumentException e) {
        log.error(e.getMessage(), e);
        saveLogToEsQueue(e, httpServletRequest, e.getMessage(), CommonConstant.ErrorEnum.ILLEGALARGUMENT_EXCEPTION_CODE.getCode(), getPhone());
        return RespUtil.fail(e.getMessage(), CommonConstant.ErrorEnum.ILLEGALARGUMENT_EXCEPTION_CODE.getCode());
    }

    @ExceptionHandler({TokenException.class})
    @ResponseBody
    public Resp tokenException(TokenException e) {
        saveLogToEsQueue(e, httpServletRequest, e.getMessage(), e.getCode(), "no login");
        return RespUtil.fail(e.getData(), e.getMessage(), e.getCode());
    }

    @ExceptionHandler({BindException.class})
    @ResponseBody
    public Resp bindException(BindException e) {
        List<ObjectError> objectErrorList = e.getBindingResult().getAllErrors();
        StringBuilder message = new StringBuilder();
        for (ObjectError objectError : objectErrorList) {
            FieldError fieldError = (FieldError) objectError;
            message.append(fieldError.getField() + " " + fieldError.getDefaultMessage() + "/n");
        }
        saveLogToEsQueue(e, httpServletRequest, message.toString(), CommonConstant.ErrorEnum.ILLEGALARGUMENT_EXCEPTION_CODE.getCode(), getPhone());
        return RespUtil.fail(message.toString(), CommonConstant.ErrorEnum.ILLEGALARGUMENT_EXCEPTION_CODE.getCode());
    }

    /**
     * 将系统中抛出来的错误日志信息保存到内存队列中
     *
     * @param e                  异常对象
     * @param httpServletRequest 请求对象，获取请求的方法和ip地址
     * @param message            自定义的错误信息
     * @param code               自定义的错误编码
     * @param phone              手机号，用来标识用户身份信息
     */
    private void saveLogToEsQueue(Exception e, HttpServletRequest httpServletRequest, String message, int code, String phone) {
        String requestUri = httpServletRequest.getRequestURL().toString();
        String userId = IpUtils.getIpAddr(httpServletRequest) + " : " + phone;
        StringBuilder url = new StringBuilder(esLogConfig.getUrl());
        url.append(esLogConfig.getDomain())
                .append("&expCode=").append(code)
                .append("&expMessage=").append(message)
                .append("&expStacktrace=").append(e.getStackTrace()[0])
                .append("&platform=KSXD")
                .append("&requestIp=").append(IpUtils.getIpAddr(httpServletRequest))
                .append("&requestUri=").append(requestUri)
                .append("&userId=").append(userId);
        if (!EsLogQueue.INS.push(url.toString())) {
            log.error("将异常信息添加到内存队列失败，队列已满！");
        }
    }

    /**
     * 获取手机号
     *
     * @return 手机号
     */
    private String getPhone() {
        String phone;
        try {
            phone = UserContext.getUser().getPhone();
        } catch (Exception exception) {
            phone = "no login";
        }
        return phone;
    }
}