Hive筆記九:Hive輸入輸出適配類(CSV,XML)

在最初使用 hive ,應該說上手還是挺快的。 Hive 提供的類 SQL 語句與 mysql 語句極爲相似,語法上有大量相同的地方,這給我們上手帶來了很大的方便,但是要得心應手地寫好這些語句,還需要對 hive 有較好的瞭解,才能結合 hive 特色寫出精妙的語句。

關於 hive 語言的詳細語法可參考官方 wiki 的語言手冊http://wiki.apache.org/hadoop/Hive/LanguageManual

雖然語法風格爲我們提供了便利,但初次使用遇到的問題還是不少的,下面針對業務場景談談我們遇到的問題,和對 hive 功能的定製。

1、 分隔符問題

首先遇到的是日誌數據的分隔符問題,我們的日誌數據的大致格式如下:

2010-05-24 00:00:02@$_$@QQ2010@$_$@all@$_$@NOKIA_1681C@$_$@1@$_$@10@$_$@@$_$@-1@$_$@10@$_$@application@$_$@1

從格式可見其分隔符是“ @$_$@ ”,這是爲了儘可能防止日誌正文出現與分隔符相同的字符而導致數據混淆。本來 hive支持在建表的時候指定自定義分隔符的,但經過多次測試發現只支持單個字符的自定義分隔符,像“ @$_$@ ”這樣的分隔符是不能被支持的,但是我們可以通過對分隔符的定製解決這個問題, hive 的內部分隔符是“ \001 ”,只要把分隔符替換成“\001 ”即可。

經過探索我們發現有兩條途徑解決這個問題。

a)自定義 outputformat 和 inputformat 。

Hive 的 outputformat/inputformat 與 hadoop 的 outputformat/inputformat 相當類似, inputformat 負責把輸入數據進行格式化,然後提供給 hive , outputformat 負責把 hive 輸出的數據重新格式化成目標格式再輸出到文件,這種對格式進行定製的方式較爲底層,對其進行定製也相對簡單,重寫 InputFormat 中 RecordReader 類中的 next 方法即可,示例代碼如下:

  1. public boolean next(LongWritable key, BytesWritable value)
  2.         throws IOException {

  3.         while ( reader .next(key, text ) ) {

  4.         String strReplace = text .toString().toLowerCase().replace( "@$_$@" , "\001" );

  5.         Text txtReplace = new Text();

  6.         txtReplace.set(strReplace );

  7.         value.set(txtReplace.getBytes(), 0, txtReplace.getLength());

  8.         return true ;

  9.       }

  10.          return false ;

  11. }

  12.         重寫 HiveIgnoreKeyTextOutputFormat 中 RecordWriter 中的 write 方法,示例代碼如下:

  13.     public void write (Writable w) throws IOException {

  14.       String strReplace = ((Text)w).toString().replace( "\001" , "@$_$@" );

  15.       Text txtReplace = new Text();

  16.       txtReplace.set(strReplace);

  17.       byte [] output = txtReplace.getBytes();

  18.       bytesWritable .set(output, 0, output. length );

  19.       writer .write( bytesWritable );

  20. }
複製代碼

自定義 outputformat/inputformat 後,在建表時需要指定 outputformat/inputformat ,如下示例:

stored as INPUTFORMAT 'com.aspire.search.loganalysis.hive.SearchLogInputFormat' OUTPUTFORMAT 'com.aspire.search.loganalysis.hive.SearchLogOutputFormat'

b) 通過 SerDe(serialize/deserialize) ,在數據序列化和反序列化時格式化數據。

這種方式稍微複雜一點,對數據的控制能力也要弱一些,它使用正則表達式來匹配和處理數據,性能也會有所影響。但它的優點是可以自定義表屬性信息 SERDEPROPERTIES ,在 SerDe 中通過這些屬性信息可以有更多的定製行爲。

2、 數據導入導出

a) 多版本日誌格式的兼容

由於 hive 的應用場景主要是處理冷數據(只讀不寫),因此它只支持批量導入和導出數據,並不支持單條數據的寫入或更新,所以如果要導入的數據存在某些不太規範的行,則需要我們定製一些擴展功能對其進行處理。

我們需要處理的日誌數據存在多個版本,各個版本每個字段的數據內容存在一些差異,可能版本 A 日誌數據的第二個列是搜索關鍵字,但版本 B 的第二列卻是搜索的終端類型,如果這兩個版本的日誌直接導入 hive 中,很明顯數據將會混亂,統計結果也不會正確。我們的任務是要使多個版本的日誌數據能在 hive 數據倉庫中共存,且表的 input/output 操作能夠最終映射到正確的日誌版本的正確字段。

