Package org.apache.flink.connector.testutils.source.reader

Class TestingReaderContext

java.lang.Object

org.apache.flink.connector.testutils.source.reader.TestingReaderContext

All Implemented Interfaces:

SourceReaderContext

public class TestingReaderContext
extends Object
implements SourceReaderContext

A testing implementation of the SourceReaderContext.

Constructor Summary

Constructors

Constructor	Description
TestingReaderContext()
TestingReaderContext(Configuration config, SourceReaderMetricGroup metricGroup)

Method Summary

Modifier and Type	Method	Description
void	clearSentEvents()
int	currentParallelism()	Get the current parallelism of this Source.
Configuration	getConfiguration()	Gets the configuration with which Flink was started.
int	getIndexOfSubtask()
String	getLocalHostName()	Gets the hostname of the machine where this reader is executed.
int	getNumSplitRequests()
List<SourceEvent>	getSentEvents()
UserCodeClassLoader	getUserCodeClassLoader()	Gets the UserCodeClassLoader to load classes that are not in system's classpath, but are part of the jar file of a user job.
SourceReaderMetricGroup	metricGroup()
void	sendSourceEventToCoordinator(SourceEvent sourceEvent)	Send a source event to the source coordinator.
void	sendSplitRequest()	Sends a split request to the source's SplitEnumerator.

Methods inherited from class java.lang.Object

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

Constructor Detail

TestingReaderContext

public TestingReaderContext()

TestingReaderContext

public TestingReaderContext(Configuration config,
                            SourceReaderMetricGroup metricGroup)

Method Detail

metricGroup

public SourceReaderMetricGroup metricGroup()

Specified by:

metricGroup in interface SourceReaderContext

Returns:

The metric group this source belongs to.

getConfiguration

public Configuration getConfiguration()

Description copied from interface: SourceReaderContext

Gets the configuration with which Flink was started.

Specified by:

getConfiguration in interface SourceReaderContext

getLocalHostName

public String getLocalHostName()

Description copied from interface: SourceReaderContext

Gets the hostname of the machine where this reader is executed. This can be used to request splits local to the machine, if needed.

Specified by:

getLocalHostName in interface SourceReaderContext

getIndexOfSubtask

public int getIndexOfSubtask()

Specified by:

getIndexOfSubtask in interface SourceReaderContext

Returns:

The index of this subtask.

sendSplitRequest

public void sendSplitRequest()

Description copied from interface: SourceReaderContext

Sends a split request to the source's SplitEnumerator. This will result in a call to the SplitEnumerator.handleSplitRequest(int, String) method, with this reader's parallel subtask id and the hostname where this reader runs.

Specified by:

sendSplitRequest in interface SourceReaderContext

sendSourceEventToCoordinator

public void sendSourceEventToCoordinator(SourceEvent sourceEvent)

Description copied from interface: SourceReaderContext

Send a source event to the source coordinator.

Specified by:

sendSourceEventToCoordinator in interface SourceReaderContext

Parameters:

sourceEvent - the source event to coordinator.

getUserCodeClassLoader

public UserCodeClassLoader getUserCodeClassLoader()

Description copied from interface: SourceReaderContext

Gets the UserCodeClassLoader to load classes that are not in system's classpath, but are part of the jar file of a user job.

Specified by:

getUserCodeClassLoader in interface SourceReaderContext

See Also:

UserCodeClassLoader

currentParallelism

public int currentParallelism()

Description copied from interface: SourceReaderContext

Get the current parallelism of this Source.

Specified by:

currentParallelism in interface SourceReaderContext

Returns:

the parallelism of the Source.

getNumSplitRequests

public int getNumSplitRequests()

getSentEvents

public List<SourceEvent> getSentEvents()

clearSentEvents

public void clearSentEvents()