1. 程式人生 > >hive的UDF實現類似於oracle的decode函式功能

hive的UDF實現類似於oracle的decode函式功能

     客戶提的要求,讓用hive實現類似orale的decode函式功能。好吧,開工。

    Oracle的decode函式語法:DECODE(value,if1,then1,if2,thne2,if3,then3,...else)。

    首先涉及到的問題是輸入引數的動態化,decode函式要求輸入函式是偶數個,然後實現類似if,else的邏輯判斷功能。這樣的話用到了java的Object ... args 來傳入多個引數,然後在方法中檢測個數是否符合偶數個。選用Object型別的原因就是為了引數型別的動態識別。

    為了實現對decode(1,1.0,'a',2.0,'b','c')的判斷型別,需要將Integer型別的轉換為double型別,不然的話這樣的內容就無法實現。用instanceof 檢測輸入的引數型別,來判斷。

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

public class Decode extends UDF {

	public String evaluate(Object... args) {
		if (args.length % 2 != 0) {
			System.out.println("輸入的引數個數錯誤,應為偶數");
		}
		int number = args.length;
		Object result = null;
		int flag = number - 1;

		int i = 1;

		if (args[0] instanceof Integer) {
			args[0] = Double.valueOf(Integer.valueOf(args[0].toString()));
		}
		while (i < flag) {

			if (args[i] instanceof Integer) {
				args[i] = Double.valueOf(Integer.valueOf(args[i].toString()));
			}

			if (String.valueOf(args[i]).equals(String.valueOf(args[0]))) {
				result = args[i + 1];
				break;
			} else {
				i += 2;
			}
		}
		if (result == null)
			result = args[flag];

		return String.valueOf(result);
	}
}