這裏我們不關心這部分繁瑣的工作,只關心技術實現的關鍵點,這個功能該在哪裏實現才能讓 hive 認得這些不同格式的數據呢?經過多方嘗試,在中間任何環節做這個版本適配都將導致複雜化,最終這個工作還是在 inputformat/outputformat 中完成最爲優雅,畢竟 inputformat 是源頭, outputformat 是最終歸宿。具體來說,是在前面提到的 inputformat 的 next 方法中和在 outputformat 的 write 方法中完成這個適配工作。

b) Hive 操作本地數據

一開始,總是把本地數據先傳到 HDFS ,再由 hive 操作 hdfs 上的數據,然後再把數據從 HDFS 上傳回本地數據。後來發現大可不必如此, hive 語句都提供了“ local ”關鍵字,支持直接從本地導入數據到 hive ,也能從 hive 直接導出數據到本地,不過其內部計算時當然是用 HDFS 上的數據,只是自動爲我們完成導入導出而已。

3、 數據處理

日誌數據的統計處理在這裏反倒沒有什麼特別之處,就是一些 SQL 語句而已,也沒有什麼高深的技巧,不過還是列舉一些語句示例,以示 hive 處理數據的方便之處,並展示 hive 的一些用法。

a) 爲 hive 添加用戶定製功能,自定義功能都位於 hive_contrib.jar 包中

add jar /opt/hadoop/hive-0.5.0-bin/lib/hive_contrib.jar;

b)  統計每個關鍵詞的搜索量,並按搜索量降序排列,然後把結果存入表 keyword_20100603 中

create table keyword_20100603 as select keyword,count(keyword) as count from searchlog_20100603 group by keyword order by count desc;

c) 統計每類用戶終端的搜索量,並按搜索量降序排列,然後把結果存入表 device_20100603 中

create table device_20100603 as select device,count(device) as count from searchlog_20100603 group by device order by count desc;

d) 創建表 time_20100603 ,使用自定義的 INPUTFORMAT 和 OUTPUTFORMAT ,並指定表數據的真實存放位置在 '/LogAnalysis/results/time_20100603' ( HDFS 路徑),而不是放在 hive 自己的數據目錄中

create external table if not exists time_20100603(time string, count int) stored as INPUTFORMAT 'com.aspire.search.loganalysis.hive.XmlResultInputFormat' OUTPUTFORMAT 'com.aspire.search.loganalysis.hive.XmlResultOutputFormat' LOCATION '/LogAnalysis/results/time_20100603';

e) 統計每秒訪問量 TPS ,按訪問量降序排列,並把結果輸出到表 time_20100603 中,這個表我們在上面剛剛定義過,其真實位置在 '/LogAnalysis/results/time_20100603' ,並且由於 XmlResultOutputFormat 的格式化,文件內容是 XML 格式。

insert overwrite table time_20100603 select time,count(time) as count from searchlog_20100603 group by time order by count desc;

f) 計算每個搜索請求響應時間的最大值,最小值和平均值

insert overwrite table response_20100603 select max(responsetime) as max,min(responsetime) as min,avg(responsetime) as avg from searchlog_20100603;

g)創建一個表用於存放今天與昨天的關鍵詞搜索量和增量及其增量比率,表數據位於 '/LogAnalysis/results/keyword_20100604_20100603' ,內容將是 XML 格式。

create external table if not exists keyword_20100604_20100603(keyword string, count int, increment int, incrementrate double) stored as INPUTFORMAT 'com.aspire.search.loganalysis.hive.XmlResultInputFormat' OUTPUTFORMAT 'com.aspire.search.loganalysis.hive.XmlResultOutputFormat' LOCATION '/LogAnalysis/results/keyword_20100604_20100603';

h)設置表的屬性,以便 XmlResultInputFormat 和 XmlResultOutputFormat 能根據 output.resulttype 的不同內容輸出不同格式的 XML 文件。

alter table keyword_20100604_20100603 set tblproperties ('output.resulttype'='keyword');

i) 關聯今天關鍵詞統計結果表( keyword_20100604 )與昨天關鍵詞統計結果表( keyword_20100603 ),統計今天與昨天同時出現的關鍵詞的搜索次數,今天相對昨天的增量和增量比率,並按增量比率降序排列,結果輸出到剛剛定義的 keyword_20100604_20100603 表中,其數據文件內容將爲 XML 格式。

insert overwrite table keyword_20100604_20100603 select cur.keyword, cur.count, cur.count-yes.count as increment, (cur.count-yes.count)/yes.count as incrementrate from keyword_20100604 cur join keyword_20100603 yes on (cur.keyword = yes.keyword) order by incrementrate desc;

4、用戶自定義函數 UDF

