Hybrid Source

HybridSource is a source that contains a list of concrete sources. It solves the problem of sequentially reading input from heterogeneous sources to produce a single input stream.

For example, a bootstrap use case may need to read several days worth of bounded input from S3 before continuing with the latest unbounded input from Kafka. HybridSource switches from FileSource to KafkaSource when the bounded file input finishes without interrupting the application.

Prior to HybridSource, it was necessary to create a topology with multiple sources and define a switching mechanism in user land, which leads to operational complexity and inefficiency.

With HybridSource the multiple sources appear as a single source in the Flink job graph and from DataStream API perspective.

For more background see FLIP-150

To use the connector, add the flink-connector-base dependency to your project:

  1. <dependency>
  2. <groupId>org.apache.flink</groupId>
  3. <artifactId>flink-connector-base</artifactId>
  4. <version>1.20.0</version>
  5. </dependency>

Copied to clipboard!

(Typically comes as transitive dependency with concrete sources.)

Start position for next source

To arrange multiple sources in a HybridSource, all sources except the last one need to be bounded. Therefore, the sources typically need to be assigned a start and end position. The last source may be bounded in which case the HybridSource is bounded and unbounded otherwise. Details depend on the specific source and the external storage systems.

Here we cover the most basic and then a more complex scenario, following the File/Kafka example.

Fixed start position at graph construction time

Example: Read till pre-determined switch time from files and then continue reading from Kafka. Each source covers an upfront known range and therefore the contained sources can be created upfront as if they were used directly:

Java

  1. long switchTimestamp = ...; // derive from file input paths
  2. FileSource<String> fileSource =
  3. FileSource.forRecordStreamFormat(new TextLineInputFormat(), Path.fromLocalFile(testDir)).build();
  4. KafkaSource<String> kafkaSource =
  5. KafkaSource.<String>builder()
  6. .setStartingOffsets(OffsetsInitializer.timestamp(switchTimestamp + 1))
  7. .build();
  8. HybridSource<String> hybridSource =
  9. HybridSource.builder(fileSource)
  10. .addSource(kafkaSource)
  11. .build();

Python

  1. switch_timestamp = ... # derive from file input paths
  2. file_source = FileSource \
  3. .for_record_stream_format(StreamFormat.text_line_format(), test_dir) \
  4. .build()
  5. kafka_source = KafkaSource \
  6. .builder() \
  7. .set_bootstrap_servers('localhost:9092') \
  8. .set_group_id('MY_GROUP') \
  9. .set_topics('quickstart-events') \
  10. .set_value_only_deserializer(SimpleStringSchema()) \
  11. .set_starting_offsets(KafkaOffsetsInitializer.timestamp(switch_timestamp)) \
  12. .build()
  13. hybrid_source = HybridSource.builder(file_source).add_source(kafka_source).build()

Dynamic start position at switch time

Example: File source reads a very large backlog, taking potentially longer than retention available for next source. Switch needs to occur at “current time - X”. This requires the start time for the next source to be set at switch time. Here we require transfer of end position from the previous file enumerator for deferred construction of KafkaSource by implementing SourceFactory.

Note that enumerators need to support getting the end timestamp. This may currently require a source customization. Adding support for dynamic end position to FileSource is tracked in FLINK-23633.

Java

  1. FileSource<String> fileSource = CustomFileSource.readTillOneDayFromLatest();
  2. HybridSource<String> hybridSource =
  3. HybridSource.<String, CustomFileSplitEnumerator>builder(fileSource)
  4. .addSource(
  5. switchContext -> {
  6. CustomFileSplitEnumerator previousEnumerator =
  7. switchContext.getPreviousEnumerator();
  8. // how to get timestamp depends on specific enumerator
  9. long switchTimestamp = previousEnumerator.getEndTimestamp();
  10. KafkaSource<String> kafkaSource =
  11. KafkaSource.<String>builder()
  12. .setStartingOffsets(OffsetsInitializer.timestamp(switchTimestamp + 1))
  13. .build();
  14. return kafkaSource;
  15. },
  16. Boundedness.CONTINUOUS_UNBOUNDED)
  17. .build();

Python

Still not supported in Python API.