关于java:Java工具类-将List转换为树状结构

9次阅读

共计 3703 个字符,预计需要花费 10 分钟才能阅读完成。

import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.function.BiConsumer;
import java.util.function.Function;
import java.util.function.Supplier;
import java.util.stream.Collectors;
/**

  • @program: myUtil
  • @description: 工具类 – 树状构造生成
  • @author: syf
  • @create: 2021-06-15 10:38
    */

public class TreeModel<T> {

/**
 * 下级字段名称
 */
private Function<T, ?> superColumn;
/**
 * 以后比对字段
 */
private Function<T, ?> column;
/**
 * 列表接管字段名
 */
private BiConsumer<T, List<T>> children;
/**
 * 数据列表
 */
private List<T> list;
/**
 * 数据解除关联性应用的对象
 */
private Supplier<T> decoupling;
/**
 * 私有化工具类
 */
private TreeModel() {}
/**
 * 初始化载入树状工具类
 *
 * @param list        须要整合的数据列表
 * @param superColumn 下级字段 (对象:: 字段 get 办法)
 * @param column      本级字段 (对象:: 字段 get 办法)
 * @param children    上级数据存储字段 (对象:: 字段 set 办法)
 * @return 树状工具类
 */
public static <R> TreeModel<R> load(List<R> list, Function<R, ?> superColumn,
                                    Function<R, ?> column, BiConsumer<R, List<R>> children) {TreeModel<R> treeModel = new TreeModel<>();
    treeModel.setSuperColumn(superColumn)
            .setColumn(column)
            .setChildren(children)
            .setList(list);
    return treeModel;
}
/**
 * 获取树状数据(不解除原汇合中对象与树状汇合对象的关联)*
 * @param initValue 根数据
 * @return 树状数据数据列表
 */
public List<T> getTree(Object initValue) {List<T> tree = treeAll(initValue);
    if (tree != null && tree.size() > 0) {return tree.stream().filter(ls -> initValue.equals(superColumn.apply(ls))).collect(Collectors.toList());
    }
    return tree;
}
/**
 * 获取树状数据(解除原汇合中对象与树状汇合对象的关联)*
 * @param initValue  根数据
 * @param decoupling 解除关联对象(对象::new)* @return 树状数据数据列表
 */
public List<T> getTree(Object initValue, Supplier<T> decoupling) {this.setDecoupling(decoupling);
    return tree(initValue);
}
/**
 * 获取树状数据
 *
 * @param initValue 根数据
 * @return 树状数据
 */
private List<T> treeAll(Object initValue) {if (list == null || list.size() < 1) {return list;}
    List<T> collect = list.stream().filter(f -> initValue.equals(superColumn.apply(f))).collect(Collectors.toList());
    if (collect.size() < 1) {return null;}
    collect.forEach(c -> children.accept(c, treeAll(column.apply(c))));
    return collect;
}
private List<T> tree(Object o) {List<T> childrenList = new ArrayList<>();
    if(o == null){return childrenList;}
    for (T entity : list) {if (o.equals(superColumn.apply(entity))) {T now = decoupling.get();
            copy(entity, now);
            childrenList.add(now);
        }
    }
    for (T cs : childrenList) {children.accept(cs, tree(column.apply(cs)));
    }
    return childrenList;
}
/**
 * 将 source 中的属性赋值给 target
 * @param source    数据起源类
 * @param target    数据接管类
 */
private void copy(T source, T target) {if (source == null) {throw new NullPointerException("dataSource");
    }
    if (target == null) {target = decoupling.get();
    }
    Field[] sourceFields = source.getClass().getDeclaredFields();
    Field[] targetFields =[ 期货](https://www.gendan5.com/p/2021-06-11/248014.html) target.getClass().getDeclaredFields();
    Map<String, Object> sourceMap = new HashMap<>(sourceFields.length);
    try {for (Field field : sourceFields) {if (Modifier.isFinal(field.getModifiers())) {continue;}
            field.setAccessible(true);
            sourceMap.put(field.getName(), field.get(source));
        }
        for (Field field : targetFields) {Object o = sourceMap.get(field.getName());
            if (o == null) {continue;}
            field.setAccessible(true);
            field.set(target, o);
        }
    } catch (IllegalAccessException e) {e.printStackTrace();
    }
}
private TreeModel<T> setSuperColumn(Function<T, ?> superColumn) {if (superColumn == null) {throw new NullPointerException("superColumn");
    }
    this.superColumn = superColumn;
    return this;
}
private TreeModel<T> setColumn(Function<T, ?> column) {if (column == null) {throw new NullPointerException("column");
    }
    this.column = column;
    return this;
}
private TreeModel<T> setChildren(BiConsumer<T, List<T>> children) {if (children == null) {throw new NullPointerException("children");
    }
    this.children = children;
    return this;
}
private void setList(List<T> list) {if (list == null || list.size() < 1) {throw new NullPointerException("list");
    }
    this.list = list;
}
private void setDecoupling(Supplier<T> decoupling) {if (decoupling == null) {throw new NullPointerException("decoupling");
    }
    this.decoupling = decoupling;
}

}

正文完
 0