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