java - 将两个 foreach 更改为一个流

Question

有人可能知道如何用 stream() 替换它？

for (Customer customer : customers) {
    if (customer.getProducts() != null && customer.getProducts().getProduct() != null) {
        for (Product product : customer.getProducts().getProduct()) {
            if (product.getId().equals(productId)) {
                return Optional.of(product);
            }
        }
    }
}

我知道，第二个 foreach 我可以替换为:

customer.getProducts().getProduct().stream().filter(a -> a.getId().equals(productId)).findAny();

但是如何用一个流替换两个 foreach？

Answer 1

使用 flatMap 获取所有 Customer 的所有 Product 的 Stream。

Optional<Product> product =
    customers.stream()
             .filter(c -> c.getProducts() != null && c.getProducts().getProduct() != null)
             .flatMap(c -> c.getProducts().getProduct().stream())
             .filter(p -> p.getId().equals(productId))
             .findFirst();