org.apache.beam.sdk.expansion.service

Interface ExpansionService.TransformProvider<InputT extends PInput,OutputT extends POutput>

Type Parameters:

InputT - input PInput type of the transform

OutputT - output POutput type of the transform

Enclosing class:

ExpansionService

public static interface ExpansionService.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,org.apache.beam.sdk.values.PCollection<?>>	apply(org.apache.beam.sdk.Pipeline p, java.lang.String name, org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec, java.util.Map<java.lang.String,org.apache.beam.sdk.values.PCollection<?>> inputs)
default InputT	createInput(org.apache.beam.sdk.Pipeline p, java.util.Map<java.lang.String,org.apache.beam.sdk.values.PCollection<?>> inputs)
default java.util.Map<java.lang.String,org.apache.beam.sdk.values.PCollection<?>>	extractOutputs(OutputT output)
default java.util.List<java.lang.String>	getDependencies(org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec, org.apache.beam.sdk.options.PipelineOptions options)
org.apache.beam.sdk.transforms.PTransform<InputT,OutputT>	getTransform(org.apache.beam.model.pipeline.v1.RunnerApi.FunctionSpec spec)

Method Detail

createInput

default InputT createInput(org.apache.beam.sdk.Pipeline p,
                           java.util.Map<java.lang.String,org.apache.beam.sdk.values.PCollection<?>> inputs)

getTransform

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

extractOutputs

default java.util.Map<java.lang.String,org.apache.beam.sdk.values.PCollection<?>> extractOutputs(OutputT output)

apply

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

getDependencies

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