org.apache.beam.sdk.fn.data

Class BeamFnDataInboundObserver<T>

java.lang.Object

org.apache.beam.sdk.fn.data.BeamFnDataInboundObserver<T>

All Implemented Interfaces:

java.util.function.Consumer<org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements.Data>, InboundDataClient

public class BeamFnDataInboundObserver<T>
extends java.lang.Object
implements java.util.function.Consumer<org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements.Data>, InboundDataClient

Decodes individually consumed BeamFnApi.Elements.Data with the provided Coder passing the individual decoded elements to the provided consumer.

Constructor Summary

Constructors

Constructor and Description
BeamFnDataInboundObserver(Coder<org.apache.beam.sdk.util.WindowedValue<T>> coder, FnDataReceiver<org.apache.beam.sdk.util.WindowedValue<T>> consumer, InboundDataClient readFuture)

Method Summary

Modifier and Type	Method and Description
void	accept(org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements.Data t)
void	awaitCompletion() Block until the client has completed reading from the inbound stream.
void	cancel() Cancels the client, causing it to drop any future inbound data.
void	complete() Mark the client as completed.
void	fail(java.lang.Throwable t) Mark the client as completed with an exception.
static <T> BeamFnDataInboundObserver<T>	forConsumer(Coder<org.apache.beam.sdk.util.WindowedValue<T>> coder, FnDataReceiver<org.apache.beam.sdk.util.WindowedValue<T>> receiver)
boolean	isDone() Returns true if the client is done, either via completing successfully or by being cancelled.

Methods inherited from class java.lang.Object

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

Methods inherited from interface java.util.function.Consumer

andThen

Constructor Detail

BeamFnDataInboundObserver

public BeamFnDataInboundObserver(Coder<org.apache.beam.sdk.util.WindowedValue<T>> coder,
                                 FnDataReceiver<org.apache.beam.sdk.util.WindowedValue<T>> consumer,
                                 InboundDataClient readFuture)

Method Detail

forConsumer

public static <T> BeamFnDataInboundObserver<T> forConsumer(Coder<org.apache.beam.sdk.util.WindowedValue<T>> coder,
                                                           FnDataReceiver<org.apache.beam.sdk.util.WindowedValue<T>> receiver)

accept

public void accept(org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements.Data t)

Specified by:

accept in interface java.util.function.Consumer<org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements.Data>

awaitCompletion

public void awaitCompletion()
                     throws java.lang.Exception

Description copied from interface: InboundDataClient

Block until the client has completed reading from the inbound stream.

Specified by:

awaitCompletion in interface InboundDataClient

Throws:

java.lang.InterruptedException - if the client is interrupted before completing.

java.util.concurrent.CancellationException - if the client is cancelled before completing.

java.lang.Exception - if the client throws an exception while awaiting completion.

isDone

public boolean isDone()

Description copied from interface: InboundDataClient

Returns true if the client is done, either via completing successfully or by being cancelled.

Specified by:

isDone in interface InboundDataClient

cancel

public void cancel()

Description copied from interface: InboundDataClient

Cancels the client, causing it to drop any future inbound data.

Specified by:

cancel in interface InboundDataClient

complete

public void complete()

Description copied from interface: InboundDataClient

Mark the client as completed.

Specified by:

complete in interface InboundDataClient

fail

public void fail(java.lang.Throwable t)

Description copied from interface: InboundDataClient

Mark the client as completed with an exception. Calls to awaitCompletion will terminate by throwing the provided exception.

Specified by:

fail in interface InboundDataClient

Parameters:

t - the throwable that caused this client to fail