Search in sources :

Example 1 with FileSink

use of org.apache.flink.connector.file.sink.FileSink in project flink by apache.

the class WindowWordCount method main.

// *************************************************************************
// PROGRAM
// *************************************************************************
public static void main(String[] args) throws Exception {
    final CLI params = CLI.fromArgs(args);
    // Create the execution environment. This is the main entrypoint
    // to building a Flink application.
    final StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
    // Apache Flink’s unified approach to stream and batch processing means that a DataStream
    // application executed over bounded input will produce the same final results regardless
    // of the configured execution mode. It is important to note what final means here: a job
    // executing in STREAMING mode might produce incremental updates (think upserts in
    // a database) while a BATCH job would only produce one final result at the end. The final
    // result will be the same if interpreted correctly, but getting there can be different.
    // 
    // The “classic” execution behavior of the DataStream API is called STREAMING execution
    // mode. Applications should use streaming execution for unbounded jobs that require
    // continuous incremental processing and are expected to stay online indefinitely.
    // 
    // By enabling BATCH execution, we allow Flink to apply additional optimizations that we
    // can only do when we know that our input is bounded. For example, different
    // join/aggregation strategies can be used, in addition to a different shuffle
    // implementation that allows more efficient task scheduling and failure recovery behavior.
    // 
    // By setting the runtime mode to AUTOMATIC, Flink will choose BATCH  if all sources
    // are bounded and otherwise STREAMING.
    env.setRuntimeMode(params.getExecutionMode());
    // This optional step makes the input parameters
    // available in the Flink UI.
    env.getConfig().setGlobalJobParameters(params);
    DataStream<String> text;
    if (params.getInputs().isPresent()) {
        // Create a new file source that will read files from a given set of directories.
        // Each file will be processed as plain text and split based on newlines.
        FileSource.FileSourceBuilder<String> builder = FileSource.forRecordStreamFormat(new TextLineInputFormat(), params.getInputs().get());
        // If a discovery interval is provided, the source will
        // continuously watch the given directories for new files.
        params.getDiscoveryInterval().ifPresent(builder::monitorContinuously);
        text = env.fromSource(builder.build(), WatermarkStrategy.noWatermarks(), "file-input");
    } else {
        text = env.fromElements(WordCountData.WORDS).name("in-memory-input");
    }
    int windowSize = params.getInt("window").orElse(250);
    int slideSize = params.getInt("slide").orElse(150);
    DataStream<Tuple2<String, Integer>> counts = // will output each words as a (2-tuple) containing (word, 1)
    text.flatMap(new WordCount.Tokenizer()).name("tokenizer").keyBy(value -> value.f0).countWindow(windowSize, slideSize).sum(1).name("counter");
    if (params.getOutput().isPresent()) {
        // Given an output directory, Flink will write the results to a file
        // using a simple string encoding. In a production environment, this might
        // be something more structured like CSV, Avro, JSON, or Parquet.
        counts.sinkTo(FileSink.<Tuple2<String, Integer>>forRowFormat(params.getOutput().get(), new SimpleStringEncoder<>()).withRollingPolicy(DefaultRollingPolicy.builder().withMaxPartSize(MemorySize.ofMebiBytes(1)).withRolloverInterval(Duration.ofSeconds(10)).build()).build()).name("file-sink");
    } else {
        counts.print().name("print-sink");
    }
    // Apache Flink applications are composed lazily. Calling execute
    // submits the Job and begins processing.
    env.execute("WindowWordCount");
}
Also used : Tuple2(org.apache.flink.api.java.tuple.Tuple2) WordCount(org.apache.flink.streaming.examples.wordcount.WordCount) WatermarkStrategy(org.apache.flink.api.common.eventtime.WatermarkStrategy) FileSink(org.apache.flink.connector.file.sink.FileSink) MemorySize(org.apache.flink.configuration.MemorySize) FileSource(org.apache.flink.connector.file.src.FileSource) DataStream(org.apache.flink.streaming.api.datastream.DataStream) TextLineInputFormat(org.apache.flink.connector.file.src.reader.TextLineInputFormat) SimpleStringEncoder(org.apache.flink.api.common.serialization.SimpleStringEncoder) DefaultRollingPolicy(org.apache.flink.streaming.api.functions.sink.filesystem.rollingpolicies.DefaultRollingPolicy) Duration(java.time.Duration) CLI(org.apache.flink.streaming.examples.wordcount.util.CLI) StreamExecutionEnvironment(org.apache.flink.streaming.api.environment.StreamExecutionEnvironment) WordCountData(org.apache.flink.streaming.examples.wordcount.util.WordCountData) CLI(org.apache.flink.streaming.examples.wordcount.util.CLI) TextLineInputFormat(org.apache.flink.connector.file.src.reader.TextLineInputFormat) FileSource(org.apache.flink.connector.file.src.FileSource) Tuple2(org.apache.flink.api.java.tuple.Tuple2) StreamExecutionEnvironment(org.apache.flink.streaming.api.environment.StreamExecutionEnvironment)

Aggregations

Duration (java.time.Duration)1 WatermarkStrategy (org.apache.flink.api.common.eventtime.WatermarkStrategy)1 SimpleStringEncoder (org.apache.flink.api.common.serialization.SimpleStringEncoder)1 Tuple2 (org.apache.flink.api.java.tuple.Tuple2)1 MemorySize (org.apache.flink.configuration.MemorySize)1 FileSink (org.apache.flink.connector.file.sink.FileSink)1 FileSource (org.apache.flink.connector.file.src.FileSource)1 TextLineInputFormat (org.apache.flink.connector.file.src.reader.TextLineInputFormat)1 DataStream (org.apache.flink.streaming.api.datastream.DataStream)1 StreamExecutionEnvironment (org.apache.flink.streaming.api.environment.StreamExecutionEnvironment)1 DefaultRollingPolicy (org.apache.flink.streaming.api.functions.sink.filesystem.rollingpolicies.DefaultRollingPolicy)1 WordCount (org.apache.flink.streaming.examples.wordcount.WordCount)1 CLI (org.apache.flink.streaming.examples.wordcount.util.CLI)1 WordCountData (org.apache.flink.streaming.examples.wordcount.util.WordCountData)1