UDF和UDTF函数的应用

package DF;

import org.apache.commons.lang.StringUtils;
import org.apache.hadoop.hive.ql.exec.UDF;
import org.json.JSONException;
import org.json.JSONObject;

public class BaseFiledJson extends UDF {

//重写evaluate方法 返回一个string 穿进去第一个值是整个json数据 line和公共字段的所有key
public String evaluate(String line,String jsonkeysString){


    StringBuilder sb = new StringBuilder();
    //获取所有key
    String[] jsonkeys = jsonkeysString.split(",");
    //line 15015616|{}
    String[] logcontents = line.split("\\|");

    //校验进来的字符串是否是正确的,blank可以一次性校验null,""," ",
    if(logcontents.length!=2|| StringUtils.isBlank(logcontents[1])){
        return "";
    }


    try {
        //此时logcontents是一个完整的json字符串{}
        JSONObject jsonObject = new JSONObject(logcontents[1]);//timestamp|{"cm":{"k":"v","k":"v"}}
        //获取公共字段的json
        JSONObject cmjson = jsonObject.getJSONObject("cm");
        //循环遍历
        for(int i=0;i<jsonkeys.length;i++){
            String jsonkey = jsonkeys[i].trim();//获取每一个key

            if(cmjson.has(jsonkey)){

                sb.append(cmjson.get(jsonkey)).append("\t");//每个字段按照\t拼接
            }else {
                sb.append("\t");//如果没有获取到则直接\t拼接
            }
        }
        //拼接事件字段和服务器时间
        sb.append(jsonObject.getString("et")).append("\t");//得到的就是[{"ett":sdasd,"en":"asdasd"}]此时就不存在et了 只存在这个数组
        sb.append(logcontents[0]).append("\t");
    } catch (JSONException e) {
        e.printStackTrace();
    }

    return sb.toString();
}

//拿到的是公共字段和事件字段的json

}

==================================================================================================
package DF;

import org.apache.commons.lang.StringUtils;
import org.apache.hadoop.hive.ql.exec.UDFArgumentException;
import org.apache.hadoop.hive.ql.metadata.HiveException;
import org.apache.hadoop.hive.ql.udf.generic.GenericUDTF;
import org.apache.hadoop.hive.serde2.objectinspector.ObjectInspector;
import org.apache.hadoop.hive.serde2.objectinspector.ObjectInspectorFactory;
import org.apache.hadoop.hive.serde2.objectinspector.PrimitiveObjectInspector;
import org.apache.hadoop.hive.serde2.objectinspector.StructObjectInspector;
import org.apache.hadoop.hive.serde2.objectinspector.primitive.PrimitiveObjectInspectorFactory;
import org.json.JSONArray;
import org.json.JSONException;

import java.util.ArrayList;
import java.util.List;

public class EventJson extends GenericUDTF {

@Deprecated
public StructObjectInspector initialize(ObjectInspector[] argOIs) throws UDFArgumentException {
    //定义返回值的名称和返回值的类型
    List<String> fieldname=new ArrayList<>();
    List<ObjectInspector> fieldType=new ArrayList<>();

    fieldname.add("event_name");
    fieldType.add(PrimitiveObjectInspectorFactory.javaStringObjectInspector);

    fieldname.add("event_json");
    fieldType.add(PrimitiveObjectInspectorFactory.javaStringObjectInspector);

    return ObjectInspectorFactory.getStandardStructObjectInspector(fieldname,fieldType);


}


@Override//一进多出  这个object[]数组传进来的不是et了而是一个整条数组[{"ett":sdasd,"en":"asdasd"}]
public void process(Object[] objects) throws HiveException {
    //具体的业务  获取输入的数据

    String input = objects[0].toString();//因为所有的json都在一条数组里面 所以只需要拿到0下标就可以算是拿到这个数组了
    if(StringUtils.isBlank(input)){
        return ;
    }else {

        try {
            JSONArray jsonArray = new JSONArray(input);//此时拿到了json数组了,然后一个个解析拿取事件和数组内的第一条json,第二条json,第三条json等等,这些json在后续用hive函数get_jsonobject解析也可以
           //循环遍历json的值,一个一个的封装到string数组里 拿到事件名称和json值
            for(int i=0; i<jsonArray.length();i++){
               String[] result= new String[2];//存放俩个值,一个是名称一个是json数据
                result[0]= jsonArray.getJSONObject(i).getString("en");//事件名称
                 result[1]=jsonArray.getString(i);//事件整体 [{"ett":"assd"}]
            forward(result);//相当于context发送

            }
        } catch (JSONException e) {
            e.printStackTrace();
        }
    }
}

@Override
public void close() throws HiveException {

}

//获取事件的字段

}

