com.yahoo.sketches.pig.tuple.UnionSketchAlgebraicIntermediateFinal Maven / Gradle / Ivy
/*
* Copyright 2016, Yahoo! Inc.
* Licensed under the terms of the Apache License 2.0. See LICENSE file at the project root for terms.
*/
package com.yahoo.sketches.pig.tuple;
import static com.yahoo.sketches.Util.DEFAULT_NOMINAL_ENTRIES;
import java.io.IOException;
import org.apache.log4j.Logger;
import org.apache.pig.EvalFunc;
import org.apache.pig.data.DataBag;
import org.apache.pig.data.DataByteArray;
import org.apache.pig.data.Tuple;
import com.yahoo.sketches.tuple.Sketch;
import com.yahoo.sketches.tuple.Summary;
import com.yahoo.sketches.tuple.SummaryFactory;
import com.yahoo.sketches.tuple.Union;
/**
* This is to calculate the intermediate pass (combiner) or the final pass
* (reducer) of an Algebraic sketch operation. This may be called multiple times
* (from the mapper and from the reducer). It will receive a bag of values
* returned by either the Intermediate or the Initial stages, so
* it needs to be able to differentiate between and interpret both types.
*
* @param Type of Summary
*/
public abstract class UnionSketchAlgebraicIntermediateFinal extends EvalFunc {
private final int sketchSize_;
private final SummaryFactory summaryFactory_;
private boolean isFirstCall_ = true;
/**
* Constructs a function given a summary factory and default sketch size
* @param summaryFactory an instance of SummaryFactory
*/
public UnionSketchAlgebraicIntermediateFinal(final SummaryFactory summaryFactory) {
this(DEFAULT_NOMINAL_ENTRIES, summaryFactory);
}
/**
* Constructs a function given a sketch size and summary factory
* @param sketchSize parameter controlling the size of the sketch and the accuracy.
* It represents nominal number of entries in the sketch. Forced to the nearest power of 2
* greater than given value.
* @param summaryFactory an instance of SummaryFactory
*/
public UnionSketchAlgebraicIntermediateFinal(final int sketchSize, final SummaryFactory summaryFactory) {
this.sketchSize_ = sketchSize;
this.summaryFactory_ = summaryFactory;
}
@Override
public Tuple exec(final Tuple inputTuple) throws IOException {
if (isFirstCall_) {
// this is to see in the log which way was used by Pig
Logger.getLogger(getClass()).info("algebraic is used");
isFirstCall_ = false;
}
final Union union = new Union(sketchSize_, summaryFactory_);
final DataBag bag = (DataBag) inputTuple.get(0);
if (bag == null) {
throw new IllegalArgumentException("InputTuple.Field0: Bag may not be null");
}
for (final Tuple dataTuple: bag) {
final Object item = dataTuple.get(0);
if (item instanceof DataBag) {
// this is from a prior call to the initial function, so there is a nested bag.
for (Tuple innerTuple: (DataBag) item) {
final Sketch incomingSketch = Util.deserializeSketchFromTuple(innerTuple);
union.update(incomingSketch);
}
} else if (item instanceof DataByteArray) {
// This is a sketch from a call to the Intermediate function
// Add it to the current union.
final Sketch incomingSketch = Util.deserializeSketchFromTuple(dataTuple);
union.update(incomingSketch);
} else {
// we should never get here.
throw new IllegalArgumentException(
"InputTuple.Field0: Bag contains unrecognized types: " + item.getClass().getName());
}
}
return Util.tupleFactory.newTuple(new DataByteArray(union.getResult().toByteArray()));
}
}