zhouxy108/plusone-commons
1
0
Fork 0
forked from plusone/plusone-commons
Code Pull Requests Activity

允许构建时进行排序。

Browse Source
This commit is contained in:
ZhouXY108
2024-04-16 21:06:02 +08:00
parent 22687605b2
commit d3b19c6cb0
2 changed files with 115 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
src/test/java/xyz/zhouxy/plusone/commons/util/TreeBuilderTests.java
View File

@@ -17,29 +17,34 @@ class TreeBuilderTests {
private static final Logger log = LoggerFactory.getLogger(TreeBuilderTests.class);
private final TreeBuilder<Menu, MenuList, String> treeBuilder = new TreeBuilder<>(
Menu::getMenuCode,
Menu::getParentMenuCode,
MenuList::addChild);
menu -> Optional.ofNullable(menu.parentMenuCode),
MenuList::addChild,
(a, b) -> Integer.compare(a.getOrderNum(), b.getOrderNum()));
@Test
void testBuildTree() {
List<Menu> menus = Lists.newArrayList(
MenuItem.of("A", "首页", "/home"),
MenuList.of("B", "系统管理"),
MenuItem.of("B", "B001", "功能管理", "/sys/function-mgmt"),
MenuItem.of("B", "B002", "角色管理", "/sys/role-mgmt"),
MenuItem.of("B", "B003", "账号管理", "/sys/account-mgmt"),
MenuItem.of("B", "B004", "系统参数管理", "/sys/param-mgmt"),
MenuList.of("C", "一级菜单C"),
MenuList.of("C", "C1", "二级菜单C1"),
MenuItem.of("C1", "C1001", "三级菜单C1001", "/c/c1/c1001"),
MenuItem.of("C1", "C1002", "三级菜单C1002", "/c/c1/c1002"),
MenuItem.of("C", "C2", "二级菜单C2", "/c/c2"),
MenuItem.of("C", "C3", "二级菜单C3", "/c/c3")
MenuList.of("B", "系统管理", 3),
MenuItem.of("A", "首页", "/home", 1),
/**/MenuItem.of("B", "B002", "角色管理", "/sys/role-mgmt", 3),
/**/MenuItem.of("B", "B001", "功能管理", "/sys/function-mgmt", 4),
/**/MenuItem.of("B", "B004", "系统参数管理", "/sys/param-mgmt", 1),
/**/MenuItem.of("B", "B003", "账号管理", "/sys/account-mgmt", 2),
MenuList.of("C", "一级菜单C", 2),
/**/MenuItem.of("C", "C3", "二级菜单C3", "/c/c3", 2),
/**/MenuList.of("C", "C1", "二级菜单C1", 2),
/**//**/MenuItem.of("C1", "C1001", "三级菜单C1001", "/c/c1/c1001", 1),
/**//**/MenuItem.of("C1", "C1002", "三级菜单C1002", "/c/c1/c1002", 2),
/**/MenuItem.of("C", "C2", "二级菜单C2", "/c/c2", 1));
List<Menu> menuTreeSortedByOrderNum = treeBuilder.buildTree(menus);
log.info("menuTreeSortedByOrderNum: {}", new Gson().toJson(menuTreeSortedByOrderNum));
List<Menu> menuTreeSortedByMenuCode = treeBuilder.buildTree(
menus,
(a, b) -> a.getMenuCode().compareTo(b.getMenuCode())
);
List<Menu> menuTree = treeBuilder.buildTree(menus);
log.info("menuTree: {}", new Gson().toJson(menuTree));
log.info("menuTreeSortedByMenuCode: {}", new Gson().toJson(menuTreeSortedByMenuCode));
}
}
@@ -48,24 +53,30 @@ abstract class Menu {
protected final String parentMenuCode;
protected final String menuCode;
protected final String title;
protected final int orderNum;
public Menu(String parentMenuCode, String menuCode, String title) {
public Menu(String parentMenuCode, String menuCode, String title, int orderNum) {
this.parentMenuCode = parentMenuCode;
this.menuCode = menuCode;
this.title = title;
this.orderNum = orderNum;
}
public String getMenuCode() {
return menuCode;
}
public Optional<String> getParentMenuCode() {
return Optional.ofNullable(parentMenuCode);
public String getParentMenuCode() {
return parentMenuCode;
}
public String getTitle() {
return title;
}
public int getOrderNum() {
return orderNum;
}
}
@ToString(callSuper = true)
@@ -73,17 +84,17 @@ class MenuItem extends Menu {
private final String url;
private MenuItem(String parentMenuCode, String menuCode, String title, String url) {
super(parentMenuCode, menuCode, title);
private MenuItem(String parentMenuCode, String menuCode, String title, String url, int orderNum) {
super(parentMenuCode, menuCode, title, orderNum);
this.url = url;
}
static MenuItem of(String parentMenuCode, String menuCode, String title, String url) {
return new MenuItem(parentMenuCode, menuCode, title, url);
static MenuItem of(String parentMenuCode, String menuCode, String title, String url, int orderNum) {
return new MenuItem(parentMenuCode, menuCode, title, url, orderNum);
}
static MenuItem of(String menuCode, String title, String url) {
return new MenuItem(null, menuCode, title, url);
static MenuItem of(String menuCode, String title, String url, int orderNum) {
return new MenuItem(null, menuCode, title, url, orderNum);
}
public String getUrl() {
@@ -96,24 +107,24 @@ class MenuList extends Menu {
private List<Menu> children;
private MenuList(String parentMenuCode, String menuCode, String title) {
super(parentMenuCode, menuCode, title);
private MenuList(String parentMenuCode, String menuCode, String title, int orderNum) {
super(parentMenuCode, menuCode, title, orderNum);
}
static MenuList of(String parentMenuCode, String menuCode, String title) {
return new MenuList(parentMenuCode, menuCode, title);
static MenuList of(String parentMenuCode, String menuCode, String title, int orderNum) {
return new MenuList(parentMenuCode, menuCode, title, orderNum);
}
static MenuList of(String menuCode, String title) {
return new MenuList(null, menuCode, title);
static MenuList of(String menuCode, String title, int orderNum) {
return new MenuList(null, menuCode, title, orderNum);
}
static MenuList of(String menuCode, String title, Iterable<Menu> children) {
return of(null, menuCode, title, children);
static MenuList of(String menuCode, String title, Iterable<Menu> children, int orderNum) {
return of(null, menuCode, title, children, orderNum);
}
static MenuList of(String parentMenuCode, String menuCode, String title, Iterable<Menu> children) {
MenuList instance = of(parentMenuCode, menuCode, title);
static MenuList of(String parentMenuCode, String menuCode, String title, Iterable<Menu> children, int orderNum) {
MenuList instance = of(parentMenuCode, menuCode, title, orderNum);
for (Menu child : children) {
instance.addChild(child);
}