org.apache.beam.runners.fnexecution.control

Class SdkHarnessClient.BundleProcessor<T>

java.lang.Object

org.apache.beam.runners.fnexecution.control.SdkHarnessClient.BundleProcessor<T>

Enclosing class:

SdkHarnessClient

public class SdkHarnessClient.BundleProcessor<T>
extends java.lang.Object

A processor capable of creating bundles for some registered BeamFnApi.ProcessBundleDescriptor.

Method Summary

Modifier and Type	Method and Description
java.util.concurrent.CompletionStage<org.apache.beam.model.fnexecution.v1.BeamFnApi.RegisterResponse>	getRegistrationFuture()
SdkHarnessClient.ActiveBundle<T>	newBundle(java.util.Map<org.apache.beam.model.fnexecution.v1.BeamFnApi.Target,RemoteOutputReceiver<?>> outputReceivers) Start a new bundle for the given BeamFnApi.ProcessBundleDescriptor identifier.
SdkHarnessClient.ActiveBundle<T>	newBundle(java.util.Map<org.apache.beam.model.fnexecution.v1.BeamFnApi.Target,RemoteOutputReceiver<?>> outputReceivers, StateRequestHandler stateRequestHandler) Start a new bundle for the given BeamFnApi.ProcessBundleDescriptor identifier.

Methods inherited from class java.lang.Object

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

Method Detail

getRegistrationFuture

public java.util.concurrent.CompletionStage<org.apache.beam.model.fnexecution.v1.BeamFnApi.RegisterResponse> getRegistrationFuture()

newBundle

public SdkHarnessClient.ActiveBundle<T> newBundle(java.util.Map<org.apache.beam.model.fnexecution.v1.BeamFnApi.Target,RemoteOutputReceiver<?>> outputReceivers)

Start a new bundle for the given BeamFnApi.ProcessBundleDescriptor identifier.

The input channels for the returned SdkHarnessClient.ActiveBundle are derived from the instructions in the BeamFnApi.ProcessBundleDescriptor.

NOTE: It is important to SdkHarnessClient.close() each bundle after all elements are emitted.

 try (ActiveBundle<InputT> bundle = SdkHarnessClient.newBundle(...)) {
   FnDataReceiver<InputT> inputReceiver = bundle.getInputReceiver();
   // send all elements ...
 }
 

newBundle

public SdkHarnessClient.ActiveBundle<T> newBundle(java.util.Map<org.apache.beam.model.fnexecution.v1.BeamFnApi.Target,RemoteOutputReceiver<?>> outputReceivers,
                                                  StateRequestHandler stateRequestHandler)

Start a new bundle for the given BeamFnApi.ProcessBundleDescriptor identifier.

The input channels for the returned SdkHarnessClient.ActiveBundle are derived from the instructions in the BeamFnApi.ProcessBundleDescriptor.

NOTE: It is important to SdkHarnessClient.close() each bundle after all elements are emitted.

 try (ActiveBundle<InputT> bundle = SdkHarnessClient.newBundle(...)) {
   FnDataReceiver<InputT> inputReceiver = bundle.getInputReceiver();
   // send all elements ...
 }