hive 类似oracle的 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);
	}
}

　　

 

参考：

https://blog.csdn.net/zhoudetiankong/article/details/24291797?utm_medium=distribute.pc_aggpage_search_result.none-task-blog-2~all~first_rank_v2~rank_v25-4-24291797.nonecase&utm_term=hive%20%E6%94%AF%E6%8C%81decode