org.apache.beam.runners.fnexecution.control

Class SdkHarnessClient

java.lang.Object

org.apache.beam.runners.fnexecution.control.SdkHarnessClient

All Implemented Interfaces:

java.lang.AutoCloseable

public class SdkHarnessClient
extends java.lang.Object
implements java.lang.AutoCloseable

A high-level client for an SDK harness.

This provides a Java-friendly wrapper around InstructionRequestHandler and CloseableFnDataReceiver, which handle lower-level gRPC message wrangling.

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	SdkHarnessClient.ActiveBundle<InputT> An active bundle for a particular BeamFnApi.ProcessBundleDescriptor.
class	SdkHarnessClient.BundleProcessor<T> A processor capable of creating bundles for some registered BeamFnApi.ProcessBundleDescriptor.

Method Summary

Modifier and Type	Method and Description
void	close()
<T> SdkHarnessClient.BundleProcessor<T>	getProcessor(org.apache.beam.model.fnexecution.v1.BeamFnApi.ProcessBundleDescriptor descriptor, RemoteInputDestination<org.apache.beam.sdk.util.WindowedValue<T>> remoteInputDesination) Provides SdkHarnessClient.BundleProcessor that is capable of processing bundles not containing any state accesses such as: Side inputs User state Remote references
<T> SdkHarnessClient.BundleProcessor<T>	getProcessor(org.apache.beam.model.fnexecution.v1.BeamFnApi.ProcessBundleDescriptor descriptor, RemoteInputDestination<org.apache.beam.sdk.util.WindowedValue<T>> remoteInputDesination, StateDelegator stateDelegator) Provides SdkHarnessClient.BundleProcessor that is capable of processing bundles containing state accesses such as: Side inputs User state Remote references
static SdkHarnessClient	usingFnApiClient(InstructionRequestHandler fnApiControlClient, FnDataService fnApiDataService) Creates a client for a particular SDK harness.
SdkHarnessClient	withIdGenerator(IdGenerator idGenerator)

Methods inherited from class java.lang.Object

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

Method Detail

usingFnApiClient

public static SdkHarnessClient usingFnApiClient(InstructionRequestHandler fnApiControlClient,
                                                FnDataService fnApiDataService)

Creates a client for a particular SDK harness. It is the responsibility of the caller to ensure that these correspond to the same SDK harness, so control plane and data plane messages can be correctly associated.

withIdGenerator

public SdkHarnessClient withIdGenerator(IdGenerator idGenerator)

getProcessor

public <T> SdkHarnessClient.BundleProcessor<T> getProcessor(org.apache.beam.model.fnexecution.v1.BeamFnApi.ProcessBundleDescriptor descriptor,
                                                            RemoteInputDestination<org.apache.beam.sdk.util.WindowedValue<T>> remoteInputDesination)

Provides SdkHarnessClient.BundleProcessor that is capable of processing bundles not containing any state accesses such as:

• Side inputs
• User state
• Remote references

Note that bundle processors are cached based upon the the process bundle descriptor id. A previously created instance may be returned.

getProcessor

public <T> SdkHarnessClient.BundleProcessor<T> getProcessor(org.apache.beam.model.fnexecution.v1.BeamFnApi.ProcessBundleDescriptor descriptor,
                                                            RemoteInputDestination<org.apache.beam.sdk.util.WindowedValue<T>> remoteInputDesination,
                                                            StateDelegator stateDelegator)

Provides SdkHarnessClient.BundleProcessor that is capable of processing bundles containing state accesses such as:

• Side inputs
• User state
• Remote references

Note that bundle processors are cached based upon the the process bundle descriptor id. A previously created instance may be returned.

close

public void close()

Specified by:

close in interface java.lang.AutoCloseable