Java 8 Steam API map和flatMap方法使用详解

java 8 stream api 中有两个方法map和flatMap非常实用，应用场景也非常广泛，能极大提升编程效率。下面我们详细介绍一下这两个方法的用法。

map方法

我们来看个示例：把一个整数列表转换成字符串列表，java 8之前常用的实现方法如下

List<Integer> numList = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8);
List<String> strList = new ArrayList<>();
for (int num : numList) {
    strList.add(Integer.toString(num));
}

这种写法比较符合直觉，但略显繁琐。如果用java 8的stream api的map方法则可以把这个过程变的非常简洁

List<String> strList = numList.stream()
        .map(it -> Integer.toString(it))
        .collect(Collectors.toList());

map方法接受一个lambda表达式，这个表达式是一个函数，输入类型是集合元素的类型，输出类型是任意类型

it -> Integer.toString(it)

我们还可以用map实现很多效果，比如转换成符合要求的bool列表

List<Boolean> boolList = numList.stream()
        .map(it -> it > 5 ? true : false)
        .collect(Collectors.toList());

或者转换成某种对象列表

public class Main {
    private static class Klass {
        private int field;
 
        public Klass(int field) {
            this.field = field;
        }
 
        @Override
        public String toString() {
            return "field=" + field;
        }
    }
 
    public static void main(String[] args) {
        List<Integer> numList = Arrays.asList(1, 2, 3, 4, 5, 6, 7, 8);
        List<Klass> objList = numList.stream()
                .map(it -> new Klass(it))
                .collect(Collectors.toList());
    }
}

都可以非常迅速的实现，和那些流水账式的代码告别

List<Klass> objList2 = new ArrayList<>();
for (int num : numList) {
    objList2.add(new Klass(num));
}

flatMap方法

我们把需求扩展下

先定义两个类型

private static class Klass {
    private int field;

    public Klass(int field) {
        this.field = field;
    }

    @Override
    public String toString() {
        return "field=" + field;
    }
}

private static class KlassGroup {
    private List<Klass> group = new ArrayList<>();

    public KlassGroup(Klass... objList) {
        for (Klass item : objList) {
            this.group.add(item);
        }
    }

    public List<Klass> getKlassList() {
        return group;
    }
}

KlassGroup类中定义了一个Klass类的列表

现在我们有一组KlassGroup对象

List<KlassGroup> groupList = Arrays.asList(
        new KlassGroup(new Klass(1), new Klass(2), new Klass(3)),
        new KlassGroup(new Klass(4), new Klass(5), new Klass(6)),
        new KlassGroup(new Klass(7), new Klass(8), new Klass(9)),
        new KlassGroup(new Klass(10))
);

需要将每个KlassGroup对象中的那些Klass类取出来，放到一个ArrayList里面，得到一个List。我们尝试着用map方法来实现

List<List<Klass>> result = groupList.stream()
        .map(it -> it.getKlassList())
        .collect(Collectors.toList());

哈，不成功，我们想要的结果是List，现在得到了 List<List>。当然，我们可以轻而易举的解决这个问题

List<Klass> result2 = new ArrayList<>();
for (KlassGroup group : groupList) {
    for (Klass klass : group.getKlassList()) {
        result2.add(klass);
    }
}

但是这种套了两层for循环的代码太丑陋了。面对这种需求，flatMap可以大展身手了

flatMap测试应用

List<Klass> result3 = groupList.stream()
        .flatMap(it -> it.getKlassList().stream())
        .collect(Collectors.toList());

一行代码就实现了

stream api 的 flatMap方法接受一个lambda表达式函数， 函数的返回值必须也是一个stream类型，flatMap方法最终会把所有返回的stream合并，map方法做不到这一点，如果用map去实现，会变成这样一个东西

List<Stream<Klass>> result3 = groupList.stream()
        .map(it -> it.getKlassList().stream())
        .collect(Collectors.toList());

flatMap的思路在其他语言中也有体现，比如C# Linq中的 SelectMany 方法，F# 中的 List.collect方法都有同样的作用。用函数式编程的说法，他们都实现了 monad，当然，monad这个概念很难是通俗的描述清楚，此文中就不展开了。

java 8 stream api 中各方法可以极大的简化集合操作，带来大幅度的编码效率提升，如果是java 8及以上的版本，一定要优先使用，stream api绝对是java中史诗级的工作效率提升利器。