Java 8 flatMapの例
Java 8では、Streamは次のようなさまざまなデータ型を保持できます。
StreamStream > Stream > Stream
>
ただし、Stream操作(filter、sum、distinct…)とコレクターはそれをサポートしていないため、次の変換を行うにはflatMap()が必要です。
Stream-> flatMap -> Stream Stream > -> flatMap -> Stream Stream > -> flatMap -> Stream
Stream > -> flatMap -> Stream
flatMap()
のしくみ:
{ {1,2}, {3,4}, {5,6} } -> flatMap -> {1,2,3,4,5,6} { {'a','b'}, {'c','d'}, {'e','f'} } -> flatMap -> {'a','b','c','d','e','f'}
1. ストリーム+ String [] + flatMap
1.1 The below example will print an empty result, because filter()
has no idea how to filter a stream of String[]
.
TestExample1.java
package com.example.java8; import java.util.Arrays; import java.util.stream.Stream; public class TestExample1 { public static void main(String[] args) { String[][] data = new String[][]{{"a", "b"}, {"c", "d"}, {"e", "f"}}; //StreamStream temp = Arrays.stream(data); //filter a stream of string[], and return a string[]? Stream stream = temp.filter(x -> "a".equals(x.toString())); stream.forEach(System.out::println); } }
出力
//empty...
1.2 In above example, we should use flatMap()
to convert Stream<String[]>
to Stream<String>
.
TestExample1.java
package com.example.java8; import java.util.Arrays; import java.util.stream.Stream; public class TestExample1 { public static void main(String[] args) { String[][] data = new String[][]{{"a", "b"}, {"c", "d"}, {"e", "f"}}; //StreamStream temp = Arrays.stream(data); //Stream , GOOD! Stream stringStream = temp.flatMap(x -> Arrays.stream(x)); Stream stream = stringStream.filter(x -> "a".equals(x.toString())); stream.forEach(System.out::println); /*Stream stream = Arrays.stream(data) .flatMap(x -> Arrays.stream(x)) .filter(x -> "a".equals(x.toString()));*/ } }
出力
a
2. ストリーム+セット+ flatMap
2.1 A student POJO.
Student.java
package com.example.java8; import java.util.HashSet; import java.util.Set; public class Student { private String name; private Setbook; public void addBook(String book) { if (this.book == null) { this.book = new HashSet<>(); } this.book.add(book); } //getters and setters }
2.2 flatMap()
and Set
example.
TestExample2.java
package com.example.java8; import java.util.ArrayList; import java.util.List; import java.util.stream.Collectors; public class TestExample2 { public static void main(String[] args) { Student obj1 = new Student(); obj1.setName("example"); obj1.addBook("Java 8 in Action"); obj1.addBook("Spring Boot in Action"); obj1.addBook("Effective Java (2nd Edition)"); Student obj2 = new Student(); obj2.setName("zilap"); obj2.addBook("Learning Python, 5th Edition"); obj2.addBook("Effective Java (2nd Edition)"); Listlist = new ArrayList<>(); list.add(obj1); list.add(obj2); List collect = list.stream() .map(x -> x.getBook()) //Stream > .flatMap(x -> x.stream()) //Stream .distinct() .collect(Collectors.toList()); collect.forEach(x -> System.out.println(x)); } }
出力
Spring Boot in Action Effective Java (2nd Edition) Java 8 in Action Learning Python, 5th Edition
flatMap(x -> x.stream())
は、Setオブジェクトのストリームを収集する方法がわからないため、Collectors.toList()
はコンパイラエラーを要求するコメントを試してください。
3. ストリーム+プリミティブ+ flatMapToInt
3.1 For primitive type, you can use flatMapToInt
.
TestExample3.java
package com.example.java8; import java.util.Arrays; import java.util.stream.IntStream; import java.util.stream.Stream; public class TestExample3 { public static void main(String[] args) { int[] intArray = {1, 2, 3, 4, 5, 6}; //1. StreamStream streamArray = Stream.of(intArray); //2. Stream -> flatMap -> IntStream IntStream intStream = streamArray.flatMapToInt(x -> Arrays.stream(x)); intStream.forEach(x -> System.out.println(x)); } }
出力
1 2 3 4 5 6