Examples with SplitEnumeratorContext org.apache.flink.api.connector.source.SplitEnumeratorContext used on opensource projects

Example 1 with SplitEnumeratorContext

use of org.apache.flink.api.connector.source.SplitEnumeratorContext in project flink by apache.

the class AbstractFileSource method createSplitEnumerator.

// ------------------------------------------------------------------------
// helpers
// ------------------------------------------------------------------------
private SplitEnumerator<SplitT, PendingSplitsCheckpoint<SplitT>> createSplitEnumerator(SplitEnumeratorContext<SplitT> context, FileEnumerator enumerator, Collection<FileSourceSplit> splits, @Nullable Collection<Path> alreadyProcessedPaths) {
    // cast this to a collection of FileSourceSplit because the enumerator code work
    // non-generically just on that base split type
    @SuppressWarnings("unchecked") final SplitEnumeratorContext<FileSourceSplit> fileSplitContext = (SplitEnumeratorContext<FileSourceSplit>) context;
    final FileSplitAssigner splitAssigner = assignerFactory.create(splits);
    if (continuousEnumerationSettings == null) {
        // bounded case
        return castGeneric(new StaticFileSplitEnumerator(fileSplitContext, splitAssigner));
    } else {
        // unbounded case
        if (alreadyProcessedPaths == null) {
            alreadyProcessedPaths = splitsToPaths(splits);
        }
        return castGeneric(new ContinuousFileSplitEnumerator(fileSplitContext, enumerator, splitAssigner, inputPaths, alreadyProcessedPaths, continuousEnumerationSettings.getDiscoveryInterval().toMillis()));
    }
}