package com.study.utils;

import com.study.entity.Per;

import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;

public class TreeUtils {

public static Per getCurrentNode(List<Per> list){
    Per per = list.stream().min(Comparator.comparing(Per::getId)).get();

    return setChildrenNode(per, list);
}
public static Per setChildrenNode(Per per,List<Per> list){
    List<Per> collect = list.stream().filter(item -> {
        return per.getId().equals(item.getPid());
    }).collect(Collectors.toList());

    per.setPerList(collect);

    if(collect.size()!=0){
        collect.stream().forEach(item -> {setChildrenNode(item,list);});
    }

    return per;
}

}