欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Spring 自定义返回异常

程序员文章站 2022-07-12 16:07:53
...

结果封装类


import org.apache.http.HttpStatus;

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

public class ResultCode extends HashMap<String, Object> {

    private static final long serialVersionUID = 1L;

    public ResultCode() {
        put("code", 0);
        put("msg", "success");
    }

    public static ResultCode error() {
        return error(HttpStatus.SC_INTERNAL_SERVER_ERROR, "未知异常");
    }

    public static ResultCode error(String msg) {
        return error(HttpStatus.SC_INTERNAL_SERVER_ERROR, msg);
    }

    public static ResultCode error(int code, String msg) {
        ResultCode r = new ResultCode();
        r.put("code", code);
        r.put("msg", msg);
        return r;
    }

    public static ResultCode ok(String msg) {
        ResultCode r = new ResultCode();
        r.put("msg", msg);
        return r;
    }

    public static ResultCode ok(Map<String, Object> map) {
        ResultCode r = new ResultCode();
        r.putAll(map);
        return r;
    }

    public static ResultCode ok() {
        return new ResultCode();
    }

    public ResultCode put(String key, Object value) {
        super.put(key, value);
        return this;
    }
}

public class RException extends RuntimeException {
    private static final long serialVersionUID = 1L;

    private String msg;
    private int code = 500;

    public RException(String msg) {
        super(msg);
        this.msg = msg;
    }

    public RException(String msg, Throwable e) {
        super(msg, e);
        this.msg = msg;
    }

    public RException(String msg, int code) {
        super(msg);
        this.msg = msg;
        this.code = code;
    }

    public RException(String msg, int code, Throwable e) {
        super(msg, e);
        this.msg = msg;
        this.code = code;
    }

    public String getMsg() {
        return msg;
    }

    public void setMsg(String msg) {
        this.msg = msg;
    }

    public int getCode() {
        return code;
    }

    public void setCode(int code) {
        this.code = code;
    }
    
}

import org.apache.shiro.authz.AuthorizationException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.dao.DuplicateKeyException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;
import org.springframework.web.servlet.NoHandlerFoundException;

@RestControllerAdvice
public class RExceptionHandler {
    private Logger logger = LoggerFactory.getLogger(getClass());

    /**
     * 处理自定义异常
     */
    @ExceptionHandler(RException.class)
    public ResultCode handleRRException(RException e){
        ResultCode r = new ResultCode();
        r.put("code", e.getCode());
        r.put("msg", e.getMessage());

        return r;
    }

    @ExceptionHandler(NoHandlerFoundException.class)
    public ResultCode handlerNoFoundException(Exception e) {
        logger.error(e.getMessage(), e);
        return ResultCode.error(404, "路径不存在,请检查路径是否正确");
    }

    @ExceptionHandler(DuplicateKeyException.class)
    public ResultCode handleDuplicateKeyException(DuplicateKeyException e){
        logger.error(e.getMessage(), e);
        return ResultCode.error("数据库中已存在该记录");
    }

    @ExceptionHandler(AuthorizationException.class)
    public ResultCode handleAuthorizationException(AuthorizationException e){
        logger.error(e.getMessage(), e);
        return ResultCode.error("没有权限,请联系管理员授权");
    }

    @ExceptionHandler(Exception.class)
    public ResultCode handleException(Exception e){
        logger.error(e.getMessage(), e);
        return ResultCode.error();
    }
}

相关标签: 笔记 spring