若依
2020-09-04 f84b1574191610803681f0c70d64b7c20a5af750
ruoyi-framework/src/main/java/com/ruoyi/framework/aspectj/DataScopeAspect.java
@@ -20,7 +20,7 @@
/**
 * 数据过滤处理
 *
 *
 * @author ruoyi
 */
@Aspect
@@ -79,6 +79,9 @@
        }
        // 获取当前的用户
        LoginUser loginUser = SpringUtils.getBean(TokenService.class).getLoginUser(ServletUtils.getRequest());
        if (loginUser == null) {
            return;
        }
        SysUser currentUser = loginUser.getUser();
        if (currentUser != null)
        {
@@ -93,10 +96,10 @@
    /**
     * 数据范围过滤
     *
     *
     * @param joinPoint 切点
     * @param user 用户
     * @param alias 别名
     * @param userAlias 别名
     */
    public static void dataScopeFilter(JoinPoint joinPoint, SysUser user, String deptAlias, String userAlias)
    {
@@ -142,8 +145,12 @@
        if (StringUtils.isNotBlank(sqlString.toString()))
        {
            BaseEntity baseEntity = (BaseEntity) joinPoint.getArgs()[0];
            baseEntity.getParams().put(DATA_SCOPE, " AND (" + sqlString.substring(4) + ")");
            Object params = joinPoint.getArgs()[0];
            if (StringUtils.isNotNull(params) && params instanceof BaseEntity)
            {
                BaseEntity baseEntity = (BaseEntity) params;
                baseEntity.getParams().put(DATA_SCOPE, " AND (" + sqlString.substring(4) + ")");
            }
        }
    }