/**
  * 解析用户组节点, 挂靠到各自的父节点下
  *
  * @param root 应用的根节点
  * @param groups
  * @param treeNodeMap
  */
 protected void parserGroup(TreeNode root, List<Group> groups, Map<Long, TreeNode> treeNodeMap) {
   for (Group group : groups) {
     TreeNode groupParent = (TreeNode) treeNodeMap.get(group.getParentId());
     TreeNode treeNode = (TreeNode) treeNodeMap.get(group.getId());
     if (groupParent == null) {
       root.addChild(treeNode);
     } else {
       groupParent.addChild(treeNode);
     }
   }
 }
  public TreeNode parse(Object data) {
    List<Group> mainAndAssistantGroups = (List<Group>) data;

    TreeNode root = new TreeNode();
    Map<Long, TreeNode> treeNodeMap = new HashMap<Long, TreeNode>();

    // 解析主用户组和辅助用户组
    if (!EasyUtils.isNullOrEmpty(mainAndAssistantGroups)) {
      for (Group group : mainAndAssistantGroups) {
        TreeNode item = new TreeNode(group);
        treeNodeMap.put(group.getId(), item);
      }

      parserGroup(root, (List<Group>) mainAndAssistantGroups, treeNodeMap);
    }

    return root;
  }
Example #3
0
  public void testRoleModule() {
    // 新建一个用户组
    Group mainGroup = new Group();
    mainGroup.setParentId(UMConstants.MAIN_GROUP_ID);
    mainGroup.setName("财务部");
    mainGroup.setGroupType(Group.MAIN_GROUP_TYPE);
    mainGroup.setApplicationId(Config.getAttribute(Config.APPLICATION_CODE));
    groupService.createNewGroup(mainGroup, "", "");
    log.debug(mainGroup);

    // 新建一个用户组子组
    Group childGroup = new Group();
    childGroup.setParentId(mainGroup.getId());
    childGroup.setName("财务一部");
    childGroup.setGroupType(mainGroup.getGroupType());
    childGroup.setApplicationId(mainGroup.getApplicationId());
    groupService.createNewGroup(childGroup, "", "");
    log.debug(childGroup);

    // 管理员直接在主组下新增用户
    User mainUser = new User();
    mainUser.setApplicationId(mainGroup.getApplicationId());
    mainUser.setLoginName("JonKing");
    mainUser.setUserName("JK");
    mainUser.setPassword("123456");
    mainUser.setGroupId(mainGroup.getId());
    userService.createOrUpdateUserInfo(mainGroup.getId(), mainUser, "" + mainGroup.getId(), "");
    log.debug(mainUser);

    // 新建角色组
    action.getRole().setIsGroup(1);
    action.getRole().setName("角色组一");
    action.getRole().setParentId(UMConstants.ROLE_ROOT_ID);
    action.saveRoleGroupInfo();
    Long roleGroupId = action.getRole().getId();

    // 新建角色
    resetAction();
    action.getRole().setIsGroup(0);
    action.getRole().setName("办公室助理");
    action.getRole().setParentId(roleGroupId);
    action.getRole().setStartDate(new Date());
    Calendar calendar = new GregorianCalendar();
    calendar.add(UMConstants.ROLE_LIFE_TYPE, UMConstants.ROLE_LIFE_TIME);
    action.getRole().setEndDate(calendar.getTime());
    action.setRole2UserIds(UMConstants.ADMIN_USER_ID + "," + mainUser.getId());
    action.setRole2GroupIds("" + mainGroup.getId());
    action.saveRole();
    Long roleId = action.getRole().getId();

    // 读取修改角色组的模板
    action.setRoleId(roleGroupId);
    action.getRoleGroupInfo();

    // 读取新增或修改角色的模板
    action.setIsNew(1);
    action.setParentRoleId(UMConstants.ROLE_ROOT_ID);
    action.getRoleInfoAndRelation();

    action.setIsNew(null);
    action.setRoleId(roleId);
    action.getRoleInfoAndRelation();

    // 读取角色树形结构
    action.getAllRole2Tree();
    action.getCanAddedGroup2Tree();

    // 停用角色组
    action.setRoleId(roleGroupId);
    action.setRoleState(UMConstants.TRUE);
    action.disable();

    // 启用角色
    action.setRoleId(roleId);
    action.setRoleState(UMConstants.FALSE);
    action.disable();

    // 再新建一个角色
    resetAction();
    action.getRole().setIsGroup(0);
    action.getRole().setName("部门经理");
    action.getRole().setParentId(roleGroupId);
    action.getRole().setStartDate(new Date());
    action.getRole().setEndDate(calendar.getTime());
    action.setRole2UserIds(UMConstants.ADMIN_USER_ID + "," + mainUser.getId());
    action.setRole2GroupIds("" + mainGroup.getId());
    action.saveRole();
    Long secondRoleId = action.getRole().getId();
    action.getAllRole2Tree();

    // 对组进行排序
    action.setRoleId(roleId);
    action.setTargetId(secondRoleId);
    action.setDirection(1);
    action.sort();
    action.getAllRole2Tree();

    // 对角色进行移动
    action.setTargetId(UMConstants.ROLE_ROOT_ID);
    action.move();

    action.getOperation();

    action.setGroupId(mainGroup.getId());
    action.getUserByGroupId();

    action.initSetPermission();
    action.setIsRole2Resource("1");
    action.getApplications();
    action.setIsRole2Resource("0");
    action.getApplications();
    action.setApplicationId("tss");
    action.getResourceTypes();

    // 授权测试 PermissionRank
    //      LOWER_PERMISSION            = "1";  普通授权
    //      AUTHORISE_PERMISSION        = "2";  可授权授权
    //      PASSON_AUTHORISE_PERMISSION = "3";  可传递授权
    //      SUB_AUTHORISE_PERMISSION    = "4";  权限转授
    log.debug("====================== 开始测试授权 ============================");

    // 一、 多个资源授权给单个角色
    action.setIsRole2Resource("1");
    action.setResourceType(UMConstants.MAINGROUP_RESOURCE_TYPE_ID);
    action.setRoleId(roleId);
    action.setPermissionRank("2");
    action.getPermissionMatrix();

    // 授权内容, 当多个资源对一个角色授权时:  resource1|2224, resource2|4022
    // 竖线后面为各个权限选项的打勾情况【0: 没打勾, 1: 仅此节点,虚勾 2: 此节点及所有子节点,实勾 3:禁用未选中 4:禁用已选中】
    action.setSetPermission(mainGroup.getId() + "|222222222222");
    action.savePermission();
    action.getPermissionMatrix();

    TestUtil.printEntity(super.permissionHelper, "MainGroupPermissions");
    TestUtil.printEntity(super.permissionHelper, "MainGroupPermissionsFull");

    // 二、单个资源授权给多个角色
    action.setIsRole2Resource(null);
    action.setRoleId(mainGroup.getId()); // 当资源对角色进行授权时, 表示resourceId
    action.setPermissionRank("2");
    action.getPermissionMatrix();

    // 授权内容, 当单个资源对多个角色授权时:  roleId1|2224, roleId2|4022
    // 竖线后面为各个权限选项的打勾情况【0: 没打勾, 1: 仅此节点,虚勾 2: 此节点及所有子节点,实勾 3:禁用未选中 4:禁用已选中】
    action.setSetPermission(secondRoleId + "|222222222222");
    action.savePermission();
    action.getPermissionMatrix();

    TestUtil.printEntity(super.permissionHelper, "MainGroupPermissions");
    TestUtil.printEntity(super.permissionHelper, "MainGroupPermissionsFull");

    login(mainUser.getId(), mainUser.getLoginName()); // 更好登录用户,看其权限
    action.getPermissionMatrix();
    TestUtil.printEntity(super.permissionHelper, "RoleUserMapping");
    printVisibleMainGroups();

    login(UMConstants.ADMIN_USER_ID, UMConstants.ADMIN_USER_NAME); // 换回Admin登录

    // 删除角色组
    action.setRoleId(roleGroupId);
    action.delete();
    action.getAllRole2Tree();
  }