OUT
- OP
- @Internal public abstract class StreamTask<OUT,OP extends StreamOperator<OUT>> extends AbstractInvokable implements AsyncExceptionHandler
StreamOperator
s which form the
Task's operator chain. Operators that are chained together execute synchronously in the same
thread and hence on the same stream partition. A common case for these chains are successive
map/flatmap/filter tasks.
The task chain contains one "head" operator and multiple chained operators. The StreamTask is specialized for the type of the head operator: one-input and two-input tasks, as well as for sources, iteration heads and iteration tails.
The Task class deals with the setup of the streams read by the head operator, and the streams produced by the operators at the ends of the operator chain. Note that the chain may fork and thus have multiple ends.
The life cycle of the task is set up as follows:
-- setInitialState -> provides state of all operators in the chain
-- invoke()
|
+----> Create basic utils (config, etc) and load the chain of operators
+----> operators.setup()
+----> task specific init()
+----> initialize-operator-states()
+----> open-operators()
+----> run()
+----> close-operators()
+----> dispose-operators()
+----> common cleanup
+----> task specific cleanup()
The StreamTask
has a lock object called lock
. All calls to methods on a StreamOperator
must be synchronized on this lock object to ensure that no methods are called
concurrently.
Modifier and Type | Field and Description |
---|---|
protected CheckpointStorage |
checkpointStorage
Our checkpoint storage.
|
protected StreamConfig |
configuration
The configuration of this streaming task.
|
protected StreamInputProcessor |
inputProcessor
The input processor.
|
protected static org.slf4j.Logger |
LOG
The logger used by the StreamTask and its subclasses.
|
protected MailboxProcessor |
mailboxProcessor |
protected OP |
mainOperator
the main operator that consumes the input streams of this task.
|
protected OperatorChain<OUT,OP> |
operatorChain
The chain of operators executed by this task.
|
protected StateBackend |
stateBackend
Our state backend.
|
protected TimerService |
systemTimerService
In contrast to
timerService we should not register any user timers here. |
protected TimerService |
timerService
The internal
TimerService used to define the current processing time (default =
System.currentTimeMillis() ) and register timers for tasks to be executed in the
future. |
static ThreadGroup |
TRIGGER_THREAD_GROUP
The thread group that holds all trigger timer threads.
|
Modifier | Constructor and Description |
---|---|
protected |
StreamTask(Environment env)
Constructor for initialization, possibly with initial state (recovery / savepoint / etc).
|
protected |
StreamTask(Environment env,
TimerService timerService)
Constructor for initialization, possibly with initial state (recovery / savepoint / etc).
|
protected |
StreamTask(Environment environment,
TimerService timerService,
Thread.UncaughtExceptionHandler uncaughtExceptionHandler) |
protected |
StreamTask(Environment environment,
TimerService timerService,
Thread.UncaughtExceptionHandler uncaughtExceptionHandler,
StreamTaskActionExecutor actionExecutor)
Constructor for initialization, possibly with initial state (recovery / savepoint / etc).
|
protected |
StreamTask(Environment environment,
TimerService timerService,
Thread.UncaughtExceptionHandler uncaughtExceptionHandler,
StreamTaskActionExecutor actionExecutor,
TaskMailbox mailbox) |
Modifier and Type | Method and Description |
---|---|
void |
abortCheckpointOnBarrier(long checkpointId,
CheckpointException cause)
Aborts a checkpoint as the result of receiving possibly some checkpoint barriers, but at
least one
CancelCheckpointMarker . |
protected void |
advanceToEndOfEventTime()
Emits the
MAX_WATERMARK so that all registered timers are fired. |
protected void |
afterInvoke() |
Future<Void> |
cancel()
This method is called when a task is canceled either as a result of a user abort or an
execution failure.
|
protected void |
cancelTask() |
protected void |
cleanup() |
protected void |
cleanUpInvoke() |
static <OUT> RecordWriterDelegate<SerializationDelegate<StreamRecord<OUT>>> |
createRecordWriterDelegate(StreamConfig configuration,
Environment environment) |
StreamTaskStateInitializer |
createStreamTaskStateInitializer() |
protected void |
declineCheckpoint(long checkpointId) |
void |
dispatchOperatorEvent(OperatorID operator,
SerializedValue<OperatorEvent> event) |
protected void |
finalize()
The finalize method shuts down the timer.
|
protected void |
finishTask()
Instructs the task to go through its normal termination routine, i.e.
|
protected long |
getAsyncCheckpointStartDelayNanos() |
ExecutorService |
getAsyncOperationsThreadPool() |
CloseableRegistry |
getCancelables() |
CheckpointStorageWorkerView |
getCheckpointStorage() |
protected CompletableFuture<Void> |
getCompletionFuture() |
StreamConfig |
getConfiguration() |
MailboxExecutorFactory |
getMailboxExecutorFactory() |
String |
getName()
Gets the name of the task, in the form "taskname (2/5)".
|
ProcessingTimeServiceFactory |
getProcessingTimeServiceFactory() |
StreamStatusMaintainer |
getStreamStatusMaintainer() |
void |
handleAsyncException(String message,
Throwable exception)
Handles an exception thrown by another thread (e.g.
|
protected abstract void |
init() |
void |
invoke()
Starts the execution.
|
boolean |
isCanceled() |
boolean |
isFailing() |
boolean |
isMailboxLoopRunning() |
boolean |
isRunning() |
boolean |
isUsingNonBlockingInput() |
Future<Void> |
notifyCheckpointAbortAsync(long checkpointId)
Invoked when a checkpoint has been aborted, i.e., when the checkpoint coordinator has
received a decline message from one task and try to abort the targeted checkpoint by
notification.
|
Future<Void> |
notifyCheckpointCompleteAsync(long checkpointId)
Invoked when a checkpoint has been completed, i.e., when the checkpoint coordinator has
received the notification from all participating tasks.
|
protected void |
processInput(MailboxDefaultAction.Controller controller)
This method implements the default action of the task (e.g.
|
void |
restore()
This method can be called before
AbstractInvokable.invoke() to restore an invokable object for the
last valid state, if it has it. |
void |
runMailboxLoop() |
boolean |
runMailboxStep() |
protected Counter |
setupNumRecordsInCounter(StreamOperator streamOperator) |
String |
toString() |
Future<Boolean> |
triggerCheckpointAsync(CheckpointMetaData checkpointMetaData,
CheckpointOptions checkpointOptions)
This method is called to trigger a checkpoint, asynchronously by the checkpoint coordinator.
|
void |
triggerCheckpointOnBarrier(CheckpointMetaData checkpointMetaData,
CheckpointOptions checkpointOptions,
CheckpointMetricsBuilder checkpointMetrics)
This method is called when a checkpoint is triggered as a result of receiving checkpoint
barriers on all input streams.
|
getCurrentNumberOfSubtasks, getEnvironment, getExecutionConfig, getIndexInSubtaskGroup, getJobConfiguration, getTaskConfiguration, getUserCodeClassLoader, setShouldInterruptOnCancel, shouldInterruptOnCancel
public static final ThreadGroup TRIGGER_THREAD_GROUP
protected static final org.slf4j.Logger LOG
@Nullable protected StreamInputProcessor inputProcessor
init()
method.protected OP extends StreamOperator<OUT> mainOperator
protected OperatorChain<OUT,OP extends StreamOperator<OUT>> operatorChain
protected final StreamConfig configuration
protected final StateBackend stateBackend
protected final CheckpointStorage checkpointStorage
protected final TimerService timerService
TimerService
used to define the current processing time (default =
System.currentTimeMillis()
) and register timers for tasks to be executed in the
future.protected final TimerService systemTimerService
timerService
we should not register any user timers here. It should
be used only for system level timers.protected final MailboxProcessor mailboxProcessor
protected StreamTask(Environment env) throws Exception
env
- The task environment for this task.Exception
protected StreamTask(Environment env, @Nullable TimerService timerService) throws Exception
env
- The task environment for this task.timerService
- Optionally, a specific timer service to use.Exception
protected StreamTask(Environment environment, @Nullable TimerService timerService, Thread.UncaughtExceptionHandler uncaughtExceptionHandler) throws Exception
Exception
protected StreamTask(Environment environment, @Nullable TimerService timerService, Thread.UncaughtExceptionHandler uncaughtExceptionHandler, StreamTaskActionExecutor actionExecutor) throws Exception
This constructor accepts a special TimerService
. By default (and if null is passes
for the timer service) a DefaultTimerService
will be
used.
environment
- The task environment for this task.timerService
- Optionally, a specific timer service to use.uncaughtExceptionHandler
- to handle uncaught exceptions in the async operations thread
poolactionExecutor
- a mean to wrap all actions performed by this task thread. Currently,
only SynchronizedActionExecutor can be used to preserve locking semantics.Exception
protected StreamTask(Environment environment, @Nullable TimerService timerService, Thread.UncaughtExceptionHandler uncaughtExceptionHandler, StreamTaskActionExecutor actionExecutor, TaskMailbox mailbox) throws Exception
Exception
protected void processInput(MailboxDefaultAction.Controller controller) throws Exception
controller
- controller object for collaborative interaction between the action and the
stream task.Exception
- on any problems in the action.protected void advanceToEndOfEventTime() throws Exception
MAX_WATERMARK
so that all registered timers are fired.
This is used by the source task when the job is TERMINATED
. In the case, we want
all the timers registered throughout the pipeline to fire and the related state (e.g.
windows) to be flushed.
For tasks other than the source task, this method does nothing.
Exception
protected void finishTask() throws Exception
StreamOperator.close()
and StreamOperator.dispose()
on its operators.
This is used by the source task to get out of the run-loop when the job is stopped with a savepoint.
For tasks other than the source task, this method does nothing.
Exception
public StreamTaskStateInitializer createStreamTaskStateInitializer()
protected Counter setupNumRecordsInCounter(StreamOperator streamOperator)
public final void restore() throws Exception
AbstractInvokable
AbstractInvokable.invoke()
to restore an invokable object for the
last valid state, if it has it.
Every implementation determinate what should be restored by itself. (nothing happens by default).
restore
in class AbstractInvokable
Exception
- Tasks may forward their exceptions for the TaskManager to handle through
failure/recovery.public final void invoke() throws Exception
AbstractInvokable
Must be overwritten by the concrete task implementation. This method is called by the task manager when the actual execution of the task starts.
All resources should be cleaned up when the method returns. Make sure to guard the code
with try-finally
blocks where necessary.
invoke
in class AbstractInvokable
Exception
- Tasks may forward their exceptions for the TaskManager to handle through
failure/recovery.@VisibleForTesting public boolean runMailboxStep() throws Exception
Exception
@VisibleForTesting public boolean isMailboxLoopRunning()
protected CompletableFuture<Void> getCompletionFuture()
public final Future<Void> cancel() throws Exception
AbstractInvokable
cancel
in class AbstractInvokable
AbstractInvokable
is fully terminated.
Note that it may never complete if the invokable is stuck.Exception
- thrown if any exception occurs during the execution of the user codepublic MailboxExecutorFactory getMailboxExecutorFactory()
public final boolean isRunning()
public final boolean isCanceled()
public final boolean isFailing()
protected void finalize() throws Throwable
This should not be relied upon! It will cause shutdown to happen much later than if manual shutdown is attempted, and cause threads to linger for longer than needed.
public final String getName()
public CheckpointStorageWorkerView getCheckpointStorage()
public StreamConfig getConfiguration()
public StreamStatusMaintainer getStreamStatusMaintainer()
public Future<Boolean> triggerCheckpointAsync(CheckpointMetaData checkpointMetaData, CheckpointOptions checkpointOptions)
AbstractInvokable
This method is called for tasks that start the checkpoints by injecting the initial
barriers, i.e., the source tasks. In contrast, checkpoints on downstream operators, which are
the result of receiving checkpoint barriers, invoke the AbstractInvokable.triggerCheckpointOnBarrier(CheckpointMetaData, CheckpointOptions, CheckpointMetricsBuilder)
method.
triggerCheckpointAsync
in class AbstractInvokable
checkpointMetaData
- Meta data for about this checkpointcheckpointOptions
- Options for performing this checkpointfalse
if the checkpoint was not carried out, true
otherwisepublic void triggerCheckpointOnBarrier(CheckpointMetaData checkpointMetaData, CheckpointOptions checkpointOptions, CheckpointMetricsBuilder checkpointMetrics) throws IOException
AbstractInvokable
triggerCheckpointOnBarrier
in class AbstractInvokable
checkpointMetaData
- Meta data for about this checkpointcheckpointOptions
- Options for performing this checkpointcheckpointMetrics
- Metrics about this checkpointIOException
public void abortCheckpointOnBarrier(long checkpointId, CheckpointException cause) throws IOException
AbstractInvokable
CancelCheckpointMarker
.
This requires implementing tasks to forward a CancelCheckpointMarker
to their outputs.
abortCheckpointOnBarrier
in class AbstractInvokable
checkpointId
- The ID of the checkpoint to be aborted.cause
- The reason why the checkpoint was aborted during alignmentIOException
protected void declineCheckpoint(long checkpointId)
public final ExecutorService getAsyncOperationsThreadPool()
public Future<Void> notifyCheckpointCompleteAsync(long checkpointId)
AbstractInvokable
notifyCheckpointCompleteAsync
in class AbstractInvokable
checkpointId
- The ID of the checkpoint that is complete.public Future<Void> notifyCheckpointAbortAsync(long checkpointId)
AbstractInvokable
notifyCheckpointAbortAsync
in class AbstractInvokable
checkpointId
- The ID of the checkpoint that is aborted.public void dispatchOperatorEvent(OperatorID operator, SerializedValue<OperatorEvent> event) throws FlinkException
dispatchOperatorEvent
in class AbstractInvokable
FlinkException
public ProcessingTimeServiceFactory getProcessingTimeServiceFactory()
public void handleAsyncException(String message, Throwable exception)
In more detail, it marks task execution failed for an external reason (a reason other than the task code itself throwing an exception). If the task is already in a terminal state (such as FINISHED, CANCELED, FAILED), or if the task is already canceling this does nothing. Otherwise it sets the state to FAILED, and, if the invokable code is running, starts an asynchronous thread that aborts that code.
This method never blocks.
handleAsyncException
in interface AsyncExceptionHandler
public final CloseableRegistry getCancelables()
@VisibleForTesting public static <OUT> RecordWriterDelegate<SerializationDelegate<StreamRecord<OUT>>> createRecordWriterDelegate(StreamConfig configuration, Environment environment)
protected long getAsyncCheckpointStartDelayNanos()
public boolean isUsingNonBlockingInput()
isUsingNonBlockingInput
in class AbstractInvokable
InputGate.getNext()
is used (as opposed to
InputGate.pollNext()
. To be removed together with the DataSet API.Copyright © 2014–2022 The Apache Software Foundation. All rights reserved.