【java_基础深入】Java8 Stream | Collections API 的 分组操作

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

    }