java~集合分组groupby的实现

Posted 2020-11-01 敢于对过去告一个段落，才有信心掀开新的篇章！

对于数据聚合来说，分组操作是很常见的，在.net里有lambda和linq，而在java里也有lambda，现在我们来实现对一个集合进行分组。

一 准备工作，有两个类型

  @Value
  class Item {
    private Date createAt;
    private int count;
    private BigDecimal price;
  }

  @Value
  class Product {
    private String name;
    private String code;
    private List<Item> items;
  }

二 为两个类型建立集合，并赋值

    List<Product> products = new ArrayList<>();
    products.add(new Product("apple", "1001", Arrays.asList(
        new Item(new Date(2018, 1, 1), 10, new BigDecimal("9.99")))));

    products.add(new Product("apple", "1001", Arrays.asList(
        new Item(new Date(2018, 2, 1), 10, new BigDecimal("19.99")))));

    products.add(new Product("apple", "1001", Arrays.asList(
        new Item(new Date(2018, 3, 1), 10, new BigDecimal("29.99")))));

三 使用lambda进行分组，主要对name字段进行分组，然后把结果存在一个新的集合里

    Map<String, List<Product>> groupByPriceMap =
        products.stream().collect(Collectors.groupingBy(Product::getName));
    products = new ArrayList<>();
    for (Map.Entry<String, List<Product>> str : groupByPriceMap.entrySet()) {
      List<Item> items = new ArrayList<>();
      for (Product product : str.getValue()) {
        items.addAll(product.getItems());
      }
      products.add(new Product(str.getKey(), "", items));
    }

四 调试代码，在断点处查看分组后的结果

五 多条件分组的实现

  Function<Product, List<Object>> compositeKey = personRecord ->
        Arrays.asList(personRecord.getName(), personRecord.getCode());

    Map<Object, List<Product>> map =
        products.stream().collect(Collectors.groupingBy(compositeKey));

 感谢阅读！