Java 8 Streams - 比较两个列表对象值并将值添加到第一个列表的子对象？

Question

我有两个班级：

public class ClassOne {

 private String id;
 private String name;
 private String school;
 private String score; //Default score is null

 ..getters and setters..
}

public class ClassTwo {

 private String id;
 private String marks; 

 ..getters and setters..
}

而且，我有两个上述类的列表，

List<ClassOne> listOne;
List<ClassTwo> listTwo;

如果 ID 相等，我如何比较两个列表并将 listTwo 中的标记分配给 listOne 的分数。我知道，我们可以使用两个 for 循环来做到这一点。但我想使用 Java8 流来实现它。

List<ClassOne> result = new ArrayList<>();

for(ClassOne one : listOne) {
    for(ClassTwo two : listTwo) {
        if(one.getId().equals(two.getId())) {
            one.setScore(two.getmarks());
            result.add(one);
        }
    }
}
return result;

如何使用 Java8 lambda 和流来实现这一点？

Answer 1

设listOne.size() 为 N，listTwo.size() 为 M。 那么 2-for-loops 解决方案的复杂度为 O(M*N)。

我们可以通过 ids 索引listTwo 将其减少到 O(M+N)。

案例 1 - 假设 listTwo 没有具有相同 id 的对象

// pair each id with its marks
Map<String, String> marksIndex = listTwo.stream().collect(Collectors.toMap(ObjectTwo::getId, ObjectTwo::getMarks));
// go through list of `ObjectOne`s and lookup marks in the index
listOne.forEach(o1 -> o1.setScore(marksIndex.get(o1.getId())));

案例 2 - 假设 listTwo 具有相同 id 的对象

    final Map<String, List<ObjectTwo>> marksIndex = listTwo.stream()
            .collect(Collectors.groupingBy(ObjectTwo::getId, Collectors.toList()));

    final List<ObjectOne> result = listOne.stream()
            .flatMap(o1 -> marksIndex.get(o1.getId()).stream().map(o2 -> {
                // make a copy of ObjectOne instance to avoid overwriting scores
                ObjectOne copy = copy(o1);
                copy.setScore(o2.getMarks());
                return copy;
            }))
            .collect(Collectors.toList());

要实现copy 方法，您需要创建一个新对象并一个一个复制字段，但在这种情况下，我更喜欢遵循Builder pattern。它还会产生更多“功能性”代码。

Answer 2

以下代码从 ObjectTwo 复制标记到 ObjectOne 中，如果两个 id 相等，则没有中间对象List<ObjectOne> result

listOne.stream()
    .forEach(one -> {listTwo.stream()
        .filter(two -> {return two.getId().equals(one.getId());})
        .limit(1)
        .forEach(two -> {one.setScore(two.getMarks());});
    });

Answer 3

这应该可行。

Map<String, String> collect = listTwo.stream().collect(Collectors.toMap(ObjectTwo::getId, ObjectTwo::getMarks));
listOne
   .stream()
   .filter(item -> collect.containsKey(item.getId()))
   .forEach(item -> item.setScore(collect.get(item.getId())));