org.apache.beam.sdk.transforms

Class FlatMapElements<InputT,OutputT>

java.lang.Object

org.apache.beam.sdk.transforms.PTransform<PCollection<? extends InputT>,PCollection<OutputT>>

org.apache.beam.sdk.transforms.FlatMapElements<InputT,OutputT>

All Implemented Interfaces:

java.io.Serializable, HasDisplayData

public class FlatMapElements<InputT,OutputT>
extends PTransform<PCollection<? extends InputT>,PCollection<OutputT>>

PTransforms for mapping a simple function that returns iterables over the elements of a PCollection and merging the results.

See Also:

Serialized Form

Field Summary

Fields inherited from class org.apache.beam.sdk.transforms.PTransform

name

Method Summary

Modifier and Type	Method and Description
PCollection<OutputT>	expand(PCollection<? extends InputT> input) Override this method to specify how this PTransform should be expanded on the given InputT.
static <OutputT> FlatMapElements<?,OutputT>	into(TypeDescriptor<OutputT> outputType) Returns a new FlatMapElements transform with the given type descriptor for the output type, but the mapping function yet to be specified using via(ProcessFunction).
void	populateDisplayData(DisplayData.Builder builder) Register display data for the given transform or component.
<NewInputT> FlatMapElements<NewInputT,OutputT>	via(Contextful<Contextful.Fn<NewInputT,java.lang.Iterable<OutputT>>> fn) Like via(ProcessFunction), but allows access to additional context.
static <InputT,OutputT> FlatMapElements<InputT,OutputT>	via(InferableFunction<? super InputT,? extends java.lang.Iterable<OutputT>> fn) For a InferableFunction<InputT, ? extends Iterable<OutputT>> fn, return a PTransform that applies fn to every element of the input PCollection<InputT> and outputs all of the elements to the output PCollection<OutputT>.
<NewInputT> FlatMapElements<NewInputT,OutputT>	via(ProcessFunction<NewInputT,? extends java.lang.Iterable<OutputT>> fn) For a ProcessFunction<InputT, ? extends Iterable<OutputT>> fn, returns a PTransform that applies fn to every element of the input PCollection<InputT> and outputs all of the elements to the output PCollection<OutputT>.

Methods inherited from class org.apache.beam.sdk.transforms.PTransform

compose, getAdditionalInputs, getDefaultOutputCoder, getDefaultOutputCoder, getDefaultOutputCoder, getKindString, getName, toString, validate

Methods inherited from class java.lang.Object

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

Method Detail

via

public static <InputT,OutputT> FlatMapElements<InputT,OutputT> via(InferableFunction<? super InputT,? extends java.lang.Iterable<OutputT>> fn)

For a InferableFunction<InputT, ? extends Iterable<OutputT>> fn, return a PTransform that applies fn to every element of the input PCollection<InputT> and outputs all of the elements to the output PCollection<OutputT>.

InferableFunction has the advantage of providing type descriptor information, but it is generally more convenient to specify output type via into(TypeDescriptor), and provide the mapping as a lambda expression to via(ProcessFunction).

Example usage:

 PCollection<String> lines = ...;
 PCollection<String> words = lines.apply(FlatMapElements.via(
     new InferableFunction<String, List<String>>() {
       public Integer apply(String line) throws Exception {
         return Arrays.asList(line.split(" "));
       }
     });
 

into

public static <OutputT> FlatMapElements<?,OutputT> into(TypeDescriptor<OutputT> outputType)

Returns a new FlatMapElements transform with the given type descriptor for the output type, but the mapping function yet to be specified using via(ProcessFunction).

via

public <NewInputT> FlatMapElements<NewInputT,OutputT> via(ProcessFunction<NewInputT,? extends java.lang.Iterable<OutputT>> fn)

For a ProcessFunction<InputT, ? extends Iterable<OutputT>> fn, returns a PTransform that applies fn to every element of the input PCollection<InputT> and outputs all of the elements to the output PCollection<OutputT>.

Example usage:

 PCollection<String> words = lines.apply(
     FlatMapElements.into(TypeDescriptors.strings())
                    .via((String line) -> Arrays.asList(line.split(" ")))
 

via

@Experimental(value=CONTEXTFUL)
public <NewInputT> FlatMapElements<NewInputT,OutputT> via(Contextful<Contextful.Fn<NewInputT,java.lang.Iterable<OutputT>>> fn)

Like via(ProcessFunction), but allows access to additional context.

expand

public PCollection<OutputT> expand(PCollection<? extends InputT> input)

Description copied from class: PTransform

Override this method to specify how this PTransform should be expanded on the given InputT.

NOTE: This method should not be called directly. Instead apply the PTransform should be applied to the InputT using the apply method.

Composite transforms, which are defined in terms of other transforms, should return the output of one of the composed transforms. Non-composite transforms, which do not apply any transforms internally, should return a new unbound output and register evaluators (via backend-specific registration methods).

Specified by:

expand in class PTransform<PCollection<? extends InputT>,PCollection<OutputT>>

populateDisplayData

public void populateDisplayData(DisplayData.Builder builder)

Description copied from class: PTransform

Register display data for the given transform or component.

populateDisplayData(DisplayData.Builder) is invoked by Pipeline runners to collect display data via DisplayData.from(HasDisplayData). Implementations may call super.populateDisplayData(builder) in order to register display data in the current namespace, but should otherwise use subcomponent.populateDisplayData(builder) to use the namespace of the subcomponent.

By default, does not register any display data. Implementors may override this method to provide their own display data.

Specified by:

populateDisplayData in interface HasDisplayData

Overrides:

populateDisplayData in class PTransform<PCollection<? extends InputT>,PCollection<OutputT>>

Parameters:

builder - The builder to populate with display data.

See Also:

HasDisplayData