org.apache.beam.sdk.fn.data

Class BeamFnDataGrpcMultiplexer

java.lang.Object

org.apache.beam.sdk.fn.data.BeamFnDataGrpcMultiplexer

All Implemented Interfaces:

java.lang.AutoCloseable

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

A gRPC multiplexer for a specific Endpoints.ApiServiceDescriptor.

Multiplexes data for inbound consumers based upon their individual org.apache.beam.model.fnexecution.v1.BeamFnApi.Targets.

Multiplexing inbound and outbound streams is as thread safe as the consumers of those streams. For inbound streams, this is as thread safe as the inbound observers. For outbound streams, this is as thread safe as the underlying stream observer.

TODO: Add support for multiplexing over multiple outbound observers by stickying the output location with a specific outbound observer.

Constructor Summary

Constructors

Constructor and Description
BeamFnDataGrpcMultiplexer(org.apache.beam.model.pipeline.v1.Endpoints.ApiServiceDescriptor apiServiceDescriptor, OutboundObserverFactory outboundObserverFactory, OutboundObserverFactory.BasicFactory<org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements,org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements> baseOutboundObserverFactory)

Method Summary

Modifier and Type	Method and Description
void	close()
org.apache.beam.vendor.grpc.v1p26p0.io.grpc.stub.StreamObserver<org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements>	getInboundObserver()
org.apache.beam.vendor.grpc.v1p26p0.io.grpc.stub.StreamObserver<org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements>	getOutboundObserver()
<T> void	registerConsumer(LogicalEndpoint inputLocation, java.util.function.BiConsumer<org.apache.beam.vendor.grpc.v1p26p0.com.google.protobuf.ByteString,java.lang.Boolean> bytesReceiver)
java.lang.String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

BeamFnDataGrpcMultiplexer

public BeamFnDataGrpcMultiplexer(@Nullable
                                 org.apache.beam.model.pipeline.v1.Endpoints.ApiServiceDescriptor apiServiceDescriptor,
                                 OutboundObserverFactory outboundObserverFactory,
                                 OutboundObserverFactory.BasicFactory<org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements,org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements> baseOutboundObserverFactory)

Method Detail

toString

public java.lang.String toString()

Overrides:

toString in class java.lang.Object

getInboundObserver

public org.apache.beam.vendor.grpc.v1p26p0.io.grpc.stub.StreamObserver<org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements> getInboundObserver()

getOutboundObserver

public org.apache.beam.vendor.grpc.v1p26p0.io.grpc.stub.StreamObserver<org.apache.beam.model.fnexecution.v1.BeamFnApi.Elements> getOutboundObserver()

registerConsumer

public <T> void registerConsumer(LogicalEndpoint inputLocation,
                                 java.util.function.BiConsumer<org.apache.beam.vendor.grpc.v1p26p0.com.google.protobuf.ByteString,java.lang.Boolean> bytesReceiver)

close

public void close()

Specified by:

close in interface java.lang.AutoCloseable