org.apache.beam.sdk.expansion.service

Interface TransformProvider<InputT extends PInput,OutputT extends POutput>

Type Parameters:

InputT - input PInput type of the transform

OutputT - output POutput type of the transform

All Known Implementing Classes:

ExpansionServiceSchemaTransformProvider

public interface TransformProvider<InputT extends PInput,OutputT extends POutput>

Provides a mapping of RunnerApi.FunctionSpec to a PTransform, together with mappings of its inputs and outputs to maps of PCollections.

Method Summary

Modifier and Type	Method and Description
default java.util.Map<java.lang.String,PCollection<?>>	apply(Pipeline p, java.lang.String name, org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec, java.util.Map<java.lang.String,PCollection<?>> inputs)
default InputT	createInput(Pipeline p, java.util.Map<java.lang.String,PCollection<?>> inputs)
default java.util.Map<java.lang.String,PCollection<?>>	extractOutputs(OutputT output)
default java.util.List<java.lang.String>	getDependencies(org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec, PipelineOptions options)
PTransform<InputT,OutputT>	getTransform(org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec, PipelineOptions options)
default java.lang.String	getTransformUniqueID(org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec)

Method Detail

getTransform

PTransform<InputT,OutputT> getTransform(org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec,
                                        PipelineOptions options)

createInput

default InputT createInput(Pipeline p,
                           java.util.Map<java.lang.String,PCollection<?>> inputs)

extractOutputs

default java.util.Map<java.lang.String,PCollection<?>> extractOutputs(OutputT output)

apply

default java.util.Map<java.lang.String,PCollection<?>> apply(Pipeline p,
                                                             java.lang.String name,
                                                             org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec,
                                                             java.util.Map<java.lang.String,PCollection<?>> inputs)

getTransformUniqueID

default java.lang.String getTransformUniqueID(org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec)

getDependencies

default java.util.List<java.lang.String> getDependencies(org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec,
                                                         PipelineOptions options)