Пример Java 8 flatMap
В Java 8 Stream может содержать разные типы данных, например:
StreamStream > Stream > Stream
>
Но операции Stream (фильтр, сумма, отдельные…) и сборщики не поддерживают его, поэтому нам нужен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. Stream + 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. Stream + Set + 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())
,Collectors.toList()
будет запрашивать ошибку компилятора, потому что он не знает, как собрать поток объекта Set.
3. Stream + Primitive + 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