迹忆客 专注技术分享

为了理解 flatMap()方法，首先需要理解Java 8的 map() 函数。map() 函数声明在 java.util.stream.Stream 类中，用于将一个Stream转换为另一个Stream， 例如，它可以将整数流转换为另一个整数流，其中每个元素都是源流中相应元素的平方。 在 map() 操作中，将一个函数应用于源流的每个元素，并将返回值插入到返回给调用者的新流中。 这里要注意的关键点是 map() 操作使用的函数返回单个值。

现在，如果 map 操作使用一个函数，而不是返回单个值，而是返回一个值流，就像当您给出一个数字并返回该数字的所有质因数时，那么您就有了一个整数流。 这就是 flatmap 函数的帮助所在。

flatMap() 方法可用于将该流展平为整数流。 例如，假设我们有一个数字列表，例如 [21, 23, 42]，我们调用 getPrimeFactors() 方法以及 map() 操作来转换此流。

结果将是 [[3,7],[23],[2,3,7]]。 如果你想将这个包含流的流扁平化为值流，你可以使用最终返回 [3,7,2,3,2,3,7] 的 flatMap()。

简而言之，flatMap() 函数用于将 Stream 的 Stream 转换为值列表。

flatMap 在 Java 8 中是如何工作的？

如果我们还没有全面了解并且难以理解 flatMap 函数究竟是如何工作的，那么这里有一个图表，它可以帮助我们理解。 据说一张图片胜过一千个单词，这是真的，下图非常容易地解释了 flatMap 在 Java 8 中的工作原理：

你可以看到有两个列表，一个用红色表示，另一个用绿色表示，它们都包含三个元素。 一旦你对它们应用了 flatMap ，你就可以得到另一个包含它们元素的列表。

我们可以看到，在第二个管道中，我们有 3 个来自绿色列表的元素，分别用 1、2 和 3 表示，另外三个来自红色列表的元素，分别用 4、5 和 6 表示。

Java 8 FlatMap 示例 - Java 中流的平面流

这是一个示例 Java 程序，用于演示如何在 Java 8 中使用 flatMap() 函数。正如我们告诉大家的，我们可以使用 flatMap() 将值流的流扁平化为值流。

在我们的示例中，我们有一个字符串列表流，通过使用 flatMap()，我们将其转换为一个字符串流，以获取参加世界杯的完整球员列表。

这是我们的 Java 程序，用于演示如何在 Java 8 中使用 Stream.flatMap()。

package test;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class Test {
    public static void main(String args[]) {
        List<String> teamIndia = Arrays.asList("Virat", "Dhoni", "Jadeja");
        List<String> teamAustralia = Arrays.asList("Warner", "Watson", "Smith");
        List<String> teamEngland = Arrays.asList("Alex", "Bell", "Broad");
        List<String> teamNewZeland = Arrays.asList("Kane", "Nathan", "Vettori");
        List<String> teamSouthAfrica = Arrays.asList("AB", "Amla", "Faf");
        List<String> teamWestIndies = Arrays.asList("Sammy", "Gayle", "Narine");
        List<String> teamSriLanka = Arrays.asList("Mahela", "Sanga", "Dilshan");
        List<String> teamPakistan = Arrays.asList("Misbah", "Afridi", "Shehzad");
        List<List<String>> playersInWorldCup2016 = new ArrayList<>();
        playersInWorldCup2016.add(teamIndia);
        playersInWorldCup2016.add(teamAustralia);
        playersInWorldCup2016.add(teamEngland);
        playersInWorldCup2016.add(teamNewZeland);
        playersInWorldCup2016.add(teamSouthAfrica);
        playersInWorldCup2016.add(teamWestIndies);
        playersInWorldCup2016.add(teamSriLanka);
        playersInWorldCup2016.add(teamPakistan);

        List<String> listOfAllPlayers = new ArrayList<>();
        for (List<String> team : playersInWorldCup2016) {
            for (String name : team) {
                listOfAllPlayers.add(name);
            }
        }
        System.out.println("Players playing in world cup 2016");
        System.out.println(listOfAllPlayers);
        // 现在让我们使用 FlatMap 在 Java 8 中执行此操作
        List<String> flatMapList = playersInWorldCup2016.stream().flatMap(pList -> pList.stream()).collect(Collectors.toList());
        System.out.println("List of all Players using Java 8");
        System.out.println(flatMapList);
    }
}

大家可以看到最终列表包含每个列表中的所有元素。 所以 flatMap() 可以很好地将字符串列表流扁平化为字符串流，这就是 Java 8 中 flatMap 操作的真正威力。

这就是关于如何在 Java 8 中使用 flatMap() 函数的全部内容。只需记住 Java 8 中 map() 和 flatMap() 之间的区别 以及何时使用 map() 与 flatMap()。 如果我们使用在 map() 操作中返回值列表的函数，我们将获得流式流，通过使用 flatMap，我们可以将其转换为值的流。

简而言之，我们可以使用 flatMap() 将几个小的值列表组合成一个大的值列表。 它被称为 flatMap() 是因为它使 Stream 扁平化。