zoukankan      html  css  js  c++  java
  • Flink 流处理API之实现UDF函数——更细粒度的控制流

    1 函数类(Function Classes

    Flink暴露了所有udf函数的接口(实现方式为接口或者抽象类)。例如MapFunction, FilterFunction, ProcessFunction等等。

    下面例子实现了FilterFunction接口:

    DataStream<String> flinkTweets = tweets.filter(new FlinkFilter());

     

    public static class FlinkFilter implements FilterFunction<String> {
        @Override
        public boolean filter(String value) throws Exception {
            return value.contains("flink");
        }
    }

    还可以将函数实现成匿名类

    DataStream<String> flinkTweets = tweets.filter(new FilterFunction<String>() {
        @Override
        public boolean filter(String value) throws Exception {
            return value.contains("flink");
        }
    });

    我们filter的字符串"flink"还可以当作参数传进去。

    DataStream<String> tweets = env.readTextFile("INPUT_FILE ");

    DataStream<String> flinkTweets = tweets.filter(new KeyWordFilter("flink"));

     

    public static class KeyWordFilter implements FilterFunction<String> {
        private String keyWord;

        KeyWordFilter(String keyWord) { this.keyWord = keyWord; }

        @Override
        public boolean filter(String value) throws Exception {
            return value.contains(this.keyWord);
        }
    }

    2 匿名函数(Lambda Functions

    DataStream<String> tweets = env.readTextFile("INPUT_FILE");

    DataStream<String> flinkTweets = tweets.filter( tweet -> tweet.contains("flink") );

    3 富函数(Rich Functions)(学习重点)

    “富函数”是DataStream API提供的一个函数类的接口,所有Flink函数类都有其Rich版本。它与常规函数的不同在于,可以获取运行环境的上下文,并拥有一些生命周期方法,所以可以实现更复杂的功能。

    l RichMapFunction

    l RichFlatMapFunction

    l RichFilterFunction

    l …​

    Rich Function有一个生命周期的概念。典型的生命周期方法有

    l open()方法是rich function的初始化方法,当一个算子例如map或者filter被调用之前open()会被调用。

    l close()方法是生命周期中的最后一个调用的方法,做一些清理工作。

    l getRuntimeContext()方法提供了函数的RuntimeContext的一些信息,例如函数执行的并行度,任务的名字,以及state状态

    代码:

     

    public static class MyMapFunction extends RichMapFunction<SensorReading, Tuple2<Integer, String>> {
        @Override
        public Tuple2<Integer, String> map(SensorReading value) throws Exception {
            return new Tuple2<>(getRuntimeContext().getIndexOfThisSubtask(), value.getId());
        }

        @Override
        public void open(Configuration parameters) throws Exception {
            System.out.println("my map open");
            // 以下可以做一些初始化工作,例如建立一个和HDFS的连接
        }

        @Override
        public void close() throws Exception {
            System.out.println("my map close");
            // 以下做一些清理工作,例如断开和HDFS的连接
        }
    }

     

  • 相关阅读:
    大数据学习13_MapReduce计数器&排序和序列化
    大数据学习12_MapReduce分区
    大数据学习11_MapReduce案例实战(单词统计)
    大数据学习10_Mapreduce
    大数据学习09_HDFS3
    周总结8
    《河北省重大技术需求征集系统》10_导入大量数据做测试
    《河北省重大技术需求征集系统》08_需求浏览
    《河北省重大技术需求征集系统》07_综合查询
    《河北省重大技术需求征集系统》06_网络审核
  • 原文地址:https://www.cnblogs.com/whdd/p/14055855.html
Copyright © 2011-2022 走看看