hive
Funkcje agregujące zdefiniowane przez użytkownika (UDAF)
Szukaj…
Średni przykład UDAF
Utwórz klasę Java, która rozszerza
org.apache.hadoop.hive.ql.exec.hive.UDAF
Utwórz wewnętrzną klasę, która implementujeUDAFEvaluator
Wdróż pięć metod
-
init()
- Ta metoda inicjuje ewaluator i resetuje jego stan wewnętrzny. Używamy nowej kolumny () w poniższym kodzie, aby wskazać, że żadne wartości nie zostały jeszcze zagregowane. -
iterate()
- Ta metoda jest wywoływana za każdym razem, gdy pojawia się nowa wartość do agregacji. Oceniający powinien zaktualizować swój stan wewnętrzny w wyniku wykonania agregacji (wykonujemy sumę - patrz poniżej). Zwracamy wartość true, aby wskazać, że dane wejściowe były prawidłowe. -
terminatePartial()
- Ta metoda jest wywoływana, gdy Hive chce wyniku częściowej agregacji. Metoda musi zwrócić obiekt, który zawiera stan agregacji. -
merge()
- Ta metoda jest wywoływana, gdy Hive decyduje się połączyć jedną częściową agregację z drugą. -
terminate()
- Ta metoda jest wywoływana, gdy potrzebny jest końcowy wynik agregacji.
-
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
Licencjonowany na podstawie CC BY-SA 3.0
Nie związany z Stack Overflow