HIVE自定义函数

发布时间:2023-09-04 14:30

Hive自定义函数

一. UDF(user defined function)

  • 背景

  • 系统内置函数无法解决所有的实际业务问题,需要开发者自己编写函数实现自身的业务实现诉求。

  • 应用场景非常多,面临的业务不同导致个性化实现很多,故udf很需要。

  • 意义

  • 函数扩展得到解决,极大丰富了可定制化的业务需求。

  • IO要求-要解决的问题

  • in:out=1:1,只能输入一条记录当中的数据,同时返回一条处理结果。

  • 属于最常见的自定义函数,像cos,sin,substring,instr等均是如此要求。

  • 实现步骤(Java创建自定义UDF类)

  • 自定义一个java类

  • 继承UDF类

  • 约定俗成的重写evaluate方法

  • 打包类所在项目成一个all-in-one的jar包并上传到hive所在机器

  • 在hive中执行add jar操作,将jar加载到classpath中。

  • 在hive中创建模板函数,使得后边可以使用该函数名称调用实际的udf函数

  • hive sql中像调用系统函数一样使用udf函数

  • 代码实现

  • 功能要求:实现当输入字符串超过2个字符的时候,多余的字符以"..."来表示。

  • 如“12”则返回“12”,如“123”返回“12..."

  • 自定义类、继承UDF、约定俗成的“重写”evaluate方法已在代码中体现

package com.tianliangedu.hive.udf;

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

/*

 * 功能:实现当输入字符串超过2个字符的时候,多余的字符以"..."来表示。

 * 输入/输出:* 如“12”则返回“12”,如“123”返回“12..."

 */

public class ValueMaskUDF extends UDF{

       public String evaluate(String input,int maxSaveStringLength,String replaceSign) {

             if(input.length()<=maxSaveStringLength){

                    return input;

             }

             return input.substring(0,maxSaveStringLength)+replaceSign;

       }

