若依如何实现的数据范围过滤

/**

• 数据过滤处理

• @author sj

• /
  @Aspect
  @Component
  public class DataScopeAspect
  {
  /**

  • 全部数据权限
  • /
    public static final String DATA_SCOPE_ALL = “1”;

  /**

  • 自定数据权限
  • /
    public static final String DATA_SCOPE_CUSTOM = “2”;

  /**

  • 部门数据权限
  • /
    public static final String DATA_SCOPE_DEPT = “3”;

  /**

  • 部门及以下数据权限
  • /
    public static final String DATA_SCOPE_DEPT_AND_CHILD = “4”;

  /**

  • 仅本人数据权限
  • /
    public static final String DATA_SCOPE_SELF = “5”;

  /**

  • 数据权限过滤关键字
  • /
    public static final String DATA_SCOPE = “dataScope”;

  // 配置织入点
  @Pointcut(“@annotation(com.sj.common.annotation.DataScope)”)
  public void dataScopePointCut()
  {
  }
  // 在织入点之前进行织入
  @Before(“dataScopePointCut()”)
  public void doBefore(JoinPoint point) throws Throwable
  {

    // 在织入点之前执行，根据当前登录的用户判断是否数据权限过滤
    handleDataScope(point);
  

  }

  protected void handleDataScope(final JoinPoint joinPoint)
  {

    // 获得注解
    DataScope controllerDataScope = getAnnotationLog(joinPoint);
    if (controllerDataScope == null)
    {
        return;
    }
    // 获取当前的用户
    LoginUser loginUser = SpringUtils.getBean(TokenService.class).getLoginUser(ServletUtils.getRequest());
    if (StringUtils.isNotNull(loginUser))
    {
        SysUser currentUser = loginUser.getUser();
        // 如果当前登录的用户不是管理员那就进行数据范围过滤
        if (StringUtils.isNotNull(currentUser) && !currentUser.isAdmin())
        {
            dataScopeFilter(joinPoint, currentUser, controllerDataScope.deptAlias(),
                    controllerDataScope.userAlias());
        }
    }
  

  }

  /**

  • 数据范围过滤
  • @param joinPoint 切点
  • @param user 用户
  • @param userAlias 别名
  • /
    public static void dataScopeFilter(JoinPoint joinPoint, SysUser user, String deptAlias, String userAlias)
    {
    // 构建字符串
    StringBuilder sqlString = new StringBuilder();
    // 遍历角色，一个用户可以有多个角色
    for (SysRole role : user.getRoles())
    {

      // 获取到数据权限范围
      String dataScope = role.getDataScope();
      // 如果是全部数据权限则不需要构造拼接字符串
      if (DATA_SCOPE_ALL.equals(dataScope))
      {
          sqlString = new StringBuilder();
          break;
      }
      else if (DATA_SCOPE_CUSTOM.equals(dataScope))
      {
          sqlString.append(StringUtils.format(
                  " OR {}.dept_id IN ( SELECT dept_id FROM sys_role_dept WHERE role_id = {} ) ", deptAlias,
                  role.getRoleId()));
      }
      // 本部门的数据
      else if (DATA_SCOPE_DEPT.equals(dataScope))
      {
          sqlString.append(StringUtils.format(" OR {}.dept_id = {} ", deptAlias, user.getDeptId()));
      }
      // 本部门及本部门以下
      else if (DATA_SCOPE_DEPT_AND_CHILD.equals(dataScope))
      {
          // dept_id等于本本部门的id或者dept_id在祖级列表里面
          sqlString.append(StringUtils.format(
                  " OR {}.dept_id IN ( SELECT dept_id FROM sys_dept WHERE dept_id = {} or find_in_set( {} , ancestors ) )",
                  deptAlias, user.getDeptId(), user.getDeptId()));
      }
      else if (DATA_SCOPE_SELF.equals(dataScope))
      {
          if (StringUtils.isNotBlank(userAlias))
          {
              sqlString.append(StringUtils.format(" OR {}.user_id = {} ", userAlias, user.getUserId()));
          }
          else
          {
              // 数据权限为仅本人且没有userAlias别名不查询任何数据
              sqlString.append(" OR 1=0 ");
          }
      }
    

    } // 将sqlString转换为String且当不为空时
    if (StringUtils.isNotBlank(sqlString.toString()))
    {

      // 获取连接点的参数，
      Object params = joinPoint.getArgs()[0];
      // 如果参数不为空并且是一个实体
      if (StringUtils.isNotNull(params) && params instanceof BaseEntity)
      {
          // 将参数转换为实体
          BaseEntity baseEntity = (BaseEntity) params;
          // 将拼接的Sql语句放到参数列表
          baseEntity.getParams().put(DATA_SCOPE, " AND (" + sqlString.substring(4) + ")");
      }
    

    }
    }

  /**

  • 是否存在注解，如果存在就获取
  • /
    private DataScope getAnnotationLog(JoinPoint joinPoint)
    {
    // 获取封装了签名信息的对象，在该对象中可以获取到目标方法名，所属类的class等信息
    Signature signature = joinPoint.getSignature();
    // 将签名对象转换为方法签名对象
    MethodSignature methodSignature = (MethodSignature) signature;
    // 获取到目标方法
    Method method = methodSignature.getMethod();
    // 如果目标方法不为空则返回目标方法上的注解
    if (method != null)
    {

      return method.getAnnotation(DataScope.class);
    

    }
    // 否则返回空
    return null;
    }
    }

// 作为前置知识需要了解角色管理页面中，可以配置下面几种数据范围
/**
* 全部数据权限
/
public static final String DATA_SCOPE_ALL = “1”;
/*
* 自定数据权限
*/
public static final String DATA_SCOPE_CUSTOM = “2”;

查询 部门角色对应表中，用户具备的角色可以管理那些部门
OR d.dept_id IN ( SELECT dept_id FROM sys_role_dept WHERE role_id = {rolelist} )

/**
* 部门数据权限
*/
public static final String DATA_SCOPE_DEPT = “3”;

查询部门编号=用户所在部门编号的数据
OR {}.dept_id = {}

/**
* 部门及以下数据权限
*/
public static final String DATA_SCOPE_DEPT_AND_CHILD = “4”;

查询部门和子部门的权限

/**
 * 仅本人数据权限
 */
public static final String DATA_SCOPE_SELF = "5";

if (StringUtils.isNotBlank(userAlias))

{
sqlString.append(StringUtils.format(“ OR {}.user_id = {} “, userAlias, user.getUserId()));
}
else
{
// 数据权限为仅本人且没有userAlias别名不查询任何数据
sqlString.append(“ OR 1=0 “);
}