如何限制 Stream API 表达式返回的元素数量

Question

我正在使用下面的表达式，来实现以下目标：

在具有相同name::String 的Product 对象之间，提取具有最高weight::BigDecimal 值的产品。

我想通过添加到这个表达式来限制返回 Collection 的长度，但我不确定如何做到这一点 - 具体来说，我想用最多 3 个 Product 对象填充列表，理想情况下，最重的那些。

Collection<Product> heaviestProductsDistinct = products.stream()
    .collect(Collectors.toMap(Product::getName, b -> b,
        BinaryOperator.maxBy(Comparator.comparing(Product::getPrice))))
    .values();

例如，假设我有：

List<Product> products = List.of(
        new Product("product1", BigDecimal.valueOf(3), "valid"),
        new Product("product1", BigDecimal.valueOf(2), "expired"),
        new Product("product1", BigDecimal.valueOf(5), "expired"),

        new Product("product2", BigDecimal.valueOf(4), "valid"),
        new Product("product2", BigDecimal.valueOf(10), "valid"),

        new Product("product3", BigDecimal.valueOf(9), "expired"),
        new Product("product3", BigDecimal.valueOf(3)), "valid",

        new Product("product4", BigDecimal.valueOf(12), "valid"),
        new Product("product4", BigDecimal.valueOf(33), "valid")
);

我的表情会返回：

[Product{name='product4', price=33, condition='valid'}, Product{name='product2', price=10, condition='valid'}, 
Product{name='product1', price=5, condition='expired'}, Product{name='product3', price=9, condition='valid'}]

因此，我只想保留两个元素，最好是价格最高的元素。

Answer 1

你的描述不清楚，所以这里有两个

最贵的 3 种产品，每个名称只选择最贵的产品

final List<Product> collect = products.stream()
    .collect(
        Collectors.groupingBy(
            Product::getName,
            Collectors.maxBy(Comparator.comparing(Product::getPrice))
        )
    )
    .values()
    .stream()
    .filter(Optional::isPresent)
    .map(Optional::get)
    .sorted(Comparator.comparing(Product::getPrice).reversed())
    .limit(3)
    .collect(Collectors.toList());

---

每个名字最贵的 3 种产品

您可以先按名称分组，然后使用collectingAndThen 获取每个名称的结果列表，对其进行排序，并获得前 3 个。

这会给你一张地图。然后，您可以根据需要迭代并连接所有值。

List<Product> collect = products.stream()
    .collect(
        Collectors.groupingBy(
            Product::getName,
            Collectors.collectingAndThen(
                Collectors.toList(),
                list -> list.stream()
                    .sorted(Comparator.comparing(Product::getPrice).reversed())
                    .limit(3)
                    .collect(Collectors.toList())
            )
        )
    ) // here you have a Map<String, List<Product>>
    .values()
    .stream()
    .flatMap(List::stream)
    .collect(Collectors.toList());