如何按对象字段将流分成两组？

Question

我有一个流，其中每个对象都由一个唯一的 ID 标识。 此外，每个对象都有一个正或负的Free 值。

我想将此流分成两个集合，其中一个包含 ids 的 Free 值为正数，另一个包含其余部分。

但我发现以下方法不是正确的方法，因为我正在收集流之外的列表。

class Foo {
    int free;
    long id;
}

public Tuple2<Set<Long>, Set<Long>> findPositiveAndNegativeIds() {
    Set<Long> positives = new HashSet<>();
    Set<Long> negatives = new HashSet<>();

    foos.stream()
            .forEach(f -> {
                if (f.free >= 0) positigves.add(f.id);
                else negatives.add(f.id);
            });
            
    return Tuple2.tuple(positives, negatives);
}

用partitionBy() 或类似的方法可以做得更好吗？

Answer 1

您确实可以使用partitioningBy。您可以在第二个参数中指定如何处理每个分区。

var map = foos.stream().collect(Collectors.partitioningBy(
    foo -> foo.free >= 0, // assuming no 0
    // for each partition, map to id and collect to set
    Collectors.mapping(foo -> foo.id, Collectors.toSet())
));

map.get(true) 将为您提供ids 与正frees 的集合，map.get(false) 将为您提供ids 与负frees 的集合。