src/main/java/studio/baxia/fo/common/TreeInfoUtil.java

这里只显示每个文件的开头 60 行。登录后可以解锁完整代码。

微信扫码关注公众号，代码整理好后第一时间通知你。

package studio.baxia.fo.common;

import java.util.ArrayList;
import java.util.List;

/**
 * Created by Pan on 2016/10/15.
 * TreeInfo工具类
 */
public class TreeInfoUtil {

    /**
     * 将List<? extends TreeInfo>转换成TreeInfoResult
     * @param treeInfos 实现了TreeInfo的类的集合形式
     * @return TreeInfoResult
     */
    public static TreeInfoResult convertToTreeInfoResult(List<? extends TreeInfo> treeInfos,TreeInfo rootTreeInfo) {
        TreeInfoResult treeRootInfoResult = new TreeInfoResult(rootTreeInfo);
        List<TreeInfoResult> treeRootInfoResultChildrens = new ArrayList<TreeInfoResult>();
        List<TreeInfo> restTreeInfos = new ArrayList<TreeInfo>();
        if(rootTreeInfo==null){
            for (int i = 0; i < treeInfos.size(); i++) {
                TreeInfo treeInfo = treeInfos.get(i);
                if (treeInfo.getParentId() == CommonConstant.MESSAGE_DEFAULT_PARENT_ID) {
                    treeRootInfoResultChildrens.add(new TreeInfoResult(treeInfo));
                } else {
                    restTreeInfos.add(treeInfo);
                }
            }
        }else{
            for (int i = 1; i < treeInfos.size(); i++) {
                TreeInfo treeInfo = treeInfos.get(i);
                if (treeInfo.getParentId() == rootTreeInfo.getId()) {
                    treeRootInfoResultChildrens.add(new TreeInfoResult(treeInfo));
                } else {
                    restTreeInfos.add(treeInfo);
                }
            }
        }
        getChildrenTree(treeRootInfoResultChildrens, restTreeInfos);
        treeRootInfoResult.setChildrens(treeRootInfoResultChildrens);
        return treeRootInfoResult;
    }

    private static void getChildrenTree(List<TreeInfoResult> treeInfoResultChildrens, List<TreeInfo> treeInfos) {
        List<TreeInfo> restTreeInfos;
        for (int i = 0; i < treeInfoResultChildrens.size(); i++) {
            TreeInfoResult treeInfoResult = treeInfoResultChildrens.get(i);
            if (treeInfoResult != null) {
                int parentId = treeInfoResult.getData().getId();
                List<TreeInfoResult> treeInfoChildrens = new ArrayList<TreeInfoResult>();
                restTreeInfos = new ArrayList<>();
                for (int j = 0; j < treeInfos.size(); j++) {
                    TreeInfo treeInfo = treeInfos.get(j);
                    if (treeInfo.getParentId() == parentId) {
                        treeInfoChildrens.add(new TreeInfoResult(treeInfo));
                    } else {
                        restTreeInfos.add(treeInfo);
                    }
                }