hive
Benutzerdefinierte Aggregatfunktionen (UDAF)
Suche…
UDAF bedeutet Beispiel
Erstellen Sie eine Java-Klasse, die
org.apache.hadoop.hive.ql.exec.hive.UDAF
erweitert.org.apache.hadoop.hive.ql.exec.hive.UDAF
Sie eine innere Klasse, dieUDAFEvaluator
implementiertImplementiere fünf Methoden
-
init()
- Diese Methode initialisiert den Evaluator und setzt seinen internen Zustand zurück. Wir verwenden new Column () im folgenden Code, um anzuzeigen, dass noch keine Werte aggregiert wurden. -
iterate()
- Diese Methode wird jedes Mal aufgerufen, wenn ein neuer Wert aggregiert wird. Der Evaluator sollte seinen internen Zustand mit dem Ergebnis der Aggregation aktualisieren (wir machen eine Summe - siehe unten). Wir geben true zurück, um anzuzeigen, dass die Eingabe gültig war. -
terminatePartial()
- Diese Methode wird aufgerufen, wenn Hive ein Ergebnis für die Teilaggregation wünscht. Die Methode muss ein Objekt zurückgeben, das den Aggregationsstatus einkapselt. -
merge()
- Diese Methode wird aufgerufen, wenn Hive eine Teilaggregation mit einer anderen kombiniert. -
terminate()
- Diese Methode wird aufgerufen, wenn das Endergebnis der Aggregation benötigt wird.
-
public class MeanUDAF extends UDAF {
// Define Logging
static final Log LOG = LogFactory.getLog(MeanUDAF.class.getName());
public static class MeanUDAFEvaluator implements UDAFEvaluator {
/**
* Use Column class to serialize intermediate computation
* This is our groupByColumn
*/
public static class Column {
double sum = 0;
int count = 0;
}
private Column col = null;
public MeanUDAFEvaluator() {
super();
init();
}
// A - Initalize evaluator - indicating that no values have been
// aggregated yet.
public void init() {
LOG.debug("Initialize evaluator");
col = new Column();
}
// B- Iterate every time there is a new value to be aggregated
public boolean iterate(double value) throws HiveException {
LOG.debug("Iterating over each value for aggregation");
if (col == null)
throw new HiveException("Item is not initialized");
col.sum = col.sum + value;
col.count = col.count + 1;
return true;
}
// C - Called when Hive wants partially aggregated results.
public Column terminatePartial() {
LOG.debug("Return partially aggregated results");
return col;
}
// D - Called when Hive decides to combine one partial aggregation with another
public boolean merge(Column other) {
LOG.debug("merging by combining partial aggregation");
if(other == null) {
return true;
}
col.sum += other.sum;
col.count += other.count;
return true;
}
// E - Called when the final result of the aggregation needed.
public double terminate(){
LOG.debug("At the end of last record of the group - returning final result");
return col.sum/col.count;
}
}
}
hive> CREATE TEMPORARY FUNCTION <FUNCTION NAME> AS 'JAR PATH.jar';
hive> select id, mean_udf(amount) from table group by id;
Modified text is an extract of the original Stack Overflow Documentation
Lizenziert unter CC BY-SA 3.0
Nicht angeschlossen an Stack Overflow