org.apache.beam.sdk.transforms.join

Class CoGroupByKey<K>

java.lang.Object

org.apache.beam.sdk.transforms.PTransform<KeyedPCollectionTuple<K>,PCollection<KV<K,CoGbkResult>>>

org.apache.beam.sdk.transforms.join.CoGroupByKey<K>

Type Parameters:

K - the type of the keys in the input and output PCollections

All Implemented Interfaces:

Serializable, HasDisplayData

public class CoGroupByKey<K>
extends PTransform<KeyedPCollectionTuple<K>,PCollection<KV<K,CoGbkResult>>>

A PTransform that performs a CoGroupByKey on a tuple of tables. A CoGroupByKey groups results from all tables by like keys into CoGbkResults, from which the results for any specific table can be accessed by the TupleTag supplied with the initial table.

Example of performing a CoGroupByKey followed by a ParDo that consumes the results:

 {@code
 PCollection> pt1 = ...;
 PCollection> pt2 = ...;

 final TupleTag t1 = new TupleTag<>();
 final TupleTag t2 = new TupleTag<>();
 PCollection> coGbkResultCollection =
   KeyedPCollectionTuple.of(t1, pt1)
                        .and(t2, pt2)
                        .apply(CoGroupByKey.create());

 PCollection finalResultCollection =
   coGbkResultCollection.apply(ParDo.of(
     new DoFn, T>() {

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<KV<K,CoGbkResult>>	apply(KeyedPCollectionTuple<K> input) Applies this PTransform on the given InputT, and returns its Output.
static <K> CoGroupByKey<K>	create() Returns a CoGroupByKey<K> PTransform.

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

getDefaultOutputCoder, getDefaultOutputCoder, getDefaultOutputCoder, getKindString, getName, populateDisplayData, toString, validate

Methods inherited from class java.lang.Object

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

Method Detail

create

public static <K> CoGroupByKey<K> create()

Returns a CoGroupByKey<K> PTransform.

Type Parameters:

K - the type of the keys in the input and output PCollections

apply

public PCollection<KV<K,CoGbkResult>> apply(KeyedPCollectionTuple<K> input)

Description copied from class: PTransform

Applies this PTransform on the given InputT, and returns its Output.

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:

apply in class PTransform<KeyedPCollectionTuple<K>,PCollection<KV<K,CoGbkResult>>>