``` /** * @author 張躍帥 * @Description: 系統角色-工具 * @date 2020/08/12 */ public class SystemRoleUtil { private static final SystemRoleService systemRoleService = SpringUtil.getBean(SystemRoleService.class); /** * 獲取-機構角色List */ public static List<SystemRoleEntity> getOrgRoleList(String orgCode, String roleName) { // 獲取參數 String appCode = SystemContext.me().getAppCode(); // 創建查詢包裝器 LambdaQueryWrapper<SystemRoleEntity> queryWrapper = new LambdaQueryWrapper<>(); // 設置條件 queryWrapper.eq(SystemRoleEntity::getAppCode, appCode) .eq(SystemRoleEntity::getOrgCode, orgCode) .eq(SystemRoleEntity::getDataFlag, CommonDataFlagEnum.NORMAL.getCode()) .eq(SystemRoleEntity::getDelFlag, CommonDelFlagEnum.NOT_DELETE.getCode()); // 設置排序 queryWrapper.orderByAsc(SystemRoleEntity::getSort); // 判斷 if (StrUtil.isNotBlank(roleName)) { // 追加條件 queryWrapper.like(SystemRoleEntity::getRoleName, roleName); } // 返回 return systemRoleService.list(queryWrapper); } /** * 獲取-機構類型tree樹 */ public static List<Tree<String>> getOrgRoleTree(String orgCode, String roleName) { // 獲取參數 String appCode = SystemContext.me().getAppCode(); // 變量創建ArrayList List<Tree<String>> treeNodeList = CollectionUtil.newArrayList(); // 創建查詢包裝器 LambdaQueryWrapper<SystemRoleEntity> queryWrapper = new LambdaQueryWrapper<>(); // 設置條件 queryWrapper.eq(SystemRoleEntity::getAppCode, appCode) .eq(SystemRoleEntity::getOrgCode, appCode) .eq(SystemRoleEntity::getDataFlag, CommonDataFlagEnum.NORMAL.getCode()) .eq(SystemRoleEntity::getDelFlag, CommonDelFlagEnum.NOT_DELETE.getCode()); // 設置排序 queryWrapper.orderByAsc(SystemRoleEntity::getSort); // 查詢 List<SystemRoleEntity> systemRoleList = systemRoleService.list(queryWrapper); // 判斷 if (systemRoleList != null && systemRoleList.size() > 0) { // 變量-tree父節點id String parentId = CommonTreeParentIdEnum.PARENTID_0.getCode(); // tree轉換器 treeNodeList = TreeUtil.build(systemRoleList, parentId, (systemRole, tree) -> { // 賦值 tree.setId(Convert.toStr(systemRole.getId())); tree.setParentId(parentId); tree.setName(systemRole.getRoleName()); // 追加自定義屬性 tree.putExtra("code", systemRole.getRoleCode()); }); } // 判斷-條件搜索 if (treeNodeList != null && treeNodeList.size() > 0 && StrUtil.isNotBlank(roleName)) { // 變量創建ArrayList List<Tree<String>> searchTreeNodeList = CollectionUtil.newArrayList(); // 遍歷 for (Tree<String> treeNode : treeNodeList) { // 搜索查詢 searchTreeNodeByName(treeNode, roleName, searchTreeNodeList); } // 賦值 treeNodeList = searchTreeNodeList; } // 返回 return treeNodeList; } /** * 遞歸遍歷樹節點-搜索匹配節點 */ private static List<Tree<String>> searchTreeNodeByName(Tree<String> treeNode, String name, List<Tree<String>> searchTreeNodeList) { // 獲取參數 String nodeName = treeNode.getName().toString(); List<Tree<String>> childNodeList = treeNode.getChildren(); // 判斷 if (nodeName.contains(name)) { // 添加 searchTreeNodeList.add(treeNode); } // 判斷 if (childNodeList != null && childNodeList.size() > 0) { // 遍歷 for (Tree<String> childNode : treeNode.getChildren()) { // 匹配子節點 searchTreeNodeByName(childNode, name, searchTreeNodeList); } } // 返回 return searchTreeNodeList; } }