Hive--基础--8.4 自定义 UDF 函数

0）需求:

1）创建一个 Maven 工程 Hive

2）导入依赖：

创建的pom.xml文件配置信息：

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>

    <groupId>com.atguigu</groupId>
    <artifactId>Hive</artifactId>
    <version>1.0-SNAPSHOT</version>

    <dependencies>
        <dependency>
            <groupId>org.apache.hive</groupId>
            <artifactId>hive-exec</artifactId>
            <version>3.1.2</version>
        </dependency>
        <dependency>
            <groupId>org.apache.hadoop</groupId>
            <artifactId>hadoop-client</artifactId>
            <version>3.1.3</version>
        </dependency>
    </dependencies>

    <properties>
        <maven.compiler.source>8</maven.compiler.source>
        <maven.compiler.target>8</maven.compiler.target>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    </properties>

</project>

3）创建一个类

import org.apache.hadoop.hive.ql.exec.UDFArgumentException;
import org.apache.hadoop.hive.ql.exec.UDFArgumentLengthException;
import org.apache.hadoop.hive.ql.exec.UDFArgumentTypeException;
import org.apache.hadoop.hive.ql.metadata.HiveException;
import org.apache.hadoop.hive.ql.udf.generic.GenericUDF;
import org.apache.hadoop.hive.serde2.objectinspector.ObjectInspector;
import org.apache.hadoop.hive.serde2.objectinspector.primitive.PrimitiveObjectInspectorFactory;
/**
 * 自定义 UDF 函数，需要继承 GenericUDF 类
 * 需求: 计算指定字符串的长度
 */
public class MyStringLength extends GenericUDF {
	/**
	 *
	 * @param arguments 输入参数类型的鉴别器对象
	 * @return 返回值类型的鉴别器对象
	 * @throws UDFArgumentException
	 */
	@Override
	public ObjectInspector initialize(ObjectInspector[] arguments) throws UDFArgumentException {// 判断输入参数的个数
		if(arguments.length !=1){
			throw new UDFArgumentLengthException("Input Args Length Error!!!");
		}
		// 判断输入参数的类型

		if(!arguments[0].getCategory().equals(ObjectInspector.Category.PRIMITIVE)
		){
			throw new UDFArgumentTypeException(0,"Input Args Type Error!!!");
		}
		//函数本身返回值为 int，需要返回 int 类型的鉴别器对象
		return PrimitiveObjectInspectorFactory.javaIntObjectInspector;
	}
	/**
	 * 函数的逻辑处理
	 * @param arguments 输入的参数
	 * @return 返回值
	 * @throws HiveException
	 */
	@Override
	public Object evaluate(DeferredObject[] arguments) throws
			HiveException {
		if(arguments[0].get() == null){
			return 0;
		}
		return arguments[0].get().toString().length();
	}
	@Override
	public String getDisplayString(String[] children) {
		return "";
	}
}

4）打成 jar 包上传到服务器/opt/module/data/myudf.jar

这里电脑出了一点问题，无法用xshell直接传送文件，因此就不先演示了。。。

接下来的步骤如下：