| | |
|---|
| | | public static List<TreeStringNode> getStringNodeTree(Map<String, TreeStringNode> treeMap){ |
|---|
| | | List<TreeStringNode> tree = new ArrayList<>(); |
|---|
| | | if (treeMap.size() > 0) { |
|---|
| | | treeMap.forEach((id, treeNode) -> { |
|---|
| | | if (treeMap.containsKey(treeNode.getParentId())) { |
|---|
| | | treeMap.get(treeNode.getParentId()).getChildren().add(treeNode); |
|---|
| | | } else { |
|---|
| | | if (treeMap.size()==1){ |
|---|
| | | treeMap.forEach((id, treeNode) -> { |
|---|
| | | tree.add(treeNode); |
|---|
| | | } |
|---|
| | | }); |
|---|
| | | }); |
|---|
| | | }else { |
|---|
| | | treeMap.forEach((id, treeNode) -> { |
|---|
| | | if (treeMap.containsKey(treeNode.getParentId())) { |
|---|
| | | treeMap.get(treeNode.getParentId()).getChildren().add(treeNode); |
|---|
| | | } else { |
|---|
| | | tree.add(treeNode); |
|---|
| | | } |
|---|
| | | }); |
|---|
| | | } |
|---|
| | | } |
|---|
| | | return tree; |
|---|
| | | } |
|---|
