优化数据权限代码

RuoYi

2024-06-05 0680d1ed1f79655d553618e0e1c007bcf07f4d7a

 ruoyi-common/ruoyi-common-security/src/main/java/com/ruoyi/common/security/handler/GlobalExceptionHandler.java

@@ -6,19 +6,24 @@
import org.springframework.validation.BindException;

import org.springframework.web.HttpRequestMethodNotSupportedException;

import org.springframework.web.bind.MethodArgumentNotValidException;

import org.springframework.web.bind.MissingPathVariableException;

import org.springframework.web.bind.annotation.ExceptionHandler;

import org.springframework.web.bind.annotation.RestControllerAdvice;

import org.springframework.web.method.annotation.MethodArgumentTypeMismatchException;

import com.ruoyi.common.core.constant.HttpStatus;

import com.ruoyi.common.core.exception.DemoModeException;

import com.ruoyi.common.core.exception.InnerAuthException;

import com.ruoyi.common.core.exception.PreAuthorizeException;

import com.ruoyi.common.core.exception.ServiceException;

import com.ruoyi.common.core.exception.auth.NotPermissionException;

import com.ruoyi.common.core.exception.auth.NotRoleException;

import com.ruoyi.common.core.text.Convert;

import com.ruoyi.common.core.utils.StringUtils;

import com.ruoyi.common.core.utils.html.EscapeUtil;

import com.ruoyi.common.core.web.domain.AjaxResult;



/**

 * 全局异常处理器

 * 
 *

 * @author ruoyi

 */

@RestControllerAdvice

@@ -27,22 +32,32 @@
    private static final Logger log = LoggerFactory.getLogger(GlobalExceptionHandler.class);



    /**

     * 权限异常

     * 权限码异常

     */

    @ExceptionHandler(PreAuthorizeException.class)

    public AjaxResult handlePreAuthorizeException(PreAuthorizeException e, HttpServletRequest request)

    @ExceptionHandler(NotPermissionException.class)

    public AjaxResult handleNotPermissionException(NotPermissionException e, HttpServletRequest request)

    {

        String requestURI = request.getRequestURI();

        log.error("请求地址'{}',权限校验失败'{}'", requestURI, e.getMessage());

        return AjaxResult.error(HttpStatus.FORBIDDEN, "没有权限，请联系管理员授权");

        log.error("请求地址'{}',权限码校验失败'{}'", requestURI, e.getMessage());

        return AjaxResult.error(HttpStatus.FORBIDDEN, "没有访问权限，请联系管理员授权");

    }



    /**

     * 角色权限异常

     */

    @ExceptionHandler(NotRoleException.class)

    public AjaxResult handleNotRoleException(NotRoleException e, HttpServletRequest request)

    {

        String requestURI = request.getRequestURI();

        log.error("请求地址'{}',角色权限校验失败'{}'", requestURI, e.getMessage());

        return AjaxResult.error(HttpStatus.FORBIDDEN, "没有访问权限，请联系管理员授权");

    }



    /**

     * 请求方式不支持

     */

    @ExceptionHandler(HttpRequestMethodNotSupportedException.class)

    public AjaxResult handleHttpRequestMethodNotSupported(HttpRequestMethodNotSupportedException e,

            HttpServletRequest request)

    public AjaxResult handleHttpRequestMethodNotSupported(HttpRequestMethodNotSupportedException e, HttpServletRequest request)

    {

        String requestURI = request.getRequestURI();

        log.error("请求地址'{}',不支持'{}'请求", requestURI, e.getMethod());

@@ -61,6 +76,33 @@
    }



    /**

     * 请求路径中缺少必需的路径变量

     */

    @ExceptionHandler(MissingPathVariableException.class)

    public AjaxResult handleMissingPathVariableException(MissingPathVariableException e, HttpServletRequest request)

    {

        String requestURI = request.getRequestURI();

        log.error("请求路径中缺少必需的路径变量'{}',发生系统异常.", requestURI, e);

        return AjaxResult.error(String.format("请求路径中缺少必需的路径变量[%s]", e.getVariableName()));

    }



    /**

     * 请求参数类型不匹配

     */

    @ExceptionHandler(MethodArgumentTypeMismatchException.class)

    public AjaxResult handleMethodArgumentTypeMismatchException(MethodArgumentTypeMismatchException e, HttpServletRequest request)

    {

        String requestURI = request.getRequestURI();

        String value = Convert.toStr(e.getValue());

        if (StringUtils.isNotEmpty(value))

        {

            value = EscapeUtil.clean(value);

        }

        log.error("请求参数类型不匹配'{}',发生系统异常.", requestURI, e);

        return AjaxResult.error(String.format("请求参数类型不匹配，参数[%s]要求类型为：'%s'，但输入值为：'%s'", e.getName(), e.getRequiredType().getName(), value));

    }



    /**

     * 拦截未知的运行时异常

     */

    @ExceptionHandler(RuntimeException.class)