java转树形结构工具类详解

java转树形结构工具类详解

本文实例为大家分享了java转树形结构工具类的具体代码，供大家参考，具体内容如下

import com.alibaba.fastjson.JSON;

import lombok.Data;

import lombok.ToString;

import org.springframework.util.ReflectionUtils;

import org.springframework.util.StringUtils;

import java.lang.reflect.Field;

import java.util.*;

/**

* @author : liyk

* @version 1.0

* @date : 2020/6/9

*/

public class TreeUtil {

/**

* 将 List 转为树形结构

*

* @param origList : 要转换的 List

* @param idFieldName : id字段名

* @param parentIdFieldName : parentId 字段名

* @param childrenFieldName : children 字段名

* @param : 拥有父子结构的 Entity

* @return : 树形结果

* @throws Exception .

*/

public static List convert(List origList, String idFieldName,

String parentIdFieldName, String childrenFieldName) throws Exception {

// 用于保存当前 id 索引的实体类

Map idMaps = new HashMap<>();

// 暂存区, 用于保存没有找到父 id 的控件

List tempList = new ArrayList<>();

List result = new ArrayList<>();

for (T entity : origList) {

// 获取 id, parentId, children

String id = Objects.toString(getFieldValue(entity, idFieldName), "");

String parentId = Objects.toString(getFieldValue(entity, parentIdFieldName), "");

if (StringUtils.isEmpty(id)) {

throw new Exception("存在id为空的资料");

}

idMaps.put(id, entity);

if (StringUtils.isEmpty(parentId)) {

// 如果父 id 为空, 则实体类为第一层

result.add(entity);

} egyIFjhlJslse {

// 根据父 id 获取实体类

T parentEntity = idMaps.get(parentId);

if (parentEntity == null) {

// 没找到先放入暂存区

tempList.add(entity);

} else {

// 父组件判断是否存在 children, 不存在新增, 存在则直接假如

setChildrenValue(childrenFieldName, entity, parentEntity);

}

}

}

// 处理暂存区, 暂存区的一定不为根节点, 所以它只要父节点存在, 那么此轮查询一定能找到父节点(上一轮已经将全部节点放入 idMaps)

for (T entity : tempList) {

// 获取 parentId

String parentId = Objects.toString(getFieldValue(entity, parentIdFieldName), "");

// 根据父id获取实体类

T parentEntity = idMaps.get(parentId);

if (parentEntity == null) {

throw new Exception("存在孤立的子节点");

} else {

// 父组件判断是否存在children, 不存在新增, 存在则直接假如

setChildrenValue(childrenFieldName, entity, parentEntity);

}

}

return result;

}

private static void setChildrenValue(String childrenFieldName, T entity, T parentEntity) throws Exception {

Object children = getFieldValue(parentEntity, childrenFieldName);

List childrenList;

if (children == null) {

childrenList = new ArrayList<>();

childrenList.add(entity);

setFieldValue(parentEntity, childrenFieldName, childrenList);

} else {

List childrenReal = (List) children;

childrenReal.add(entity);

}

}

private static Object getFieldValue(T entity, String fieldName) throws Exception {

Field field = ReflectionUtils.findField(entity.getClass(), fieldName);

if (field == null) {

throw newgyIFjhlJs Exception(String.format("字段名称[%s]不存在", fieldName));

}

boolean accessible = field.isAccessible();

field.setAccessible(true);

Object result =http:// ReflectionUtils.getField(field, entity);

field.setAccessible(accessible);

return result;

}

private static void setFieldValue(T entity, String fieldName, Object value) throws Exception {

Field field = ReflectionUtils.findField(entity.getClass(), fieldName);

if (field == null) {

throw new Exception(String.format("字段名称[%s]不存在", fieldName));

}

boolean accessible = field.isAccessible();

field.setAccessible(true);

ReflectionUtils.setField(field, entity, value);

field.setAccessible(accessible);

}

public static void main(String[] args) throws Exception {

List list = new ArrayList<>();

for (int i = 0; i < 5; i++) {

Demo demo = new Demo(i, "一级节点" + i);

list.add(demo);

}

for (int i = 5; i < 15; i++) {

Demo demo = new Demo(i, i % 5, "二级节点" + i);

list.add(demo);

}

for (int i = 15; i < 100; i++) {

Demo demo = new Demo(i, i % 10 + 5, "三级节点" + i);

list.add(demo);

}

Demo demo = new Demo(100, 102, "非法节点");

list.add(demo);

List convert = TreeUtil.convert(list, "id", "pid", "children");

String s = JSON.toJSONString(convert);

System.out.println(s);

}

}

@Data

@ToString

class Demo {

private Integer id;

private Integer pid;

private String name;

private List children;

public Demo(Integer id, Integer pid, String name) {

this.id = id;

this.pid = pid;

this.name = name;

}

public Demo(Integer id, String name) {

this.id = id;

this.name = name;

}

}

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。