Welcome 微信登录

首页 / 数据库 / MySQL / 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 检测输入的参数类型,来判断。相关阅读:基于Hadoop集群的Hive安装 http://www.linuxidc.com/Linux/2013-07/87952.htmHive内表和外表的区别 http://www.linuxidc.com/Linux/2013-07/87313.htmHadoop + Hive + Map +reduce 集群安装部署 http://www.linuxidc.com/Linux/2013-07/86959.htmHive本地独立模式安装 http://www.linuxidc.com/Linux/2013-06/86104.htmHive学习之WordCount单词统计 http://www.linuxidc.com/Linux/2013-04/82874.htm-------------------------------分割线-------------------------------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);
 }
}Hive 的详细介绍:请点这里
Hive 的下载地址:请点这里本文永久更新链接地址