Package org.apache.flink.core.state

Class StateFutureImpl<T>

java.lang.Object

org.apache.flink.core.state.StateFutureImpl<T>

All Implemented Interfaces:

org.apache.flink.api.common.state.v2.StateFuture<T>, InternalStateFuture<T>

@Internal
public class StateFutureImpl<T>
extends Object
implements InternalStateFuture<T>

The default implementation of StateFuture. This is managed by the runtime framework and should never be directly created in user code. It will handle the completion and callback trigger, and most of the design are borrowed from the CompletableFuture. In the basic version of this implementation, we wrap CompletableFuture for simplification. TODO: remove CompletableFuture.

This class is on hot path and very complex, please take care of the performance as well as the running thread of each block when you decide to touch it.

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
static interface	StateFutureImpl.AsyncFrameworkExceptionHandler	Handle exceptions thrown by async state callback framework.
static interface	StateFutureImpl.CallbackRunner	The entry for a state future to submit task to mailbox.

Field Summary

Fields

Modifier and Type	Field	Description
protected StateFutureImpl.CallbackRunner	callbackRunner	The callback runner.
protected StateFutureImpl.AsyncFrameworkExceptionHandler	exceptionHandler	The exception handler that handles callback framework's error.

Constructor Summary

Constructors

Constructor	Description
StateFutureImpl(StateFutureImpl.CallbackRunner callbackRunner, StateFutureImpl.AsyncFrameworkExceptionHandler exceptionHandler)

Method Summary

Modifier and Type	Method	Description
void	callbackFinished()	Will be triggered when a callback finishes processing.
void	callbackRegistered()	Will be triggered when a callback is registered.
void	complete(T result)	Complete this future.
void	completeExceptionally(String message, Throwable ex)	Fail this future and pass the given exception to the runtime.
<A> StateFutureImpl<A>	makeNewStateFuture()	Make a new future based on context of this future.
void	postComplete(boolean inCallbackRunner)	Will be triggered when this future completes.
org.apache.flink.api.common.state.v2.StateFuture<Void>	thenAccept(org.apache.flink.util.function.ThrowingConsumer<? super T,? extends Exception> action)
<U> org.apache.flink.api.common.state.v2.StateFuture<U>	thenApply(org.apache.flink.util.function.FunctionWithException<? super T,? extends U,? extends Exception> fn)
<U,V> org.apache.flink.api.common.state.v2.StateFuture<V>	thenCombine(org.apache.flink.api.common.state.v2.StateFuture<? extends U> other, org.apache.flink.util.function.BiFunctionWithException<? super T,? super U,? extends V,? extends Exception> fn)
<U> org.apache.flink.api.common.state.v2.StateFuture<U>	thenCompose(org.apache.flink.util.function.FunctionWithException<? super T,? extends org.apache.flink.api.common.state.v2.StateFuture<U>,? extends Exception> action)
void	thenSyncAccept(org.apache.flink.util.function.ThrowingConsumer<? super T,? extends Exception> action)	Accept the action in the same thread with the one of complete (or current thread if it has been completed).

Methods inherited from class java.lang.Object

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

Field Detail

callbackRunner

protected final StateFutureImpl.CallbackRunner callbackRunner

The callback runner.

exceptionHandler

protected final StateFutureImpl.AsyncFrameworkExceptionHandler exceptionHandler

The exception handler that handles callback framework's error.

Constructor Detail

StateFutureImpl

public StateFutureImpl(StateFutureImpl.CallbackRunner callbackRunner,
                       StateFutureImpl.AsyncFrameworkExceptionHandler exceptionHandler)

Method Detail

thenApply

public <U> org.apache.flink.api.common.state.v2.StateFuture<U> thenApply(org.apache.flink.util.function.FunctionWithException<? super T,? extends U,? extends Exception> fn)

Specified by:

thenApply in interface org.apache.flink.api.common.state.v2.StateFuture<T>

thenAccept

public org.apache.flink.api.common.state.v2.StateFuture<Void> thenAccept(org.apache.flink.util.function.ThrowingConsumer<? super T,? extends Exception> action)

Specified by:

thenAccept in interface org.apache.flink.api.common.state.v2.StateFuture<T>

thenCompose

public <U> org.apache.flink.api.common.state.v2.StateFuture<U> thenCompose(org.apache.flink.util.function.FunctionWithException<? super T,? extends org.apache.flink.api.common.state.v2.StateFuture<U>,? extends Exception> action)

Specified by:

thenCompose in interface org.apache.flink.api.common.state.v2.StateFuture<T>

thenCombine

public <U,V> org.apache.flink.api.common.state.v2.StateFuture<V> thenCombine(org.apache.flink.api.common.state.v2.StateFuture<? extends U> other,
                                                                                   org.apache.flink.util.function.BiFunctionWithException<? super T,? super U,? extends V,? extends Exception> fn)

Specified by:

thenCombine in interface org.apache.flink.api.common.state.v2.StateFuture<T>

makeNewStateFuture

public <A> StateFutureImpl<A> makeNewStateFuture()

Make a new future based on context of this future. Subclasses need to overload this method to generate their own instances (if needed).

Returns:

the new created future.

complete

public void complete(T result)

Description copied from interface: InternalStateFuture

Complete this future.

Specified by:

complete in interface InternalStateFuture<T>

completeExceptionally

public void completeExceptionally(String message,
                                  Throwable ex)

Description copied from interface: InternalStateFuture

Fail this future and pass the given exception to the runtime.

Specified by:

completeExceptionally in interface InternalStateFuture<T>

Parameters:

message - the description of this exception

ex - the exception

callbackRegistered

public void callbackRegistered()

Will be triggered when a callback is registered.

postComplete

public void postComplete(boolean inCallbackRunner)

Will be triggered when this future completes.

callbackFinished

public void callbackFinished()

Will be triggered when a callback finishes processing.

thenSyncAccept

public void thenSyncAccept(org.apache.flink.util.function.ThrowingConsumer<? super T,? extends Exception> action)

Description copied from interface: InternalStateFuture

Accept the action in the same thread with the one of complete (or current thread if it has been completed).

Specified by:

thenSyncAccept in interface InternalStateFuture<T>

Parameters:

action - the action to perform.