如何在流中使用Map.computeIfAbsend（）？

Question

我有一个有趣的拼图游戏。假设我有一个字符串值列表：

["A", "B", "C"]

然后我必须在另一个系统中查询地图

{name="Annie", key="A"} -> 23
{name="Paul", key="C"} -> 16

我需要返回一个新的列表

 {key="A", count=23},
 {key="B", count=0},
 {key="C", count=16}

但是当我的一个User对象在地图中没有相应的count时，我很难进行计算。

我知道那张地图。computeIfAbsent（）可以满足我的需要，但是如何根据原始列表的内容应用它呢？

我想我需要在原始列表上进行流式处理，然后应用compute？所以我有：

valuesList.stream()
 .map(it -> valuesMap.computeIfAbsent(it.getKey(), k-> OL))
 ...

但这就是我被卡住的地方。有人能提供一些关于我如何完成我需要的东西的见解吗？




Answer 1

// First, map keys to users (assuming keys are unique for each user) ...
final Map<String,User> keyToUserMap = valuesMap.keySet().stream()
    .collect(Collectors.toMap(u -> u.key, u -> u));
        
Map<String,Long> map2 = valuesList.stream()
   .map(key -> {
     // Form a pair of (key, count)
     final User user = keyToUserMap.get(key);
     return new Object[] { 
        key, 
        user == null? 0L : valuesMap.get(user).count };
   })
  .collect(Collectors.toMap(pair -> (String) pair[0], pair -> (Long) pair[1]));

Answer 2

您可以创建一个辅助映射

示例：

public record User(String name, String key) {}
public record UserCount(String key, long count) {}

public static void main(String[] args) {
    List<String> keys = List.of("A", "B", "C");
    
    Map<User, Long> countByUser =
        Map.of(new User("Annie", "A"), 23L,
               new User("Paul", "C"), 16L));
    
    Map<String, Long> countByKey = countByUser.entrySet().stream()
        .collect(Collectors.groupingBy(entry -> entry.getKey().key(), 
            Collectors.summingLong(Map.Entry::getValue)));

    List<UserCount> userCounts = keys.stream()
        .map(key -> new UserCount(key, countByKey.getOrDefault(key, 0L)))
        .collect(Collectors.toList());

    System.out.println(userCounts);
}

输出

[UserCount[key=A, count=23], UserCount[key=B, count=0], UserCount[key=C, count=16]]

关于将computeIfAbsent（）用于流的想法，这种方法是错误的，流API的文档不鼓励这种方法。

当然，您可以使用computeIfAbsent（）来解决这个问题，但不能与streams结合使用。创建一个通过副作用运行的流不是一个好主意（至少没有令人信服的理由）。

我猜你甚至不需要Java8computeIfAbsent（），简单明了的putIfAbsent（）就足够了。

以下代码将产生相同的结果：

Map<String, Long> countByKey = new HashMap<>();

countByUser.forEach((k, v) -> countByKey.merge(k.key(), v, Long::sum));
keys.forEach(k -> countByKey.putIfAbsent(k, 0L));

List<UserCount> userCounts = keys.stream()
            .map(key -> new UserCount(key, countByKey.getOrDefault(key, 0L)))
            .collect(Collectors.toList());

如果这个选项看起来很复杂，你可以为循环创建两个增强的，而不是在地图和列表上应用forEach（）。