java~集合分组groupby的实现

时间:2021-12-16 06:19:51

对于数据聚合来说,分组操作是很常见的,在.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));
}

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

java~集合分组groupby的实现

五 多条件分组的实现

  Function<Product, List<Object>> compositeKey = personRecord ->
Arrays.asList(personRecord.getName(), personRecord.getCode()); Map<Object, List<Product>> map =
products.stream().collect(Collectors.groupingBy(compositeKey));

感谢阅读!