函数式编程-收集器

集合对象收集器

1、toList

 @Test
    public void givenUsers_withMap_thenTransformUsingStream() {
        List<UserDTO> userDTOList = userList.stream().map(user -> UserDTO.builder()
                .username(user.getUsername())
                .enabled(user.isEnabled() ? "可用" : "不可用").build()).collect(Collectors.toList());
        assertEquals(3, userDTOList.size());
    }

2、toSet

 @Test
    public void givenUsers_withToSet_thenSuccess() {
       Set<String> usernames = userList.stream()
               .map(User::getName)
               .collect(toSet());
       assertEquals(2, usernames.size());
    }
    
    @Test
    public void givenUsers_withToCollection_thenSuccess() {
        Comparator<User> byAge = Comparator.comparing(User::getAge);
        TreeSet<User> users = userList.stream()
                .collect(toCollection(() -> new TreeSet<>(byAge)));
        assertEquals(30, users.stream().map(User::getAge).findFirst().orElse(-1));
    }

3、toMap

 @Test
    public void givenUsers_withToMap_thenSuccess() {
        Map<String, User> userMap = userList.stream()
                .collect(toMap(
                        User::getUsername,
                        user -> user
                ));
        assertTrue(userMap.containsKey("lisi"));
        Map<String, User> duplicateMap = Stream.concat(userList.stream(), userList.stream())
                .peek(user -> log.debug("username, {}", user.getUsername()))
                .collect(toMap(
                        User::getUsername,
                        user -> user,
                        (existing, replace) -> existing
                ));
        assertEquals(3, duplicateMap.keySet().size());
        TreeMap<String, User> sortedMap = Stream.concat(userList.stream(), userList.stream())
                .peek(user -> log.debug("username, {}", user.getUsername()))
                .collect(toMap(
                        User::getUsername,
                        user -> user,
                        (existing, replace) -> existing,
                        TreeMap::new
                ));
        assertEquals("lisi", sortedMap.keySet().stream().findFirst().get());
    }