Flink-1.9流计算开发：三、flatMap函数

Flink-1.9流计算开发：三、flatMap函数

在上一篇文章Flink-1.9流计算开发：二、Map函数文章中我们使用了map方法。在本篇文章中我们将使用flatMap，来验证一下它与map方法的差异。

需求

本篇文章实现一个很简单的逻辑，创建一个数据流{“apple”,“orange”,“banana”,“watermelon”}，将数据流的单词做大写转换，并保留其小写，然后将其打印出来。

解决方案

public class StreamTest {
	private static final Logger LOG = LoggerFactory.getLogger(StreamTest.class);
	public static void main(String[] args) throws Exception {
		final StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
		String[] words = {"apple","orange","banana","watermelon"};
		//创建DataSource
		//根据给定的元素创建一个数据流，元素必须是相同的类型，比如全部为String，或者全部为int。
		DataStreamSource<String> ds = env.fromElements(words);
		//Transformations
		//对DataStream应用一个flatMap转换。对DataStream中的每一个元素都会调用FlatMapFunction接口的具体实现类。flatMap方法可以返回任意个元素，当然也可以什么都不返回。
		SingleOutputStreamOperator<String> flatMap = ds.flatMap(new FlatMapFunction<String, String>() {
			@Override
			public void flatMap(String value, Collector<String> out) throws Exception {
				out.collect(value);
				out.collect(value.toUpperCase());
			}
		});
		//sinks打印出信息
		//给DataStream添加一个Sinks
		flatMap.addSink(new SinkFunction<String>() {
			@Override
			public void invoke(String value) throws Exception {
				LOG.info(value);
			}
		});
		
		env.execute("Flink Streaming Java API Skeleton");
	}
}

执行效果

15:44:46,305 INFO  fun.cosmozhu.session3.StreamTest                              - watermelon
15:44:46,305 INFO  fun.cosmozhu.session3.StreamTest                              - apple
15:44:46,305 INFO  fun.cosmozhu.session3.StreamTest                              - WATERMELON
15:44:46,305 INFO  fun.cosmozhu.session3.StreamTest                              - banana
15:44:46,305 INFO  fun.cosmozhu.session3.StreamTest                              - BANANA
15:44:46,305 INFO  fun.cosmozhu.session3.StreamTest                              - orange
15:44:46,305 INFO  fun.cosmozhu.session3.StreamTest                              - ORANGE
15:44:46,305 INFO  fun.cosmozhu.session3.StreamTest                              - APPLE

小结

对比来看，map函数返回的元素有且只能有一个。而flatMap则可以返回0~N个。

代码地址

https://github.com/chaoxxx/learn-flink-stream-api/blob/master/src/main/java/fun/cosmozhu/session3/StreamTest.java

---

作者：cosmozhu --90后的老父亲，专注于保护地球的程序员
个人网站：http://www.cosmozhu.fun
欢迎转载，转载时请注明出处。