时间戳

首页 > 解决方案 > 我们可以使用 java Stream 来重构它吗？

java - 我们可以使用 java Stream 来重构它吗？

问题描述

是否可以在 Java 8 中使用 s重构以下for循环？Stream

将状态从内部转移到外部，同时收集不同来源的列表？

Map<StateEnum, List<List<ThreadDo>>> stateDumpListMap = new HashMap<>();
for (ThreadDumpDo dumpDo : dumpDoList) {
    Map<StateEnum, List<ThreadDo>> stateDoListMap = getStateGroup(dumpDo);
    stateDoListMap.entrySet().stream().forEach(entry -> {
       stateDumpListMap.putIfAbsent(entry.getKey(), new ArrayList<>());
       stateDumpListMap.get(entry.getKey()).add(entry.getValue());
     });
}

最近我尝试了几个用例，但对我来说似乎很难看。

我试图用它stream().map()来实现这一点，但我失败了。有人可以帮忙吗？

标签： javajava-8java-stream

解决方案

这是一种方法，使用collect而不是forEach生成Map, 并使用flatMap来消除for循环：

Map<StateEnum, List<List<ThreadDo>>> stateDumpListMap =
    dumpDoList.stream ()
              .flatMap (d->getStateGroup(d).entrySet().stream ())
              .collect(Collectors.toMap (Map.Entry::getKey,
                                         e -> {
                                            List<List<ThreadDo>> l = new ArrayList<>(); 
                                            l.add (e.getValue());
                                            return l;
                                         },
                                         (v1,v2)->{v1.addAll(v2);return v1;}));

正如 Aominè 评论的那样，可以简化为：

Map<StateEnum, List<List<ThreadDo>>> stateDumpListMap =
    dumpDoList.stream ()
              .flatMap (d->getStateGroup(d).entrySet().stream ())
              .collect(Collectors.toMap (Map.Entry::getKey,
                                         e -> new ArrayList<>(Collections.singletonList(e.getValue())),
                                         (v1,v2)->{v1.addAll(v2);return v1;}));

推荐阅读