InputT - type of input valuesAccumT - type of mutable accumulator valuesOutputT - type of output valuespublic abstract static class Combine.CombineFn<InputT,AccumT,OutputT>
extends java.lang.Object
CombineFn<InputT, AccumT, OutputT> specifies how to combine a
collection of input values of type InputT into a single
output value of type OutputT. It does this via one or more
intermediate mutable accumulator values of type AccumT.
The overall process to combine a collection of input
InputT values into a single output OutputT value is as
follows:
InputT values are partitioned into one or more
batches.
createAccumulator() operation is
invoked to create a fresh mutable accumulator value of type
AccumT, initialized to represent the combination of zero
values.
InputT value in a batch, the
addInput(AccumT, InputT) operation is invoked to add the value to that
batch's accumulator AccumT value. The accumulator may just
record the new value (e.g., if AccumT == List<InputT>, or may do
work to represent the combination more compactly.
mergeAccumulators(java.lang.Iterable<AccumT>) operation is invoked to
combine a collection of accumulator AccumT values into a
single combined output accumulator AccumT value, once the
merging accumulators have had all all the input values in their
batches added to them. This operation is invoked repeatedly,
until there is only one accumulator value left.
extractOutput(AccumT) operation is invoked on the final
accumulator AccumT value to get the output OutputT value.
For example:
public class AverageFn extends CombineFn<Integer, AverageFn.Accum, Double> {
public static class Accum {
int sum = 0;
int count = 0;
}
public Accum createAccumulator() {
return new Accum();
}
public Accum addInput(Accum accum, Integer input) {
accum.sum += input;
accum.count++;
return accum;
}
public Accum mergeAccumulators(Iterable<Accum> accums) {
Accum merged = createAccumulator();
for (Accum accum : accums) {
merged.sum += accum.sum;
merged.count += accum.count;
}
return merged;
}
public Double extractOutput(Accum accum) {
return ((double) accum.sum) / accum.count;
}
}
PCollection<Integer> pc = ...;
PCollection<Double> average = pc.apply(Combine.globally(new AverageFn()));
Combining functions used by Combine.Globally,
Combine.PerKey, Combine.GroupedValues, and
PTransforms derived from them should be
associative and commutative. Associativity is
required because input values are first broken up into subgroups
before being combined, and their intermediate results further
combined, in an arbitrary tree structure. Commutativity is
required because any order of the input values is ignored when
breaking up input values into groups.
| Constructor and Description |
|---|
CombineFn() |
| Modifier and Type | Method and Description |
|---|---|
abstract AccumT |
addInput(AccumT accumulator,
InputT input)
Adds the given input value to the given accumulator, returning the
new accumulator value.
|
OutputT |
apply(java.lang.Iterable<? extends InputT> inputs)
Applies this
CombineFn to a collection of input values
to produce a combined output value. |
AccumT |
compact(AccumT accumulator)
Returns an accumulator that represents the same logical value as the
input accumulator, but may have a more compact representation.
|
abstract AccumT |
createAccumulator()
Returns a new, mutable accumulator value, representing the accumulation of zero input values.
|
OutputT |
defaultValue()
Returns the default value when there are no values added to the accumulator.
|
abstract OutputT |
extractOutput(AccumT accumulator)
Returns the output value that is the result of combining all
the input values represented by the given accumulator.
|
java.lang.reflect.TypeVariable<?> |
getAccumTVariable()
Returns the
TypeVariable of AccumT. |
Coder<AccumT> |
getAccumulatorCoder(CoderRegistry registry,
Coder<InputT> inputCoder)
Returns the
Coder to use for accumulator AccumT
values, or null if it is not able to be inferred. |
Coder<OutputT> |
getDefaultOutputCoder(CoderRegistry registry,
Coder<InputT> inputCoder)
Returns the
Coder to use by default for output
OutputT values, or null if it is not able to be inferred. |
java.lang.String |
getIncompatibleGlobalWindowErrorMessage()
Returns the error message for not supported default values in Combine.globally().
|
java.lang.reflect.TypeVariable<?> |
getInputTVariable()
Returns the
TypeVariable of InputT. |
java.lang.reflect.TypeVariable<?> |
getOutputTVariable()
Returns the
TypeVariable of OutputT. |
TypeDescriptor<OutputT> |
getOutputType()
Returns a
TypeDescriptor capturing what is known statically
about the output type of this CombineFn instance's
most-derived class. |
abstract AccumT |
mergeAccumulators(java.lang.Iterable<AccumT> accumulators)
Returns an accumulator representing the accumulation of all the
input values accumulated in the merging accumulators.
|
void |
populateDisplayData(DisplayData.Builder builder)
Register display data for the given transform or component.
|
public abstract AccumT createAccumulator()
public abstract AccumT addInput(AccumT accumulator, InputT input)
For efficiency, the input accumulator may be modified and returned.
public abstract AccumT mergeAccumulators(java.lang.Iterable<AccumT> accumulators)
May modify any of the argument accumulators. May return a fresh accumulator, or may return one of the (modified) argument accumulators.
public abstract OutputT extractOutput(AccumT accumulator)
public AccumT compact(AccumT accumulator)
For most CombineFns this would be a no-op, but should be overridden by CombineFns that (for example) buffer up elements and combine them in batches.
For efficiency, the input accumulator may be modified and returned.
By default returns the original accumulator.
public OutputT apply(java.lang.Iterable<? extends InputT> inputs)
CombineFn to a collection of input values
to produce a combined output value.
Useful when using a CombineFn separately from a
Combine transform. Does not invoke the
mergeAccumulators(java.lang.Iterable<AccumT>) operation.
public OutputT defaultValue()
By default returns the extract output of an empty accumulator.
public TypeDescriptor<OutputT> getOutputType()
TypeDescriptor capturing what is known statically
about the output type of this CombineFn instance's
most-derived class.
In the normal case of a concrete CombineFn subclass with
no generic type parameters of its own, this will be a complete
non-generic type.
public Coder<AccumT> getAccumulatorCoder(CoderRegistry registry, Coder<InputT> inputCoder) throws CannotProvideCoderException
CombineFnBase.GlobalCombineFnCoder to use for accumulator AccumT
values, or null if it is not able to be inferred.
By default, uses the knowledge of the Coder being used
for InputT values and the enclosing Pipeline's
CoderRegistry to try to infer the Coder for AccumT
values.
This is the Coder used to send data through a communication-intensive shuffle step, so a compact and efficient representation may have significant performance benefits.
getAccumulatorCoder in interface CombineFnBase.GlobalCombineFn<InputT,AccumT,OutputT>CannotProvideCoderExceptionpublic Coder<OutputT> getDefaultOutputCoder(CoderRegistry registry, Coder<InputT> inputCoder) throws CannotProvideCoderException
CombineFnBase.GlobalCombineFnCoder to use by default for output
OutputT values, or null if it is not able to be inferred.
By default, uses the knowledge of the Coder being
used for input InputT values and the enclosing
Pipeline's CoderRegistry to try to infer the
Coder for OutputT values.
getDefaultOutputCoder in interface CombineFnBase.GlobalCombineFn<InputT,AccumT,OutputT>CannotProvideCoderExceptionpublic java.lang.String getIncompatibleGlobalWindowErrorMessage()
CombineFnBase.GlobalCombineFngetIncompatibleGlobalWindowErrorMessage in interface CombineFnBase.GlobalCombineFn<InputT,AccumT,OutputT>public java.lang.reflect.TypeVariable<?> getInputTVariable()
TypeVariable of InputT.public java.lang.reflect.TypeVariable<?> getAccumTVariable()
TypeVariable of AccumT.public java.lang.reflect.TypeVariable<?> getOutputTVariable()
TypeVariable of OutputT.public void populateDisplayData(DisplayData.Builder builder)
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.
populateDisplayData in interface HasDisplayDatabuilder - The builder to populate with display data.HasDisplayData