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

2020-06-15 06:46:27 浏览数 (1)

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

需求

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

解决方案

代码语言:javascript复制
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");
    }
}

执行效果

代码语言:javascript复制
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个。

代码地址

代码语言:javascript复制
https://github.com/chaoxxx/learn-flink-stream-api/blob/master/src/main/java/fun/cosmozhu/session3/StreamTest.java

作者:cosmozhu --90后的老父亲,专注于保护地球的程序员

个人网站:https://www.cosmozhu.fun

欢迎转载,转载时请注明出处。

相关文章

  • Flink-1.9流计算开发:十六、intervalJoin函数
  • Flink-1.9流计算开发:十五、join函数
  • Flink-1.9流计算开发:十四、union函数
  • Flink-1.9流计算开发:十三、min、minBy、max、maxBy函数
  • Flink-1.9流计算开发:十二、apply函数

0 人点赞