Flink中自定义Source和Sink的使用

原创

已于 2024-11-10 20:31:22 修改 · 726 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#flink

于 2024-11-10 17:17:09 首次发布

只要自定一个Source类实现SourceFunction接口，一个Sink类实现SinkFunction接口，就能正常使用自定义的Source和Sink，或者直接extends继承RichSourceFunction和RichSinkFunction,RichSinkFunction:多个open和close方法

1、自定义Source

import org.apache.flink.streaming.api.datastream.DataStream;
import org.apache.flink.streaming.api.datastream.DataStreamSource;
import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment;
import org.apache.flink.streaming.api.functions.source.SourceFunction;

public class Demo3SourceFunction {
    public static void main(String[] args) throws Exception {
        StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
        //使用自定义source
        DataStream<Integer> myDS = env.addSource(new MySource());
        myDS.print();

        env.execute();
    }
}

//自定义source
//实现SourceFunction接口
class MySource implements SourceFunctio