部分統計結果需要以 CSV 的格式輸出,對於這類文件體全是有效內容的文件,不需要像 XML 一樣包含 version , encoding 等信息的文件頭,最適合用 UDF(user define function) 了。

UDF 函數可直接應用於 select 語句,對查詢結構做格式化處理之後,再輸出內容。自定義 UDF 需要繼承 org.apache.hadoop.hive.ql.exec.UDF ,並實現 evaluate 函數, Evaluate 函數支持重載,還支持可變參數。我們實現了一個支持可變字符串參數的 UDF ,支持把 select 得出的任意個數的不同類型數據轉換爲字符串後,按 CSV 格式輸出,由於代碼較簡單,這裏給出源碼示例:

  1. public String evaluate(String... strs) {
  2.        StringBuilder sb = new StringBuilder();

  3.        for ( int i = 0; i < strs. length ; i++) {

  4.            sb.append(ConvertCSVField(strs[i])).append( ',' );

  5.        }

  6.        sb.deleteCharAt(sb.length()-1);

  7.        return sb.toString();

  8. }
複製代碼

需要注意的是,要使用 UDF 功能,除了實現自定義 UDF 外,還需要加入包含 UDF 的包,示例:

add jar /opt/hadoop/hive-0.5.0-bin/lib/hive_contrib.jar;

然後創建臨時方法,示例:

CREATE TEMPORARY FUNCTION Result2CSv AS ‘com.aspire.search.loganalysis.hive. Result2CSv';

使用完畢還要 drop 方法,示例:

DROP TEMPORARY FUNCTION Result2CSv;

5、輸出 XML 格式的統計結果

前面看到部分日誌統計結果輸出到一個表中,藉助 XmlResultInputFormat 和 XmlResultOutputFormat 格式化成 XML 文件,考慮到創建這個表只是爲了得到 XML 格式的輸出數據,我們只需實現 XmlResultOutputFormat 即可,如果還要支持 select 查詢,則我們還需要實現 XmlResultInputFormat ,這裏我們只介紹 XmlResultOutputFormat 。

前面介紹過,定製 XmlResultOutputFormat 我們只需重寫 write 即可,這個方法將會把 hive 的以 ’\001’ 分隔的多字段數據格式化爲我們需要的 XML 格式,被簡化的示例代碼如下:

  1. public void write(Writable w) throws IOException {
  2.            String[] strFields = ((Text) w).toString().split( "\001" );

  3.            StringBuffer sbXml = new StringBuffer();

  4.            if ( strResultType .equals( "keyword" )) {

  5.     sbXml.append( "<record><keyword>" ).append(strFields[0]).append(

  6.     "</keyword><count>" ).append(strFields[1]).append(           "</count><increment>" ).append(strFields[2]).append(

  7.     "</increment><rate>" ).append(strFields[3]).append(

  8. "</rate></result>" );

  9.            }

  10.            Text txtXml = new Text();

  11.            byte [] strBytes = sbXml.toString().getBytes( "utf-8" );

  12.            txtXml.set(strBytes, 0, strBytes. length );

  13.            byte [] output = txtXml.getBytes();

  14.            bytesWritable .set(output, 0, output. length );

  15.            writer .write( bytesWritable );

  16.     }
複製代碼

其中的 strResultType .equals( "keyword" ) 指定關鍵詞統計結果,這個屬性來自以下語句對結果類型的指定,通過這個屬性我們還可以用同一個 outputformat 輸出多種類型的結果。

alter table keyword_20100604_20100603 set tblproperties ('output.resulttype'='keyword');

仔細看看 write 函數的實現便可發現,其實這裏只輸出了 XML 文件的正文,而 XML 的文件頭和結束標籤在哪裏輸出呢?所幸我們採用的是基於 outputformat 的實現,我們可以在構造函數輸出 version , encoding 等文件頭信息,在 close() 方法中輸出結束標籤。

這也是我們爲什麼不使用 UDF 來輸出結果的原因,自定義 UDF 函數不能輸出文件頭和文件尾,對於 XML 格式的數據無法輸出完整格式,只能輸出 CSV 這類所有行都是有效數據的文件。

五、總結

Hive 是一個可擴展性極強的數據倉庫工具,藉助於 hadoop 分佈式存儲計算平臺和 hive 對 SQL 語句的理解能力,我們所要做的大部分工作就是輸入和輸出數據的適配,恰恰這兩部分 IO 格式是千變萬化的,我們只需要定製我們自己的輸入輸出適配器, hive將爲我們透明化存儲和處理這些數據,大大簡化我們的工作。本文的重心也正在於此,這部分工作相信每一個做數據分析的朋友都會面對的,希望對您有益。

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