### Hive UDF、UDAF UDTF 的实际应用案例 #### 1. 用户自定义函数 (UDF) 用户自定义函数用于处理单个输入并返回单个输出。下面是一个将字符串转换为大写的例子: ```java import org.apache.hadoop.hive.ql.exec.Description; import org.apache.hadoop.hive.ql.exec.UDF; @Description(name = "to_upper", value = "Converts input string to upper case") public class ToUpperUDF extends UDF { public String evaluate(String input) { if (input == null) { return null; } return input.toUpperCase(); } } ``` 此代码实现了 `evaluate` 方法,当传入一个字符串时会将其转成大写形式[^2]。 为了使上述 UDF 可用,在 Hive 中需加载 jar 文件以及注册该函数: ```sql ADD JAR /path/to/your/jarfile.jar; CREATE TEMPORARY FUNCTION to_upper AS 'com.example.ToUpperUDF'; SELECT to_upper('hello world') FROM your_table LIMIT 1; ``` 以上命令首先添加了包含 UDF 类的 jar 文件到当前 session;接着创建了一个临时函数名为 `to_upper` 并关联至 Java 类路径;最后查询表中任意一行数据测试新函数的效果。 #### 2. 聚合函数 (UDAF) 聚合函数通常用来计算一组记录上的汇总统计量,如平均数、总等。这里展示如何编写一个简单的求 UDAF 函数: ```java package com.example.udaf; import org.apache.hadoop.hive.ql.metadata.HiveException; import org.apache.hadoop.hive.ql.parse.SemanticException; import org.apache.hadoop.hive.ql.udf.generic.GenericUDAFEvaluator; import org.apache.hadoop.hive.serde2.objectinspector.ObjectInspector; import org.apache.hadoop.hive.serde2.objectinspector.primitive.PrimitiveObjectInspectorFactory; public final class SumEvaluator extends GenericUDAFEvaluator { private transient DoubleWritable result = new DoubleWritable(); static class Buffer implements AggregationBuffer { double sum = 0d; } @Override protected ObjectInspector initMode(ObjectInspector[] args) throws SemanticException { for (int i = 0; i < args.length; ++i) { PrimitiveObjectInspectorFactory.writableDoubleObjectInspector; } return PrimitiveObjectInspectorFactory.javaDoubleObjectInspector; } @Override public void iterate(AggregationBuffer agg, Object[] parameters) throws HiveException { ((SumEvaluator.Buffer)agg).sum += ((Number)parameters[0]).doubleValue(); } @Override public Object terminatePartial(AggregationBuffer agg) throws HiveException { return ((SumEvaluator.Buffer)agg).sum; } @Override public void merge(AggregationBuffer agg, Object partial) throws HiveException { ((SumEvaluator.Buffer)agg).sum += ((Number)partial).doubleValue(); } @Override public Object terminate(AggregationBuffer agg) throws HiveException { this.result.set(((SumEvaluator.Buffer)agg).sum); return this.result; } } ``` 这段代码展示了怎样构建一个基本的求运算逻辑,并且遵循了 UDAF 接口的要求。需要注意的是,这只是一个简化版本的例子,实际上还需要实现更多方法来支持分布式环境下的中间结果传递等功能[^3]。 同样地,在使用之前也需要先引入相应的 jar 包并将这个新的聚集函数映射给 Hive SQL 关键字: ```sql ADD JAR /path/to/your/jarfile.jar; CREATE AGGREGATE FUNCTION my_sum AS 'com.example.udaf.SumEvaluator'; SELECT my_sum(column_name) FROM table_name GROUP BY another_column; ``` #### 3. 表生成函数 (UDTF) 这类特殊类型的 UDF 将单一输入拆分成多条输出行。例如,我们可以设计一个分割逗号分隔列表的功能: ```java package com.example.udtf; import java.util.StringTokenizer; import org.apache.hadoop.hive.ql.exec.UDTF; import org.apache.hadoop.io.Text; public class ExplodeCSV extends UDF { Text outText = new Text(); public boolean process(Object[] forwardArgs) throws Exception { StringTokenizer st = new StringTokenizer((String)forwardArgs[0], ","); while(st.hasMoreTokens()){ outText.set(st.nextToken()); forward(forward(outText)); } return true; } } ``` 在此基础上,可以通过如下方式调用它来进行 CSV 字符串解析操作: ```sql ADD JAR /path/to/your/jarfile.jar; CREATE TEMPORARY FUNCTION explode_csv AS 'com.example.udtf.ExplodeCSV'; WITH sample_data AS ( SELECT 'apple,banana,cherry' as fruits UNION ALL SELECT 'dog,cat,bird' ) SELECT fruit FROM sample_data LATERAL VIEW explode_csv(fruits) exploded_fruit AS fruit; ``` 这样就可以把每一项都单独提取出来作为独立的一列显示出来了。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值