如何用Lambda与Stream优化Map嵌套循环构建JavaFX TreeItem

嘿，很高兴帮你优化这段代码！你的需求是把Map的每个键转化为TreeItem父节点，再将对应List中的元素作为子节点，用Java 8的Stream和Lambda完全可以让这段代码更简洁、更符合函数式风格，同时避免嵌套循环的冗余。

首先提个小细节：原代码里的myMap.get(k)其实是多余的——在forEach((k,v) -> {...})中，v已经是当前键k对应的List了，没必要再从Map里重新获取，这是个小小的性能浪费，优化的时候可以直接用v。

下面是两种优化方案，你可以根据自己的偏好选择：

方案一：用Stream处理子节点，保持直观 #

Map<MyEnum, List<MyObject>> myMap = new HashMap<>();
List<TreeItem<String>> product = myMap.entrySet()
    .stream()
    .map(entry -> {
        // 创建父节点，使用entry的键名称
        TreeItem<String> fatherItem = new TreeItem<>(entry.getKey().getName());
        // 将子元素Stream转换为TreeItem并添加到父节点
        entry.getValue().stream()
            .map(obj -> new TreeItem<>(obj.getName()))
            .forEach(fatherItem.getChildren()::add);
        return fatherItem;
    })
    .collect(Collectors.toList());

这里我们直接遍历Map的entrySet()流，每个键值对（Entry）映射成一个父TreeItem：

1. 先创建父节点，用Entry的键名称初始化
2. 把Entry对应的List转成Stream，每个元素映射为子TreeItem，再通过方法引用fatherItem.getChildren()::add批量添加到父节点的子列表中
3. 最后把所有父节点收集成List

方案二：先收集子节点再批量添加（性能更优） #

如果想彻底去掉内部的forEach，可以先把所有子节点收集成List，再一次性添加到父节点：

Map<MyEnum, List<MyObject>> myMap = new HashMap<>();
List<TreeItem<String>> product = myMap.entrySet()
    .stream()
    .map(entry -> {
        // 先把所有子元素转换为TreeItem并收集成List
        List<TreeItem<String>> childItems = entry.getValue().stream()
            .map(obj -> new TreeItem<>(obj.getName()))
            .collect(Collectors.toList());
        // 创建父节点并批量添加子节点
        TreeItem<String> fatherItem = new TreeItem<>(entry.getKey().getName());
        fatherItem.getChildren().addAll(childItems);
        return fatherItem;
    })
    .collect(Collectors.toList());

这个版本用collect先把子节点汇总，再调用addAll一次性添加，减少了多次调用add的开销（虽然差异不大，但在元素量很大时会更高效），代码结构也更清晰。

另外提醒下：Java命名规范里类名要大写开头，所以你的myObject建议改成MyObject哦～

内容的提问来源于stack exchange，提问作者Ross