package com.qianhe.common.utils;

import com.qianhe.common.core.domain.entity.SysDept;

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

public class MtTreeUtils {
    /**
     * 查询指定单位的下一级单位
     * @param deptList 所有单位的列表
     * @param parentId 目标单位的ID（查询其下一级）
     * @return 下一级单位列表（无则返回空列表）
     */
    public static List<SysDept> getDirectChildren(List<SysDept> deptList, Long parentId) {
        if (deptList == null || parentId == null) {
            return new ArrayList<>();
        }

        // 筛选出父级ID等于目标parentId的单位
        return deptList.stream()
                .filter(org -> parentId.equals(org.getParentId()))
                .collect(Collectors.toList());
    }
}
