本文是 用Java DIY 函数式方法—— map 续篇， 解决如何使用java实现函数式方法-flatmap。
注意
- 不适合对函数式一点基础都没有的读者
- DIY实现不是完美的，仅仅是用实例表达函数式方法的理解
- 这个系列文章不是分析java 8 stream中的方法源码，而是对java 8 stream特性，结合Kotlin， Rxjava之类的理解， 使用纯java的方式实现类似的函数式方法。
- 需要对java 中的泛型以及Collection有了解
- 会用到java 8 lambda表达式
- 要实际代码验证，需要 jdk 1.8
讲解的模式如下：
- 给出某个场景
- 使用 java 8
- 使用DIY 函数实现
那就进入主题吧： 用Java DIY 函数式方法——flatmap
DIY 函数式方法flatmap
作用： T -> Collection<R> 注意，这是DIY实现理解，Kotlin，java 8 stream各有自己的集合表达
/** 需求：
* 给定 1 个 Integer集合[1,2,3,4,5]
* 将该集合转换为String类型集合["1a", "1b", "2a", "2b", "3a", "3b", "4a","4b", "5a", "5b"]
* 思路： [1,2,3,4,5] -> ["1a", "1b"], ["2a", "2b"], ["3a", "3b"], ["4a","4b"], ["5a", "5b"]
* -> ["1a", "1b", "2a", "2b", "3a", "3b", "4a","4b", "5a", "5b"]
*/
分析： map能解决的是 T -> R的变化， 上述需求有点特别，需要将 [1, 2, 3, 4, 5] 其中的每一个都要添加 "a", "b"， 然后把他们整合成一个集合。
要是只能使用map实现，也可以， 可行思路是, 如有好的map思路，请留言！！！
- [1,2,3,4,5] -> ["1a", "2a", "3a", "4a", "5a"]
- [1,2,3,4,5] -> ["1b", "2b", "3b", "4b", "5b"]
- 手动合并，还需要排序达到 ["1a", "1b", "2a", "2b", "3a", "3b", "4a","4b", "5a", "5b"]
1. java 8 stream实现
List<Integer> integerList = Arrays.asList(1,2,3,4,5);
integerList.stream()
.flatMap(new Function<Integer, Stream<String>>() {
@Override
public Stream<String> apply(Integer integer) {
return Arrays.asList(integer + "a", integer + "b").stream();
}
}).forEach( item -> out.print(item + " ") );
说明： 上述一条链式调用，就解决了我们的需求
lamdba表达式，简洁如下：
integerList.stream()
.flatMap(integer -> Arrays.asList(integer + "a", integer + "b").stream())
.forEach(item -> out.print(item + " ") );
2. DIY flatmap
在DIY 之前，需要梳理 flatmap的 核心是什么？
T -> Collection<R>
注意:这是DIY flatmap基于Collection实现，Kotlin， java 8 stream各有自己的转换关系！
所以，需要三个东西： 输入 T， 输出 Collection<R>， 映射关系！
public static <T, R> Collection<? super R> flatMap(Collection<? extends T> collection,
Function<T, Collection<R>> function) {
Collection<? super R> result = new ArrayList<>();//这里仅仅是演示
for(T item: collection){
result.addAll(function.call(item));
}
return result;
}
public interface Function<T, R>{
R call(T item);//T -> R
}
其中： Collection<? super R> 是flatmap返回值类型， Collection<? extends T> 是输入参数的类型
Function<T, Collection<R>> function 是映射关系， T -> Collection<R>
//这里仅仅是演示
Collection<? super R> result = new ArrayList<>();
一直强调 DIY的实现是有局限性的，我这里是在java集合的基础上，而且选用ArrayList作为实际的主体，要是其他数据结构类型，肯定就没法使用， 但是，不影响 讲解flatmap实现的思路！
如上， 使用者，只需要关注 Function 接口的具体实现方法call的设计。
使用方式如下：
List<Integer> integerList = Arrays.asList(1,2,3,4,5);
flatMap(integerList, new Function<Integer, Collection<String>>() {
@Override
public Collection<String> call(Integer item) {
return Arrays.asList(item + "a", item + "b");
}
}).forEach(item -> out.print(item + " ") );
lambda简化：
flatMap(integerList, integer -> Arrays.asList(integer + "a", integer + "b") )
.forEach(item -> out.print(item + " "));
其他实例：将三个IntegerList整合为一个List [1], [2,3], [4,5,6] -> [1,2,3,4,5,6]
Collection<List<Integer>> input = Arrays.asList(
Arrays.asList(1),
Arrays.asList(2,3),
Arrays.asList(4, 5, 6));
flatMap(input, item -> new ArrayList<Integer>(item))
.forEach(out::println);
小结
理解 T -> Collection<R>，就明白了flatmap的原理了，遇到实际的情况，可以考虑用flatmap实现，体会跟传统的不同之处！
代码上传到 csdn 资源下载
喜欢，用实际点赞支持我吧！ 欢迎留言讨论！
