Java8 Stream - 在单行中使用过滤器收集 foreach

Question

filter()、collect() 和 foreach() 不能在单个语句而不是多个语句中使用吗？

我有一张地图，需要根据某些条件进行过滤，并为内容设置一些值并返回地图。我的当前如下所示，但我希望所有 3 个都包含在单个语句中。

地图 inputMap（包含所有信息）

Map<String, Person> returnMap; 
            returnMap = map.entrySet().stream()
            .filter(p ->  p.getValue().getCourse() == 123)
            .collect(Collectors.toMap(p -> p.getKey(), p -> p.getValue()));

            returnMap.entrySet().stream().forEach((entry) -> {
                Person person= entry.getValue();                
                person.setAction("update");
                person.setLastUpdatedTime(new Date());
            });

这个可以转换成，

  Map<String, Person> returnMap; 
                returnMap = map.entrySet().stream()
                .filter(p ->  p.getValue().getCourse() == 123)
                .collect(Collectors.toMap(p -> p.getKey(), p -> p.getValue())) 
 .forEach((entry) -> {
                    Person person= entry.getValue();                
                    person.setAction("update");
                    person.setLastUpdatedTime(new Date());
                });

（此代码不起作用）

Answer 1

问题是forEach 没有返回一个对象，所以你必须以不同的方式处理它。你可以这样做：

Map<String, Person> returnMap = new HashMap<>(); 
map.entrySet().stream()
              .filter(p ->  p.getValue().getCourse() == 123)
              .forEach((entry) -> {
                    Person person = entry.getValue();                
                    person.setAction("update");
                    person.setLastUpdatedTime(new Date());
                    returnMap.put(entry.getKey(), person);
                });

Answer 2

坚持一次操作完成是没有意义的。不管你怎么写，这是两个操作。

但您应该考虑的一件事是，除了entrySet().stream() 之外，还有更多方法可以处理所有元素：

Map<String, Person> returnMap = map.entrySet().stream()
    .filter(p ->  p.getValue().getCourse() == 123)
    .collect(Collectors.toMap(p -> p.getKey(), p -> p.getValue()));

returnMap.values().forEach(person -> {          
    person.setAction("update");
    person.setLastUpdatedTime(new Date());
});

如果你仍然坚持让它看起来像一个单一的操作，你可以这样做：

Map<String, Person> returnMap = map.entrySet().stream()
    .filter(p ->  p.getValue().getCourse() == 123)
    .collect(Collectors.collectingAndThen(
        Collectors.toMap(p -> p.getKey(), p -> p.getValue()),
        tmp -> {
            tmp.values().forEach(person -> {          
                person.setAction("update");
                person.setLastUpdatedTime(new Date());
            });
            return tmp;
        })
    );

这在语法上是一个单一的语句，但它与前一个变体完全相同。

Answer 3

我不明白有必要按照建议将其分为两个步骤。看起来可以在终端操作之前完成转换，如下所示：

returnMap = map.entrySet().stream()
                .filter(p -> p.getValue().getCourse() == 123)
                .map(e -> {
                    Person p = e.getValue();
                    p.setAction("update");
                    p.setLastUpdatedTime(new Date());
                    return e;
                })
                .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));

如果您对流式传输的Map.Entry 的副作用感到不安，您可以返回一个新的AbstractMap.SimpleEntry，并将更新后的Person 作为其值，但到目前为止，没有一个解决方案没有副作用，因为原始地图中的值正在被修改。无论如何，这里是变化：

returnMap = map.entrySet().stream()
                .filter(p -> p.getValue().getCourse() == 123)
                .map(e -> {
                    Person p = e.getValue();
                    p.setAction("update");
                    p.setLastUpdatedTime(new Date());
                    return new AbstractMap.SimpleEntry<>(e.getKey(), p);
                })
                .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));