hive中UDF跟UDAF使用說明

Hive進行UDF開發十分簡單,此處所說UDF爲Temporary的function,所以需要hive版本在0.4.0以上纔可以。

一、背景:

1、Hive是基於Hadoop中的MapReduce,提供HQL查詢的數據倉庫。Hive是一個很開放的系統,很多內容都支持用戶定製,包括:

a)文件格式:Text FileSequence File

b)內存中的數據格式: Java Integer/String, Hadoop IntWritable/Text

c)用戶提供的 map/reduce 腳本:不管什麼語言,利用 stdin/stdout 傳輸數據

d)用戶自定義函數: Substr, Trim, 1 – 1

e)用戶自定義聚合函數: Sum, Average…… n – 1

2、定義:UDF(User-Defined-Function),用戶自定義函數對數據進行處理。

二、用法

1UDF函數可以直接應用於select語句,對查詢結構做格式化處理後,再輸出內容。

2、編寫UDF函數的時候需要注意一下幾點:

a)自定義UDF需要繼承org.apache.hadoop.hive.ql.UDF

b)需要實現evaluate函。

cevaluate函數支持重載。

3、以下是兩個數求和函數的UDFevaluate函數代表兩個整型數據相加,兩個浮點型數據相加,可變長數據相加

    Hive的UDF開發只需要重構UDF類的evaluate函數即可。例:

package hive.connect;

import org.apache.hadoop.hive.ql.exec.UDF;

public final class Add extends UDF {

public Integer evaluate(Integer a, Integer b) {

               if (null == a || null == b) {

                               return null;

               } return a + b;

}

public Double evaluate(Double a, Double b) {

               if (a == null || b == null)

                               return null;

                               return a + b;

               }

public Integer evaluate(Integer... a) {

               int total = 0;

               for (int i = 0; i < a.length; i++)

                               if (a[i] != null)

                                              total += a[i];

                                              return total;

                               }

}

4、步驟

a)把程序打包放到目標機器上去;

b)進入hive客戶端,添加jar包:hive>add jar /run/jar/udf_test.jar;

c)創建臨時函數:hive>CREATE TEMPORARY FUNCTION add_example AS 'hive.udf.Add';

d)查詢HQL語句:

SELECT add_example(8, 9) FROM scores;

SELECT add_example(scores.math, scores.art) FROM scores;

SELECT add_example(6, 7, 8, 6.8) FROM scores;

e)銷燬臨時函數:hive> DROP TEMPORARY FUNCTION add_example;

5、細節在使用UDF的時候,會自動進行類型轉換,例如:

SELECT add_example(8,9.1) FROM scores;

注:

1.   UDF只能實現一進一出的操作,如果需要實現多進一出,則需要實現UDAF

下面來看下UDAF:

1Hive查詢數據時,有些聚類函數在HQL沒有自帶,需要用戶自定義實現。

2、用戶自定義聚合函數: Sum, Average…… n – 1

UDAFUser- Defined Aggregation Funcation

一、用法

1、一下兩個包是必須的import org.apache.hadoop.hive.ql.exec.UDAForg.apache.hadoop.hive.ql.exec.UDAFEvaluator

2、函數類需要繼承UDAF類,內部類EvaluatorUDAFEvaluator接口。

3Evaluator需要實現 inititerateterminatePartialmergeterminate這幾個函數。

ainit函數實現接口UDAFEvaluatorinit函數。

biterate接收傳入的參數,並進行內部的輪轉。其返回類型爲boolean

cterminatePartial無參數,其爲iterate函數輪轉結束後,返回輪轉數據,terminatePartial類似於hadoopCombiner

dmerge接收terminatePartial的返回結果,進行數據merge操作,其返回類型爲boolean

eterminate返回最終的聚集函數結果。

package hive.udaf;

import org.apache.hadoop.hive.ql.exec.UDAF;

import org.apache.hadoop.hive.ql.exec.UDAFEvaluator;

public class Avg extends UDAF {

         public static class AvgState {

         private long mCount;

         private double mSum;

}

public static class AvgEvaluator implements UDAFEvaluator {

         AvgState state;

         public AvgEvaluator() {

                   super();

                   state = new AvgState();

                   init();

}

/** * init函數類似於構造函數,用於UDAF的初始化 */

public void init() {

         state.mSum = 0;

         state.mCount = 0;

}

/** * iterate接收傳入的參數,並進行內部的輪轉。其返回類型爲boolean * * @param o * @return */

public boolean iterate(Double o) {

         if (o != null) {

                   state.mSum += o;

                   state.mCount++;

         } return true;

}

/** * terminatePartial無參數,其爲iterate函數輪轉結束後,返回輪轉數據, * terminatePartial類似於hadoopCombiner * * @return */

public AvgState terminatePartial() {

         // combiner

         return state.mCount == 0 ? null : state;

}

/** * merge接收terminatePartial的返回結果,進行數據merge操作,其返回類型爲boolean * * @param o * @return */

public boolean terminatePartial(Double o) {                

         if (o != null) {

                   state.mCount += o.mCount;

                   state.mSum += o.mSum;

         }

         return true;

}

/** * terminate返回最終的聚集函數結果 * * @return */

public Double terminate() {

         return state.mCount == 0 ? null : Double.valueOf(state.mSum / state.mCount);

}

}

5、執行求平均數函數的步驟

a)將java文件編譯成Avg_test.jar。

b)進入hive客戶端添加jar包:

hive>add jar /run/jar/Avg_test.jar。

c)創建臨時函數:

hive>create temporary function avg_test 'hive.udaf.Avg';

d)查詢語句:

hive>select avg_test(scores.math) from scores;

e)銷燬臨時函數:

hive>drop temporary function avg_test;

總結

1、重載evaluate函數。

2、UDF函數中參數類型可以爲Writable,也可爲java中的基本數據對象。

3、UDF支持變長的參數。

4、Hive支持隱式類型轉換。

5、客戶端退出時,創建的臨時函數自動銷燬。

6、evaluate函數必須要返回類型值,空的話返回null,不能爲void類型。

7、UDF是基於單條記錄的列進行的計算操作,而UDFA則是用戶自定義的聚類函數,是基於表的所有記錄進行的計算操作。

8、UDF和UDAF都可以重載。

9、查看函數

SHOW FUNCTIONS; 

DESCRIBE FUNCTION <function_name>;

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章