       public static void main(String[] args) {

             System.out.println(new ValueMaskUDF().evaluate("河北省",2,"..."));
  • 布署步骤

  • maven管理

采用maven管理打包的方式,maven pom配置项为:



    4.0.0

    com.tianliangedu.course

    TlHadoopCore

    0.0.1-SNAPSHOT



    

      UTF-8

        UTF-8

    

    

    

        

            nexus-aliyun

            Nexus aliyun

            http://maven.aliyun.com/nexus/content/groups/public

        

    

                

    

        

        

            org.apache.hadoop

            hadoop-client

            2.7.4

            provided

        

             

        

            org.apache.hive

            hive-cli

            1.2.1

            provided

                 

    

    

        TlHadoopCore

        

            

                maven-assembly-plugin

                

                    

                        jar-with-dependencies

                    

                

                

                    

                        make-assembly

                        package

                        

                            assembly

                        

                    

                

            

            

                maven-compiler-plugin

                2.3.2

                

                    1.7

                    1.7

                    UTF-8

                

            

        

    

  • maven打包操作

  • 上传jar包至hive操作环境中

  • 进入到自己的所操作的hive环境目录中。

  • rz命令上传至服务器上

  • 加载jar包、声明函数、使用函数

  • 加载jar包(输入hive进入到hive cli)

HIVE自定义函数_第1张图片

  • 将jar包加入hive 交互中

add jar hdfs:///home/hive/tianliangedu_course/04_udf/TlHadoopCore-jar-with-dependencies.jar; 

  • 声明函数

create temporary function mask 

as 

'com.tianliangedu.hive.udf.ValueMaskUDF';

  • 使用函数

二. UDAF(user defined aggregation function)

  • 自定义udaf函数self_count,实现系统udaf count的功能

  • in:out=n:1,即输入N条数据,返回一条处理结果,即列转行。

  • 最常见的系统聚合函数,如count,sum,avg,max等

  • 实现步骤

  • 自定义一个java类

  • 继承UDAF类

  • 内部定义一个静态类,实现UDAFEvaluator接口

  • 实现方法init,iterate,terminatePartial,merge,terminate共5个方法.

 HIVE自定义函数_第2张图片

  • 在hive中执行add jar操作,将jar加载到classpath中。

  • 在hive中创建模板函数,使得后边可以使用该函数名称调用实际的udf函数

  • hive sql中像调用系统函数一样使用udaf函数

  • 代码实现

实现与hive原生的count相似的计数功能。

  1. 如:select count(1) from tablename 或者select key,count(1) from tablename group by key;

package com.tianliangedu.hive.udaf;

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

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

import org.apache.log4j.Logger;

/**

* 自行实现sql的count操作

*/

//主类继承UDAF

public class DIYCountUDAF extends UDAF {  

    //日志对象初始化,使访类有输出日志的能力

    public static Logger logger=Logger.getLogger(DIYCountUDAF.class);

    

    //静态类实现UDAFEvaluator

    public static class Evaluator implements UDAFEvaluator {  

        //设置成员变量,存储每个统计范围内的总记录数

        private int totalRecords;  

        //初始化函数,map和reduce均会执行该函数,起到初始化所需要的变量的作用

        public Evaluator() {  

            init();  

        }  

        //初始化,初始值为0,并日志记录下相应输出

        public void init() {  

            totalRecords = 0;  

            logger.info("init totalRecords="+totalRecords);

        }  

        //map阶段,返回值为boolean类型,当为true则程序继续执行,当为false则程序退出  

        public boolean iterate(String input) {

            //当input输入不为空的时候,即为有值存在,即为存在1行,故做+1操作

            if (input != null) {  

                totalRecords += 1;  

            }  

            //输出当前组处理到第多少条数据了

            logger.info("iterate totalRecords="+totalRecords);

            return true;  

        }  

        /**

         * 类似于combiner,在map范围内做部分聚合,将结果传给merge函数中的形参mapOutput  

         * 如果需要聚合,则对iterator返回的结果处理,否则直接返回iterator的结果即可

         */

        public int terminatePartial() {  

            logger.info("terminatePartial totalRecords="+totalRecords);

            return totalRecords;  

        }

        

        // reduce 阶段,用于逐个迭代处理map当中每个不同key对应的 terminatePartial的结果

        public boolean merge(int mapOutput) {  

            totalRecords +=mapOutput;  

            logger.info("merge totalRecords="+totalRecords);

            return true;  

        }  

        //处理merge计算完成后的结果,此时的count在merge完成时候,结果已经得出,无需再进一次对整体结果做处理,故直接返回即可

        public int terminate() {  

            logger.info("terminate totalRecords="+totalRecords);

            return totalRecords;  

        }  

    }  

}    
  • 布署步骤

跟udf完全一致

  • 加载jar包、声明函数、使用函数

跟udf完全一致

  • 测试运行

与count一样,使用前边定义的临时udaf函数。

  • 案例2

自定义udaf函数,实现多条学生成绩的合并

  • 数据输

HIVE自定义函数_第3张图片

  • 数据输出

  • 代码

package com.tianliangedu.hive.udaf;

import java.util.HashMap;

import java.util.Map;

import java.util.Set;

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

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

import org.apache.log4j.Logger;

/**

* 实现多条数据合并成一条数据

*/

// 主类继承UDAF

public class StudentScoreAggUDAF extends UDAF {

    // 日志对象初始化

    public static Logger logger = Logger.getLogger(StudentScoreAggUDAF.class);

    // 静态类实现UDAFEvaluator

    public static class Evaluator implements UDAFEvaluator {

        // 设置成员变量,存储每个统计范围内的总记录数

        private Map courseScoreMap;

        

        //初始化函数,map和reduce均会执行该函数,起到初始化所需要的变量的作用

        public Evaluator() {

            init();

        }

        // 初始化函数间传递的中间变量

        public void init() {

            courseScoreMap = new HashMap();

        }

        

         //map阶段,返回值为boolean类型,当为true则程序继续执行,当为false则程序退出  

        public boolean iterate(String course, String score) {

            if (course == null || score == null) {

                return true;

            }

            courseScoreMap.put(course, score);

            return true;

        }

         /**

         * 类似于combiner,在map范围内做部分聚合,将结果传给merge函数中的形参mapOutput  

         * 如果需要聚合,则对iterator返回的结果处理,否则直接返回iterator的结果即可

         */

        public Map terminatePartial() {

            return courseScoreMap;

        }

         // reduce 阶段,用于逐个迭代处理map当中每个不同key对应的 terminatePartial的结果

        public boolean merge(Map mapOutput) {

            this.courseScoreMap.putAll(mapOutput);

            return true;

        }

        // 处理merge计算完成后的结果,即对merge完成后的结果做最后的业务处理

        public String terminate() {

            return courseScoreMap.toString();

        }

    }

}
  • 布署过程与之前相同

  • 测试脚本

select 

id,username,score_agg(course,score) 

from 

student_score 

group by id,username;

HIVE自定义函数_第4张图片

三. UDTF(User-Defined Table-Generating Functions) 

  • 解决一行输入多行输出,即1:n,即行转列应用

  • 往往被lateral view explode+udf等替代实现,比直接用udtf会更简单、直接、更灵活一些

  • 本节由学员自学实现,如何用lateral view explode+udf替代udtf实现

  • lateral view explode+udf替代udtf应用案例

  • 需求

将一个array类型按列存储的学生成绩表,转变成按行来显示,学生名字超过2个字符的,后边用"..."来代替。

  • 数据准备

学生成绩表HIVE自定义函数_第5张图片

  • 通过lateral view explode实现行转列

select id,name,score 

from test_array 

lateral view 

explode(score_array) score_table as score;

  • 加入udf处理业务需求

select id,mask(name,2,'...'),score 

from test_array 

lateral view 

explode(score_array) score_table as score;

HIVE自定义函数_第6张图片

四. 练习题

1. 实现UDF,实现给定任意明文密码,返回md5加密后32位的密文密码。

  • 有用户密码表

userinfo(id string,username string,password string);

  • 使用insert into插入以下数据

{id:"001",username:"张三",明文密码password:"123456"}

  • 实际效果select * from userinfo的结果如下图,使用户看不到真实的用户密码。

001    张三     XXXXXXXXXXXXXXXXX

2. 自定义UDAF名字为self_max,求取给定一整数值列表中的最大值函数max功能。

  • 数据样例

HIVE自定义函数_第7张图片

  • 系统自带的max函数使用 

select id,username,max(score) 

from student_score 

group by id,username;

HIVE自定义函数_第8张图片

3. 自定义UDAF名字为self_min,求取给定一整数值列表中的最小值函数min功能。

  • 数据样例

HIVE自定义函数_第9张图片

  • 系统自带的min函数使用 

select id,username,min(score) 

from student_score 

group by id,username;

   HIVE自定义函数_第10张图片

ItVuer - 免责声明 - 关于我们 - 联系我们

本网站信息来源于互联网,如有侵权请联系:561261067@qq.com

桂ICP备16001015号