Package org.apache.beam.sdk.io.iceberg

Class BundleLifter<T>

java.lang.Object

org.apache.beam.sdk.transforms.PTransform<PCollection<T>,PCollectionTuple>

org.apache.beam.sdk.io.iceberg.BundleLifter<T>

Type Parameters:

T - The type of elements in the input PCollection.

All Implemented Interfaces:

Serializable, HasDisplayData

public class BundleLifter<T>
extends PTransform<PCollection<T>,PCollectionTuple>

A PTransform that buffers elements and outputs them to one of two TupleTags based on the total size of the bundle in finish_bundle.

See Also:

• Serialized Form

Field Summary

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

annotations, displayData, name, resourceHints

Method Summary

Modifier and Type	Method	Description
PCollectionTuple	expand(PCollection<T> input)	Override this method to specify how this PTransform should be expanded on the given InputT.
static <T> BundleLifter<T>	of(TupleTag<T> smallBatchTag, TupleTag<T> largeBatchTag, int threshold)
static <T> BundleLifter<T>	of(TupleTag<T> smallBatchTag, TupleTag<T> largeBatchTag, int threshold, SerializableFunction<T,Integer> elementSizer)

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

addAnnotation, compose, compose, getAdditionalInputs, getAnnotations, getDefaultOutputCoder, getDefaultOutputCoder, getDefaultOutputCoder, getKindString, getName, getResourceHints, populateDisplayData, setDisplayData, setResourceHints, toString, validate, validate

Methods inherited from class java.lang.Object

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

Method Details

of

public static <T> BundleLifter<T> of(TupleTag<T> smallBatchTag,
 TupleTag<T> largeBatchTag,
 int threshold)

of

public static <T> BundleLifter<T> of(TupleTag<T> smallBatchTag,
 TupleTag<T> largeBatchTag,
 int threshold,
 SerializableFunction<T,Integer> elementSizer)

expand

public PCollectionTuple expand(PCollection<T> 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<T>,PCollectionTuple>