Java 8 流 - 嵌套映射上的过滤器和重建

Question

我的表单中有一个嵌套 map

TreeMap<LocalDate, Map<Long,TreeMap<BigDecimal,String>>>

我必须处理这个 map 并最终得到一个与嵌套 TreeMap 结构相同的 map

TreeMap<BigDecimal, String>>

恰好有两个元素。

我可以通过

找到我想要的元素

values.entrySet().stream().flatMap(date -> date.getValue().entrySet().stream()
                                             .map(type -> type.getValue().entrySet()))
                                             .filter(valueMap -> valueMap.size() == 2 )

但我不知道如何表达 .collect() 来重新组装结构。欢迎任何指点。

Answer 1

调用 flatMap 和 map 时您会丢失信息。您需要保留 key 以便能够重建您的结构。

稍加改动，您就可以过滤内部映射并使用内部流收集它们，而不会影响外部流的结构:

Map<LocalDate, Map<Long, TreeMap<BigDecimal, String>>> result = values.entrySet().stream()
        .collect(
                Collectors.toMap(
                        Entry::getKey,
                        entry -> entry.getValue()
                           .entrySet()
                           .stream()
                           .filter(subEntry -> subEntry.getValue().size() == 2)
                           .collect(Collectors.toMap(Entry::getKey,
                                 Entry::getValue)))
                        );