GlobalExceptionHandler.java 2.86 KB
package com.bsth.exception;

import com.bsth.common.ResponseCode;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.security.access.AccessDeniedException;
import org.springframework.security.authentication.AccountExpiredException;
import org.springframework.security.core.userdetails.UsernameNotFoundException;
import org.springframework.validation.BindException;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.servlet.NoHandlerFoundException;

import java.util.HashMap;
import java.util.Map;

/**
 * 全局异常处理器
 *
 */
@RestControllerAdvice
public class GlobalExceptionHandler
{
    private static final Logger log = LoggerFactory.getLogger(GlobalExceptionHandler.class);

    @ExceptionHandler(NoHandlerFoundException.class)
    public Map<String, Object> handlerNoFoundException(Exception e)
    {
        log.error(e.getMessage(), e);
        return error("路径不存在,请检查路径是否正确");
    }

    @ExceptionHandler(AccessDeniedException.class)
    public Map<String, Object> handleAuthorizationException(AccessDeniedException e)
    {
        log.error(e.getMessage());
        return error("没有权限,请联系管理员授权");
    }

    @ExceptionHandler(AccountExpiredException.class)
    public Map<String, Object> handleAccountExpiredException(AccountExpiredException e)
    {
        log.error(e.getMessage(), e);
        return error(e.getMessage());
    }

    @ExceptionHandler(UsernameNotFoundException.class)
    public Map<String, Object> handleUsernameNotFoundException(UsernameNotFoundException e)
    {
        log.error(e.getMessage(), e);
        return error(e.getMessage());
    }

    @ExceptionHandler(Exception.class)
    public Map<String, Object> handleException(Exception e)
    {
        log.error(e.getMessage(), e);
        return error(e.getMessage());
    }

    /**
     * 自定义验证异常
     */
    @ExceptionHandler(BindException.class)
    public Map<String, Object> validatedBindException(BindException e)
    {
        log.error(e.getMessage(), e);
        String message = e.getAllErrors().get(0).getDefaultMessage();
        return error(message);
    }

    /**
     * 自定义验证异常
     */
    @ExceptionHandler(MethodArgumentNotValidException.class)
    public Object validExceptionHandler(MethodArgumentNotValidException e)
    {
        log.error(e.getMessage(), e);
        String message = e.getBindingResult().getFieldError().getDefaultMessage();
        return error(message);
    }

    private Map<String, Object> error(String message) {
        Map<String, Object> result = new HashMap<>();
        result.put("status", ResponseCode.ERROR);
        result.put("msg", "");

        return result;
    }
}