集合对象收集器
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());
}