Java8 Stream | Collections API 的 分组操作
初始数据
// 初始数据
List<User> userList = Arrays.asList(
new User("zhangsan", 10),
new User("zhangsan", 12),
new User("lisi", 10),
new User("wangwu", 15),
new User("zhaoliu", 0)
);
1. 普通分组 Collectors.groupingBy
// 使用grouping分组,根据Name分组
Map<String, List<User>> map = userList.stream().collect(Collectors.groupingBy(user -> user.getName()));
log.info("使用grouping分组: \n {} \n", JSON.toJSON(map));
2. 普通分组 Collectors.groupingBy + Collectors.mapping
mapping 用于返回新的对象,即不单单分组,还是用分组后的组内元素返回
/**
* 测试steam分组功能,过滤
* @param args
*/
public static void main(String[] args) {
// 使用grouping和mapping分组
Map<String, List<UserEnhance>> map1 = userList.stream()
.collect(
Collectors.groupingBy(
user -> user.getName(),
Collectors.mapping(p -> new UserEnhance(p.getName(), p.getAge()) ,Collectors.toList())
)
);
log.info("使用grouping和mapping分组: \n {} \n", JSON.toJSON(map1));
// 使用grouping和mapping分组,分组的内容具体到某个属性
Map<String, List<Integer>> map2 = userList.stream()
.collect(
Collectors.groupingBy(
user -> user.getName(),
Collectors.mapping(user -> user.getAge(),Collectors.toList())
)
);
log.info("使用grouping和mapping分组,分组的内容具体到某个属性, \n {} \n", JSON.toJSON(map2));
// 去重后的分组
Map<String, User> map3 = userList.stream()
.filter(p -> Objects.nonNull(p.getAge()) && p.getAge() != 0)
.collect(
Collectors.toMap(
user -> user.getName(),
user -> new User(user.getName(), user.getAge()),
(v1, v2) -> v1.getAge() > v2.getAge() ? v1 : v2)
);
log.info("去重后的分组: \n {} \n", JSON.toJSON(map3));
// 使用map()抽取
List<String> userNameList = userList.stream().map(p -> p.getName()).collect(Collectors.toList());
// 使用mapping抽取后放入新的list
List<String> userNameListWithMapper = new ArrayList<>();
// 另外,map(...)规则可以作为参数传入
Function<User,String> mapper= p -> p.getName();
userNameListWithMapper = userList.stream().map(mapper).collect(Collectors.toList());
log.info("普通map抽取: \n {} \n", JSON.toJSON(userNameList));
log.info("使用外置mapper: \n {} \n", JSON.toJSON(userNameListWithMapper));
}