org.apache.flink.connector.testframe.source

Class FromElementsSource<OUT>

java.lang.Object

org.apache.flink.connector.testframe.source.FromElementsSource<OUT>

All Implemented Interfaces:

Serializable, Source<OUT,FromElementsSplit,NoOpEnumState>, SourceReaderFactory<OUT,FromElementsSplit>

public class FromElementsSource<OUT>
extends Object
implements Source<OUT,FromElementsSplit,NoOpEnumState>

A Source implementation that reads data from a list and stops reading at the fixed position. The source will wait until the checkpoint or savepoint triggered, the source is useful for connector tests.

Note: This parallelism of source must be 1.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
FromElementsSource(Boundedness boundedness, List<OUT> elements, Integer emittedElementsNum)
FromElementsSource(List<OUT> elements)

Method Summary

Modifier and Type	Method and Description
SplitEnumerator<FromElementsSplit,NoOpEnumState>	createEnumerator(SplitEnumeratorContext<FromElementsSplit> enumContext) Creates a new SplitEnumerator for this source, starting a new input.
SourceReader<OUT,FromElementsSplit>	createReader(SourceReaderContext readerContext) Creates a new reader to read data from the splits it gets assigned.
Boundedness	getBoundedness() Get the boundedness of this source.
SimpleVersionedSerializer<NoOpEnumState>	getEnumeratorCheckpointSerializer() Creates the serializer for the SplitEnumerator checkpoint.
SimpleVersionedSerializer<FromElementsSplit>	getSplitSerializer() Creates a serializer for the source splits.
SplitEnumerator<FromElementsSplit,NoOpEnumState>	restoreEnumerator(SplitEnumeratorContext<FromElementsSplit> enumContext, NoOpEnumState checkpoint) Restores an enumerator from a checkpoint.

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

FromElementsSource

public FromElementsSource(List<OUT> elements)

FromElementsSource

public FromElementsSource(Boundedness boundedness,
                          List<OUT> elements,
                          Integer emittedElementsNum)

Method Detail

getBoundedness

public Boundedness getBoundedness()

Description copied from interface: Source

Get the boundedness of this source.

Specified by:

getBoundedness in interface Source<OUT,FromElementsSplit,NoOpEnumState>

Returns:

the boundedness of this source.

createReader

public SourceReader<OUT,FromElementsSplit> createReader(SourceReaderContext readerContext)
                                                 throws Exception

Description copied from interface: SourceReaderFactory

Creates a new reader to read data from the splits it gets assigned. The reader starts fresh and does not have any state to resume.

Specified by:

createReader in interface SourceReaderFactory<OUT,FromElementsSplit>

Parameters:

readerContext - The context for the source reader.

Returns:

A new SourceReader.

Throws:

Exception - The implementor is free to forward all exceptions directly. Exceptions thrown from this method cause task failure/recovery.

createEnumerator

public SplitEnumerator<FromElementsSplit,NoOpEnumState> createEnumerator(SplitEnumeratorContext<FromElementsSplit> enumContext)
                                                                  throws Exception

Description copied from interface: Source

Creates a new SplitEnumerator for this source, starting a new input.

Specified by:

createEnumerator in interface Source<OUT,FromElementsSplit,NoOpEnumState>

Parameters:

enumContext - The context for the split enumerator.

Returns:

A new SplitEnumerator.

Throws:

Exception - The implementor is free to forward all exceptions directly. Exceptions thrown from this method cause JobManager failure/recovery.

restoreEnumerator

public SplitEnumerator<FromElementsSplit,NoOpEnumState> restoreEnumerator(SplitEnumeratorContext<FromElementsSplit> enumContext,
                                                                          NoOpEnumState checkpoint)
                                                                   throws Exception

Description copied from interface: Source

Restores an enumerator from a checkpoint.

Specified by:

restoreEnumerator in interface Source<OUT,FromElementsSplit,NoOpEnumState>

Parameters:

enumContext - The context for the restored split enumerator.

checkpoint - The checkpoint to restore the SplitEnumerator from.

Returns:

A SplitEnumerator restored from the given checkpoint.

Throws:

Exception - The implementor is free to forward all exceptions directly. Exceptions thrown from this method cause JobManager failure/recovery.

getSplitSerializer

public SimpleVersionedSerializer<FromElementsSplit> getSplitSerializer()

Description copied from interface: Source

Creates a serializer for the source splits. Splits are serialized when sending them from enumerator to reader, and when checkpointing the reader's current state.

Specified by:

getSplitSerializer in interface Source<OUT,FromElementsSplit,NoOpEnumState>

Returns:

The serializer for the split type.

getEnumeratorCheckpointSerializer

public SimpleVersionedSerializer<NoOpEnumState> getEnumeratorCheckpointSerializer()

Description copied from interface: Source

Creates the serializer for the SplitEnumerator checkpoint. The serializer is used for the result of the SplitEnumerator.snapshotState(long) method.

Specified by:

getEnumeratorCheckpointSerializer in interface Source<OUT,FromElementsSplit,NoOpEnumState>

Returns:

The serializer for the SplitEnumerator checkpoint.