項目中經常會遇到各種需要以樹形結構展示的功能，比較常見的，如菜單樹，分類樹，部門樹等等，如果為每種類型都遍歷遞歸生成樹形結構返回給前端，顯得有些冗余且麻煩，并且其實邏輯都是一致的，只是遍歷的對象不同而已，故其實可以通過面向接口思維，來實現這種通用工具類的實現。

TreeNode用來表示每個樹節點的抽象，即需要生成樹的對象需要實現此接口。

/** * 樹節點父類，所有需要使用{@linkplain TreeUtils}工具類形成樹形結構等操作的節點都需要實現該接口 * * @param <T> 節點id類型 */ public interface TreeNode<T> { /** * 獲取節點id * * @return 樹節點id */ T id(); /** * 獲取該節點的父節點id * * @return 父節點id */ T parentId(); /** * 是否是根節點 * * @return true：根節點 */ boolean root(); /** * 設置節點的子節點列表 * * @param children 子節點 */ void setChildren(List<? extends TreeNode<T>> children); /** * 獲取所有子節點 * * @return 子節點列表 */ List<? extends TreeNode<T>> getChildren(); }

TreeUtils用來生成樹形結構，以及獲取所有葉子節點等操作

/** * 樹形結構工具類 * * @author meilin.huang * @version 1.0 * @date 2019-08-24 1:57 下午 */public class TreeUtils { /** * 根據所有樹節點列表，生成含有所有樹形結構的列表 * * @param nodes 樹形節點列表 * @param <T> 節點類型 * @return 樹形結構列表 */ public static <T extends TreeNode<?>> List<T> generateTrees(List<T> nodes) { List<T> roots = new ArrayList<>(); for (Iterator<T> ite = nodes.iterator(); ite.hasNext(); ) { T node = ite.next(); if (node.root()) {roots.add(node);// 從所有節點列表中刪除該節點，以免后續重復遍歷該節點ite.remove(); } } roots.forEach(r -> { setChildren(r, nodes); }); return roots; } /** * 從所有節點列表中查找并設置parent的所有子節點 * * @param parent 父節點 * @param nodes 所有樹節點列表 */ @SuppressWarnings('all') public static <T extends TreeNode> void setChildren(T parent, List<T> nodes) { List<T> children = new ArrayList<>(); Object parentId = parent.id(); for (Iterator<T> ite = nodes.iterator(); ite.hasNext(); ) { T node = ite.next(); if (Objects.equals(node.parentId(), parentId)) {children.add(node);// 從所有節點列表中刪除該節點，以免后續重復遍歷該節點ite.remove(); } } // 如果孩子為空，則直接返回,否則繼續遞歸設置孩子的孩子 if (children.isEmpty()) { return; } parent.setChildren(children); children.forEach(m -> { // 遞歸設置子節點 setChildren(m, nodes); }); } /** * 獲取指定樹節點下的所有葉子節點 * * @param parent 父節點 * @param <T> 實際節點類型 * @return 葉子節點 */ public static <T extends TreeNode<?>> List<T> getLeafs(T parent) { List<T> leafs = new ArrayList<>(); fillLeaf(parent, leafs); return leafs; } /** * 將parent的所有葉子節點填充至leafs列表中 * * @param parent 父節點 * @param leafs 葉子節點列表 * @param <T> 實際節點類型 */ @SuppressWarnings('all') public static <T extends TreeNode> void fillLeaf(T parent, List<T> leafs) { List<T> children = parent.getChildren(); // 如果節點沒有子節點則說明為葉子節點 if (CollectionUtils.isEmpty(children)) { leafs.add(parent); return; } // 遞歸調用子節點，查找葉子節點 for (T child : children) { fillLeaf(child, leafs); } }}

具體使用方式之聲明樹節點對象

@Getter@Setterpublic class ResourceListVO implements TreeNode<Long> { private Long id; private Long pid; private Integer type; private String name; private String icon; private String code; private Integer status; private List<ResourceListVO> children; @Override public Long id() { return this.id; } @Override public Long parentId() { return this.pid; } @Override public boolean root() { return Objects.equals(this.pid, 0L); } @Override public void setChildren(List children) { this.children = children; }}

具體使用方式之調用

/** * 獲取賬號的資源樹 */ public List<ResourceListVO> listByAccountId(Long accountId) { return TreeUtils.generateTrees(BeanUtils.copyProperties(mapper.selectByAccountId(userId), ResourceListVO.class)); }

通過使用TreeUtils工具可以統一方便地生成一切對象的樹形結構以及其他一些對樹的操作，避免對每個對象都用特定代碼生成。使用起來就是幾個字簡潔方便爽歪歪biu特否。

補充知識：TreeUtil 數據庫菜單生成無限級樹形結構

1、項目需求：

從數據庫從加載所有的菜單出來，菜單中有

id,parentId,name字段

希望能有一個工具幫我進行樹形結構重組；

實例類：

package com.lming.chcservice.util; import lombok.Data; import java.util.List;@Datapublic class TreeNode { /** * 節點id */ private String id; /** * 父節點 默認0為根節點 */ private String parentId; /** * 節點名稱 */ private String name; /** * 是否有子節點 */ private boolean hasChild; public TreeNode(String id, String parentId, String name) { this.id = id; this.parentId = parentId; this.name = name; }}

工具類：

package com.lming.chcservice.util; import java.util.ArrayList;import java.util.LinkedHashMap;import java.util.List;import java.util.Map; /** * 樹形結構工具類 * * 將一組list對象轉成樹形結構 * 該list需符合設定的字段類型 * */public class TreeUtil { public static Map<String,Object> mapArray = new LinkedHashMap<String, Object>(); public List<TreeNode> menuCommon; public List<Object> list = new ArrayList<Object>(); public List<Object> treeMenu(List<TreeNode> menu){ this.menuCommon = menu; for (TreeNode treeNode : menu) { Map<String,Object> mapArr = new LinkedHashMap<String, Object>(); if(treeNode.getParentId().equals('0')){setTreeMap(mapArr,treeNode);list.add(mapArr); } } return list; } public List<?> menuChild(String id){ List<Object> lists = new ArrayList<Object>(); for(TreeNode a:menuCommon){ Map<String,Object> childArray = new LinkedHashMap<String, Object>(); if(a.getParentId() .equals(id)){setTreeMap(childArray,a);lists.add(childArray); } } return lists; } private void setTreeMap(Map<String,Object> mapArr,TreeNode treeNode){ mapArr.put('id', treeNode.getId()); mapArr.put('name', treeNode.getName()); mapArr.put('parentId', treeNode.getParentId()); List<?> childrens = menuChild(treeNode.getId()); if(childrens.size()>0){ mapArr.put('hasChild',true); } else{ mapArr.put('hasChildren',false); } mapArr.put('childrens', menuChild(treeNode.getId())); } public static void main(String[] args){ List<TreeNode> treeNodeList = new ArrayList<>(); TreeNode treeNode1 = new TreeNode('1','0','首頁'); TreeNode treeNode2 = new TreeNode('2','0','訂單'); TreeNode treeNode3 = new TreeNode('3','1','預約'); TreeNode treeNode4 = new TreeNode('4','2','捐獻'); TreeNode treeNode5 = new TreeNode('5','4','我的訂單'); TreeNode treeNode6 = new TreeNode('6','5','個人中心'); TreeNode treeNode7 = new TreeNode('7','6','個人中心2'); TreeNode treeNode8 = new TreeNode('8','99','個人中心3'); treeNodeList.add(treeNode1); treeNodeList.add(treeNode6); treeNodeList.add(treeNode5); treeNodeList.add(treeNode3); treeNodeList.add(treeNode4); treeNodeList.add(treeNode2); treeNodeList.add(treeNode7); treeNodeList.add(treeNode8); TreeUtil treeUtil = new TreeUtil(); System.out.print(JsonUtil.toJson(treeUtil.treeMenu(treeNodeList))); } }

測試結果：

[ { 'id': '1', 'name': '首頁', 'parentId': '0', 'hasChild': true, 'childrens': [ { 'id': '3', 'name': '預約', 'parentId': '1', 'hasChildren': false, 'childrens': [] } ] }, { 'id': '2', 'name': '訂單', 'parentId': '0', 'hasChild': true, 'childrens': [ { 'id': '4', 'name': '捐獻', 'parentId': '2', 'hasChild': true, 'childrens': [ { 'id': '5', 'name': '我的訂單', 'parentId': '4', 'hasChild': true, 'childrens': [ {'id': '6','name': '個人中心','parentId': '5','hasChild': true,'childrens': [ { 'id': '7', 'name': '個人中心2', 'parentId': '6', 'hasChildren': false, 'childrens': [] }] } ] } ] } ] }]

實力類不一致怎么辦？ 自己寫一個實體轉換類，將類的對象屬性轉換成上面的實體類，然后在調用，當然最快的方式直接修改實體類即可用。

以上這篇java之TreeUtils生成一切對象樹形結構案例就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持好吧啦網。