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

Class TestingSplitEnumeratorContext<SplitT extends org.apache.flink.api.connector.source.SourceSplit>

java.lang.Object

org.apache.flink.connector.testutils.source.reader.TestingSplitEnumeratorContext<SplitT>

Type Parameters:

SplitT - The generic type of the splits.

All Implemented Interfaces:

org.apache.flink.api.connector.source.SplitEnumeratorContext<SplitT>

public class TestingSplitEnumeratorContext<SplitT extends org.apache.flink.api.connector.source.SourceSplit>
extends Object
implements org.apache.flink.api.connector.source.SplitEnumeratorContext<SplitT>

A test implementation of the SplitEnumeratorContext, with manual, non-concurrent interaction and intercepting of state.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	TestingSplitEnumeratorContext.SplitAssignmentState<SplitT extends org.apache.flink.api.connector.source.SourceSplit> The state of the split assignment for a subtask.

Constructor Summary

Constructors

Constructor and Description
TestingSplitEnumeratorContext(int parallelism)

Method Summary

Modifier and Type	Method and Description
void	assignSplits(org.apache.flink.api.connector.source.SplitsAssignment<SplitT> newSplitAssignments)
<T> void	callAsync(Callable<T> callable, java.util.function.BiConsumer<T,Throwable> handler)
<T> void	callAsync(Callable<T> callable, java.util.function.BiConsumer<T,Throwable> handler, long initialDelay, long period)
int	currentParallelism()
org.apache.flink.core.testutils.ManuallyTriggeredScheduledExecutorService	getExecutorService()
Map<Integer,List<org.apache.flink.api.connector.source.SourceEvent>>	getSentEvents()
Map<Integer,TestingSplitEnumeratorContext.SplitAssignmentState<SplitT>>	getSplitAssignments()
org.apache.flink.metrics.MetricGroup	metricGroup()
Map<Integer,org.apache.flink.api.connector.source.ReaderInfo>	registeredReaders()
void	registerReader(int subtask, String hostname)
void	runInCoordinatorThread(Runnable runnable)
void	sendEventToSourceReader(int subtaskId, org.apache.flink.api.connector.source.SourceEvent event)
void	signalNoMoreSplits(int subtask)
void	triggerAllActions()

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.flink.api.connector.source.SplitEnumeratorContext

assignSplit

Constructor Detail

TestingSplitEnumeratorContext

public TestingSplitEnumeratorContext(int parallelism)

Method Detail

triggerAllActions

public void triggerAllActions()

getExecutorService

public org.apache.flink.core.testutils.ManuallyTriggeredScheduledExecutorService getExecutorService()

getSplitAssignments

public Map<Integer,TestingSplitEnumeratorContext.SplitAssignmentState<SplitT>> getSplitAssignments()

getSentEvents

public Map<Integer,List<org.apache.flink.api.connector.source.SourceEvent>> getSentEvents()

registerReader

public void registerReader(int subtask,
                           String hostname)

metricGroup

public org.apache.flink.metrics.MetricGroup metricGroup()

Specified by:

metricGroup in interface org.apache.flink.api.connector.source.SplitEnumeratorContext<SplitT extends org.apache.flink.api.connector.source.SourceSplit>

sendEventToSourceReader

public void sendEventToSourceReader(int subtaskId,
                                    org.apache.flink.api.connector.source.SourceEvent event)

Specified by:

sendEventToSourceReader in interface org.apache.flink.api.connector.source.SplitEnumeratorContext<SplitT extends org.apache.flink.api.connector.source.SourceSplit>

currentParallelism

public int currentParallelism()

Specified by:

currentParallelism in interface org.apache.flink.api.connector.source.SplitEnumeratorContext<SplitT extends org.apache.flink.api.connector.source.SourceSplit>

registeredReaders

public Map<Integer,org.apache.flink.api.connector.source.ReaderInfo> registeredReaders()

Specified by:

registeredReaders in interface org.apache.flink.api.connector.source.SplitEnumeratorContext<SplitT extends org.apache.flink.api.connector.source.SourceSplit>

assignSplits

public void assignSplits(org.apache.flink.api.connector.source.SplitsAssignment<SplitT> newSplitAssignments)

Specified by:

assignSplits in interface org.apache.flink.api.connector.source.SplitEnumeratorContext<SplitT extends org.apache.flink.api.connector.source.SourceSplit>

signalNoMoreSplits

public void signalNoMoreSplits(int subtask)

Specified by:

signalNoMoreSplits in interface org.apache.flink.api.connector.source.SplitEnumeratorContext<SplitT extends org.apache.flink.api.connector.source.SourceSplit>

callAsync

public <T> void callAsync(Callable<T> callable,
                          java.util.function.BiConsumer<T,Throwable> handler)

Specified by:

callAsync in interface org.apache.flink.api.connector.source.SplitEnumeratorContext<SplitT extends org.apache.flink.api.connector.source.SourceSplit>

callAsync

public <T> void callAsync(Callable<T> callable,
                          java.util.function.BiConsumer<T,Throwable> handler,
                          long initialDelay,
                          long period)

Specified by:

callAsync in interface org.apache.flink.api.connector.source.SplitEnumeratorContext<SplitT extends org.apache.flink.api.connector.source.SourceSplit>

runInCoordinatorThread

public void runInCoordinatorThread(Runnable runnable)

Specified by:

runInCoordinatorThread in interface org.apache.flink.api.connector.source.SplitEnumeratorContext<SplitT extends org.apache.flink.api.connector.source.SourceSplit>