InputT - type of input valuesAccumT - type of mutable accumulator valuesOutputT - type of output valuespublic abstract static class Combine.CombineFn<InputT,AccumT,OutputT> extends Object implements Serializable
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 void addInput(Accum accum, Integer input) {
accum.sum += input;
accum.count++;
}
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(Iterable<? extends InputT> inputs)
Applies this
CombineFn to a collection of input values
to produce a combined output value. |
<K> Combine.KeyedCombineFn<K,InputT,AccumT,OutputT> |
asKeyedFn()
Converts this
CombineFn into an equivalent
Combine.KeyedCombineFn that ignores the keys passed to it and
combines the values according to this CombineFn. |
abstract AccumT |
createAccumulator()
Returns a new, mutable accumulator value, representing the
accumulation of zero input values.
|
abstract OutputT |
extractOutput(AccumT accumulator)
Returns the output value that is the result of combining all
the input values represented by the given accumulator.
|
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. |
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(Iterable<AccumT> accumulators)
Returns an accumulator representing the accumulation of all the
input values accumulated in the merging accumulators.
|
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(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 OutputT apply(Iterable<? extends InputT> inputs)
CombineFn to a collection of input values
to produce a combined output value.
Useful when testing the behavior of a CombineFn
separately from a Combine transform.
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
Coder 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.
CannotProvideCoderExceptionpublic Coder<OutputT> getDefaultOutputCoder(CoderRegistry registry, Coder<InputT> inputCoder) throws CannotProvideCoderException
Coder 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.
CannotProvideCoderExceptionpublic <K> Combine.KeyedCombineFn<K,InputT,AccumT,OutputT> asKeyedFn()
CombineFn into an equivalent
Combine.KeyedCombineFn that ignores the keys passed to it and
combines the values according to this CombineFn.K - the type of the (ignored) keys