Package 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	Description
default Map<String,PCollection<?>>	apply(Pipeline p, String name, org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec, Map<String,PCollection<?>> inputs)
default InputT	createInput(Pipeline p, Map<String,PCollection<?>> inputs)
default Map<String,PCollection<?>>	extractOutputs(OutputT output)
default List<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 String	getTransformUniqueID(org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec)

Method Details

getTransform

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

createInput

default InputT createInput(Pipeline p,
 Map<String,PCollection<?>> inputs)

extractOutputs

default Map<String,PCollection<?>> extractOutputs(OutputT output)

apply

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

getTransformUniqueID

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

getDependencies

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