Stream流收集器的购物车DDD聚合真实示例 - foojay

21-05-11 banq

Java Stream的Collectors方法适合大多数用例。它们允许返回aCollection或标量。对于前者,使用一种toXXX()方法,对于后者,使用一种方法reducing()。

让我们想象一个实现购物车的电子商务平台。该购物车的建模如下:

这非常类似DDD设计中聚合,Cart作为一个聚合根实体。

public class Product {

    private final Long id;                           // 1
    private final String label;                      // 1
    private final BigDecimal price;                  // 1

    public Product(Long id, String label, BigDecimal price) {
        this.id = id;
        this.label = label;
        this.price = price;
    }

    @Override
    public boolean equals(Object object ) { ... }    // 2

    @Override
    public int hashCode() { ... }                    // 2
}
public class Cart {

    private final Map<Product, Integer> products = new HashMap<>(); // 1

    public void add(Product product) {
        add(product, 1);
    }

    public void add(Product product, int quantity) {
        products.merge(product, quantity, Integer::sum);
    }

    public void remove(Product product) {
        products.remove(product);
    }

    public void setQuantity(Product product, int quantity) {
        products.put(product, quantity);
    }

    public Map<Product, Integer> getProducts() {
        return Collections.unmodifiableMap(products);               // 2
    }
}

定义了如何在内存中存储数据后,我们需要设计如何在屏幕上显示购物车。我们知道,结帐屏幕需要显示两个不同的信息位:

  • 行的列表,其中每一行的价格,即每种产品的价格乘以数量。
  • 购物车的整体价格。

相应代码:

public record CartRow(Product product, int quantity) {                // 1

    public CartRow(Map.Entry<Product, Integer> entry) {
        this(entry.getKey(), entry.getValue());
    }

    public BigDecimal getRowPrice() {
        return product.getPrice().multiply(new BigDecimal(quantity));
    }
}
var rows = cart.getProducts()
    .entrySet()
    .stream()
    .map(CartRow::new)
    .collect(Collectors.toList());                                    // 1

var price = cart.getProducts()
    .entrySet()
    .stream()
    .map(CartRow::new)
    .map(CartRow::getRowPrice)                                        // 2
    .reduce(BigDecimal.ZERO, BigDecimal::add);                        // 3

Java流的主要限制之一是只能使用一次。原因是流对象不一定是不变的(尽管它们可以是不变的)。因此,两次执行相同的流可能不是幂等的。

因此,要获取行和价格,我们需要从购物车创建两个流。从一个流中,我们将获得行,而从另一流中,将获得价格。

如果我们想从单个流中收集行和价格。我们需要一个可Collector在一次通过中将两个对象都作为单个对象返回的自定义。

public class PriceAndRows {

    private BigDecimal price;                              // 1
    private final List<CartRow> rows = new ArrayList<>();  // 2

    PriceAndRows(BigDecimal price, List<CartRow> rows) {
        this.price = price;
        this.rows.addAll(rows);
    }

    PriceAndRows() {
        this(BigDecimal.ZERO, new ArrayList<>());
    }
}

这是Collector接口的摘要。有关更多详细信息,请检查此以前的帖子

  • supplier()    提供基础对象以开始
  • accumulator()    描述如何将当前流式项目累积到容器中
  • combiner()    如果流是并行的,请描述如何合并它们
  • finisher()    如果可变容器类型不是返回的类型,请描述如何将前者转换为后者
  • characteristics()    提供元数据以优化流

鉴于此,我们可以相应地实现Collector:

private static class PriceAndRowsCollector
    implements Collector<Map.Entry<Product, Integer>, PriceAndRows, PriceAndRows> {

    @Override
    public Supplier<PriceAndRows> supplier() {
        return PriceAndRows::new;                                                // 1
    }

    @Override
    public BiConsumer<PriceAndRows, Map.Entry<Product, Integer>> accumulator() {
        return (priceAndRows, entry) -> {                                        // 2
            var row = new CartRow(entry);
            priceAndRows.price = priceAndRows.price.add(row.getRowPrice());
            priceAndRows.rows.add(row);
        };
    }

    @Override
    public BinaryOperator<PriceAndRows> combiner() {
        return (c1, c2) -> {                                                     // 3
            c1.price = c1.price.add(c2.price);
            var rows = new ArrayList<>(c1.rows);
            rows.addAll(c2.rows);
            return new PriceAndRows(c1.price, rows);
        };
    }

    @Override
    public Function<PriceAndRows, PriceAndRows> finisher() {
        return Function.identity();                                              // 4
    }

    @Override
    public Set<Characteristics> characteristics() {
        return Set.of(Characteristics.IDENTITY_FINISH);                          // 4
    }
}

设计Collector涉及一些工作,但是使用自定义收集器很容易:

var priceAndRows = cart.getProducts()
                       .entrySet()
                       .stream()
                       .collect(new PriceAndRowsCollector());
 

您可以使用Collectors该类中提供的即用型收集器来解决大多数用例。但是,有些需要实现自定义Collector,例如,当您需要收集多个单个集合或单个标量时,则需要实现一个custom 。

如果您以前从未开发过它,可能看起来很复杂,但事实并非如此。您只需要一点练习即可。希望这篇文章对您有所帮助。

您可以在GitHub上以Maven格式找到此帖子的源代码。

猜你喜欢