Hive自定义函数（字母大小写转换）

亦凉 2022-05-29 07:30 184阅读 0赞

> 自定义Hive的函数，首先需要继承UDF类，然后定义方法名字为evaluate的方法，该方法不能设置void，如果需要返回为空，可以返回为null.

**一个简单的大小写转换案例**  
**特别注意看注释**

package com.madman.hive.function;
    
    import org.apache.hadoop.hive.ql.exec.UDF;
    import org.apache.hadoop.io.IntWritable;
    import org.apache.hadoop.io.Text;
    
    /** * 注释很重要。。 * 看注释。 * 看注释 * 看英文注释。。。。。。。。。 * */
    /** * A User-defined function (UDF) for the use with Hive. * * New UDF classes need to inherit from this UDF class. * * Required for all UDF classes: 1. Implement one or more methods named * "evaluate" which will be called by Hive. The following are some examples: * public int evaluate(); public int evaluate(int a); public double evaluate(int * a, double b); public String evaluate(String a, int b, String c); * * "evaluate" should never be a void method. However it can return "null" if * needed. */
    
    /** * 需求：转换大小写，0代表小写，1代表大写，默认转换小写 * * @author ibf * */
    public class HiveFunction extends UDF { 
        public Text evaluate(Text str) {
            return this.evaluate(str, new IntWritable(0));
        }
    
        public Text evaluate(Text str, IntWritable flag) {
    
            // 如果不等于null，就进行转换大小写的操作，说明数据合法
            if (str != null) {
                // 如果等于0，表示转换小写
                if (flag.get() == 0) {
                    return new Text(str.toString().toLowerCase());
                    // 如果等于1，表示转换大写
                } else if (flag.get() == 1) {
                    return new Text(str.toString().toUpperCase());
                } else
                    return null;
            } else
                return null;
        }
    
        public static void main(String[] args) {
    
            System.out.println(new HiveFunction().evaluate(new Text("Hadoop"), new IntWritable(0)));
        }
    
    }

**大概步骤**  
`继承UDF类，import org.apache.hadoop.hive.ql.exec.UDF;`  
`写完程序，打jar包(指定main类)上传到Linux系统中与jar包进行关联 add jar /opt/moduels/alldata/udf.jar;`  
`创建function函数方法 create temporary function my_udf as 'com.madman.hive.function.HiveFunction';后面这个就是你的类路劲`  
`测试：执行将emp表中的ename全部转换成小写 select ename,my_udf(ename) low_ename from emp;`