Java 맵 스트림 사용
1. 기본
Stream은 Collection에서 쉽게 얻을 수 있는 요소의 시퀀스이다.
맵은 시퀀스 없이 키에서 값으로 매핑되는 다른 구조를 가진다. 그러나 이것이 Map 구조를 다른 시퀀스로 변환할 수 없다는 의미는 아니다. 그러면 Stream API를 사용하여 자연스럽게 작업할 수 있다.
맵에서 다른 Collection을 얻는 방법을 살펴보고 Stream으로 피벗할 수 있다.
1
Map<String, Integer> someMap = new HashMap<>();
키-값 쌍 세트를 얻을 수 있다.
1
Set<Map.Entry<String, Integer>> entries = someMap.entrySet();
맵과 관련된 키 세트를 얻을 수도 있다.
1
Set<String> keySet = someMap.keySet();
또는 값 세트로 직접 작업할 수 있다.
1
Collection<Integer> values = someMap.values();
이들 각각은 컬렉션에서 스트림을 가져와 컬렉션을 처리할 수 있는 진입점을 제공한다.
1
2
3
Stream<Map.Entry<String, Integer>> entriesStream = entries.stream();
Stream<Integer> valuesStream = values.stream();
Stream<String> keysStream = keySet.stream();
2. 스트림을 사용하여 맵의 키 얻기
1) 입력 데이터
맵이 있다고 가정한다.
1
2
3
4
5
Map<String, String> books = new HashMap<>();
books.put(
"978-0201633610", "Design patterns : elements of reusable object-oriented software");
books.put("978-1617291999", "Java 8 in Action: Lambdas, Streams, and functional-style programming");
books.put("978-0134685991", "Effective Java");
“Effective Java”라는 책의 ISBN을 찾는 데 관심이 있다.
2) 일치 검색
책 제목이 맵에 존재할 수 없으므로 연결된 ISBN이 없음을 나타낼 수 있기를 원한다. Optional을 사용하여 다음과 같이 표현할 수 있다.
아래 예에서 해당 제목과 일치하는 책의 키에 관심이 있다고 가정한다.
1
2
3
4
5
6
Optional<String> optionalIsbn = books.entrySet().stream()
.filter(e -> "Effective Java".equals(e.getValue()))
.map(Map.Entry::getKey)
.findFirst();
assertEquals("978-0134685991", optionalIsbn.get());
맵에서 entrySet을 얻는다.
제목이 “Effective Java”인 항목만 찾기를 원하므로 첫 번째 중간 작업은 필터가 된다.
우리는 전체 맵 항목이 아니라 각 항목의 키에 관심이 있습니다. 따라서 다음 연결 중간 작업은 바로 그 작업을 수행합니다. 이것은 우리가 찾고 있는 제목과 일치하는 항목에 대한 키만 포함하는 출력으로 새 스트림을 생성하는 맵 작업입니다.
하나의 결과만 원하기 때문에 findFirst() 터미널 작업을 적용할 수 있다. 이 작업은 Stream의 초기 값을 선택적 개체로 제공한다.
제목이 존재하지 않는 경우이다.
1
2
3
4
5
Optional<String> optionalIsbn = books.entrySet().stream()
.filter(e -> "Non Existent Title".equals(e.getValue()))
.map(Map.Entry::getKey).findFirst();
assertEquals(false, optionalIsbn.isPresent());
3) 여러 결과 검색
하나가 아닌 여러 결과를 반환하는 방법이다.
여러 결과가 반환되도록 하려면 맵에 다음 책을 추가한다.
1
books.put("978-0321356680", "Effective Java: Second Edition");
이제 “Effective Java”로 시작하는 모든 책을 찾으면 다음과 같은 결과가 하나 이상 표시된다.
1
2
3
4
5
6
7
List<String> isbnCodes = books.entrySet().stream()
.filter(e -> e.getValue().startsWith("Effective Java"))
.map(Map.Entry::getKey)
.collect(Collectors.toList());
assertTrue(isbnCodes.contains("978-0321356680"));
assertTrue(isbnCodes.contains("978-0134685991"));
이 경우 수행한 작업은 필터 조건을 대체하여 맵의 값이 String 동등성을 비교하는 대신 “Effective Java”로 시작하는지 확인하는 것이다.
이번에는 첫 번째를 선택 하는 대신 결과를 수집 하고 일치 항목을 List에 넣는다.
3. 스트림을 사용하여 맵의 값 가져오기
제목을 기반으로 ISBN을 얻는 대신 ISBN을 기반으로 제목을 얻으려고 한다.
원래 맵을 사용한다. ISBN이 “978-0”으로 시작하는 제목을 찾고 싶다.
1
2
3
4
5
6
7
8
9
List<String> titles = books.entrySet().stream()
.filter(e -> e.getKey().startsWith("978-0"))
.map(Map.Entry::getValue)
.collect(Collectors.toList());
assertEquals(2, titles.size());
assertTrue(titles.contains(
"Design patterns : elements of reusable object-oriented software"));
assertTrue(titles.contains("Effective Java"));
이 솔루션은 이전 세트의 솔루션과 유사하다. 항목 집합을 스트리밍한 다음 필터링, 매핑 및 수집한다.
이전과 마찬가지로 첫 번째 일치 항목만 반환하려는 경우 map 메서드 다음에 List에 모든 결과를 수집하는 대신 findFirst() 메서드를 호출할 수 있다.