package org.sxkj.odm.utils;

import org.springframework.util.CollectionUtils;
import org.sxkj.common.func.Streams;
import org.sxkj.common.utils.NoNullUtils;
import org.sxkj.system.vo.TreeVo;

import java.util.List;
import java.util.Map;

/**
 * tree 工具类
 * @author zhongrj
 * @date 2025-06-18
 */
public class TreeUtils {

	/**
	 * 构建树
	 *
	 * @param listMap key=区域code,value=列表
	 * @param data     区域树
	 * @param map      key=区域code,value=数量
	 * @param listMap
	 * @return
	 */
	public static TreeVo buildTreeVo(TreeVo data, Map<String, Double> map, Map<String, ? extends List<?>> listMap) {
		if (data == null) {
			return null;
		}
		if (CollectionUtils.isEmpty(map)) {
			return data;
		}
		String id = data.getId();
		//明细列表
		List list = listMap.get(id);
		data.setData(list);
		List<TreeVo> children = data.getChildrens();
		//总数
		Double totalNum = Streams.reduce(children, entity -> NoNullUtils.doubleValue(map.get(entity.getId())), 0.0, Double::sum);
		data.setNumber(totalNum);
		Double num = NoNullUtils.doubleValue(map.get(data.getId()));
		if (num > 0) {
			data.setNumber(num);
		}
		if (!CollectionUtils.isEmpty(children)) {
			for (TreeVo treeVo : children) {
				buildTreeVo(treeVo, map, listMap);
			}
		}
		return data;
	}
}