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));