org.apache.flink.streaming.api.runners.python.beam

Class BeamPythonFunctionRunner

java.lang.Object

org.apache.flink.streaming.api.runners.python.beam.BeamPythonFunctionRunner

All Implemented Interfaces:

PythonFunctionRunner

Direct Known Subclasses:

BeamDataStreamPythonFunctionRunner, BeamTableStatefulPythonFunctionRunner, BeamTableStatelessPythonFunctionRunner

@Internal
public abstract class BeamPythonFunctionRunner
extends Object
implements PythonFunctionRunner

A BeamPythonFunctionRunner used to execute Python functions.

Field Summary

Fields

Modifier and Type	Field and Description
protected static org.slf4j.Logger	LOG
protected org.apache.beam.sdk.fn.data.FnDataReceiver<org.apache.beam.sdk.util.WindowedValue<byte[]>>	mainInputReceiver The receiver which forwards the input elements to a remote environment for processing.
protected LinkedBlockingQueue<byte[]>	resultBuffer The Python function execution result receiver.
protected Tuple2<byte[],Integer>	resultTuple The Python function execution result tuple.

Constructor Summary

Constructors

Constructor and Description
BeamPythonFunctionRunner(String taskName, PythonEnvironmentManager environmentManager, String functionUrn, Map<String,String> jobOptions, FlinkMetricContainer flinkMetricContainer, KeyedStateBackend keyedStateBackend, TypeSerializer keySerializer, MemoryManager memoryManager, double managedMemoryFraction)

Method Summary

Modifier and Type	Method and Description
void	close() Tear-down the Python function runner.
org.apache.beam.runners.fnexecution.control.JobBundleFactory	createJobBundleFactory(org.apache.beam.vendor.grpc.v1p26p0.com.google.protobuf.Struct pipelineOptions)
void	flush() Forces to finish the processing of the current bundle of elements.
protected abstract org.apache.beam.model.pipeline.v1.RunnerApi.Coder	getInputCoderProto()
protected abstract org.apache.beam.model.pipeline.v1.RunnerApi.Coder	getOutputCoderProto()
protected abstract byte[]	getUserDefinedFunctionsProtoBytes()
void	open(PythonConfig config) Prepares the Python function runner, such as preparing the Python execution environment, etc.
Tuple2<byte[],Integer>	pollResult() Retrieves the Python function result.
void	process(byte[] data) Executes the Python function with the input byte array.
protected void	startBundle()

Methods inherited from class java.lang.Object

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

Field Detail

LOG

protected static final org.slf4j.Logger LOG

resultTuple

protected final Tuple2<byte[],Integer> resultTuple

The Python function execution result tuple.

resultBuffer

protected transient LinkedBlockingQueue<byte[]> resultBuffer

The Python function execution result receiver.

mainInputReceiver

protected transient org.apache.beam.sdk.fn.data.FnDataReceiver<org.apache.beam.sdk.util.WindowedValue<byte[]>> mainInputReceiver

The receiver which forwards the input elements to a remote environment for processing.

Constructor Detail

BeamPythonFunctionRunner

public BeamPythonFunctionRunner(String taskName,
                                PythonEnvironmentManager environmentManager,
                                String functionUrn,
                                Map<String,String> jobOptions,
                                FlinkMetricContainer flinkMetricContainer,
                                @Nullable
                                KeyedStateBackend keyedStateBackend,
                                @Nullable
                                TypeSerializer keySerializer,
                                @Nullable
                                MemoryManager memoryManager,
                                double managedMemoryFraction)

Method Detail

open

public void open(PythonConfig config)
          throws Exception

Description copied from interface: PythonFunctionRunner

Prepares the Python function runner, such as preparing the Python execution environment, etc.

Specified by:

open in interface PythonFunctionRunner

Throws:

Exception

close

public void close()
           throws Exception

Description copied from interface: PythonFunctionRunner

Tear-down the Python function runner.

Specified by:

close in interface PythonFunctionRunner

Throws:

Exception

process

public void process(byte[] data)
             throws Exception

Description copied from interface: PythonFunctionRunner

Executes the Python function with the input byte array.

Specified by:

process in interface PythonFunctionRunner

Parameters:

data - the byte array data.

Throws:

Exception

pollResult

public Tuple2<byte[],Integer> pollResult()
                                  throws Exception

Description copied from interface: PythonFunctionRunner

Retrieves the Python function result.

Specified by:

pollResult in interface PythonFunctionRunner

Returns:

the head of he Python function result buffer, or null if the result buffer is empty. f0 means the byte array buffer which stores the Python function result. f1 means the length of the Python function result byte array.

Throws:

Exception

flush

public void flush()
           throws Exception

Description copied from interface: PythonFunctionRunner

Forces to finish the processing of the current bundle of elements. It will flush the data cached in the data buffer for processing and retrieves the state mutations (if exists) made by the Python function. The call blocks until all of the outputs produced by this bundle have been received.

Specified by:

flush in interface PythonFunctionRunner

Throws:

Exception

createJobBundleFactory

public org.apache.beam.runners.fnexecution.control.JobBundleFactory createJobBundleFactory(org.apache.beam.vendor.grpc.v1p26p0.com.google.protobuf.Struct pipelineOptions)
                                                                                    throws Exception

Throws:

Exception

startBundle

protected void startBundle()

getUserDefinedFunctionsProtoBytes

protected abstract byte[] getUserDefinedFunctionsProtoBytes()

getInputCoderProto

protected abstract org.apache.beam.model.pipeline.v1.RunnerApi.Coder getInputCoderProto()

getOutputCoderProto

protected abstract org.apache.beam.model.pipeline.v1.RunnerApi.Coder getOutputCoderProto()