GlobalExceptionHandler.java 4.38 KB
package com.diligrp.tax.central.config;

import com.diligrp.tax.central.exception.TaxAgentServiceException;
import com.diligrp.tax.central.message.Message;
import com.diligrp.tax.central.type.TaxSystemType;
import jakarta.validation.ConstraintViolation;
import jakarta.validation.ConstraintViolationException;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.support.DefaultMessageSourceResolvable;
import org.springframework.http.converter.HttpMessageNotReadableException;
import org.springframework.validation.BindException;
import org.springframework.validation.BindingResult;
import org.springframework.validation.ObjectError;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.MissingServletRequestParameterException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import tools.jackson.core.JacksonException;

import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * 统一异常处理器
 */
@RestControllerAdvice
@Slf4j
public class GlobalExceptionHandler {
    /**
     * 处理业务异常
     */
    @ExceptionHandler(TaxAgentServiceException.class)
    public Message<?> handleBusinessException(TaxAgentServiceException e) {
        log.error("Method throw TaxAgentServiceException", e);
        return Message.failure(e.getCode(), e.getMessage());
    }

    /**
     * 统一拦截参数转换异常 JSON -> Object
     *
     * @param ex InvalidFormatException
     * @return 响应消息对象
     */
    @ExceptionHandler({HttpMessageNotReadableException.class, JacksonException.class})
    public Message<?> invalidFormatExceptionHandler(Exception ex) {
        return Message.failure(TaxSystemType.ABNORMAL_PARAMETERS.code, ex.getMessage());
    }

    /**
     * 参数验证
     *
     * @param exs 参数异常
     * @return BaseOutput
     */
    @ExceptionHandler(value = ConstraintViolationException.class)
    public Message<?> constraintViolationException(ConstraintViolationException exs) {
        Set<ConstraintViolation<?>> violations = exs.getConstraintViolations();
        String msg = violations.stream().map(ConstraintViolation::getMessage)
                .collect(Collectors.joining(","));
        return Message.failure(TaxSystemType.ABNORMAL_PARAMETERS.code, msg);
    }

    /**
     * 参数验证
     *
     * @param e MethodArgumentNotValidException @RequestBody
     *          BindException 普通form表单
     * @return BaseOutput
     */
    @ExceptionHandler(value = {MethodArgumentNotValidException.class, BindException.class})
    public Message<?> methodArgumentNotValidException(Exception e) {
        BindingResult bindingResult;
        if (e instanceof MethodArgumentNotValidException) {
            bindingResult = ((MethodArgumentNotValidException) e).getBindingResult();
        } else {
            bindingResult = ((BindException) e).getBindingResult();
        }
        List<ObjectError> allErrors = bindingResult.getAllErrors();
        String msg = allErrors.stream().map(DefaultMessageSourceResolvable::getDefaultMessage)
                .collect(Collectors.joining("、"));

        return Message.failure(TaxSystemType.ABNORMAL_PARAMETERS.code, msg);
    }

    /**
     * 缺少请求参数异常
     *
     * @param ex 请求参数异常
     * @return 异常信息
     */
    @ExceptionHandler(MissingServletRequestParameterException.class)
    public Message<?> missingServletRequestParameterExceptionHandler(MissingServletRequestParameterException ex) {
        return Message.failure(TaxSystemType.ABNORMAL_PARAMETERS.code, String.format("缺少参数:%s", ex.getParameterName()));
    }

    /**
     * 断言验证
     *
     * @param ex IllegalArgumentException
     * @return BaseOutput
     */
    @ExceptionHandler(value = IllegalArgumentException.class)
    public Message<?> illegalArgumentException(IllegalArgumentException ex) {
        return Message.failure(TaxSystemType.ABNORMAL_PARAMETERS.code, ex.getMessage());
    }

    /**
     * 统一拦截系统其它异常
     *
     * @param ex 系统异常
     * @return 响应消息对象
     */
    @ExceptionHandler(Exception.class)
    public Message<?> exceptionHandler(Exception ex) {
        log.error("unknown error :", ex);
        return Message.failure(TaxSystemType.UNKNOWN_ERROR.code, TaxSystemType.UNKNOWN_ERROR.message);
    }
}