编写UDF函数可以穿数组么

Question

我的在hive中传的类型是hive中的数组   

Answer 1

当然可以了~我看是你的udf写的有问题吧.你传的array <int> 怎么看报错提示的是要传入struct类型啊?

Answer 2

在给你个demo吧.用于得到数据里最大的一个值package cn.com.diditaxi.hive.cf;import org.apache.hadoop.hive.ql.exec.Description;
import org.apache.hadoop.hive.ql.exec.UDFArgumentException;
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.ql.udf.generic.GenericUDFUtils;
import org.apache.hadoop.hive.serde2.objectinspector.ObjectInspector;
import org.apache.hadoop.hive.serde2.objectinspector.ObjectInspectorUtils;
import org.apache.hive.pdk.HivePdkUnitTest;
import org.apache.hive.pdk.HivePdkUnitTests;/**
 * GenericUDF Class for SQL construct "greatest(value1, value2, value3, ....)". 
 * Oracle's CREATEST returns the greatest of the list of one or more expressions.
 * 
 */
@Description(name = "greatest", value = "_FUNC_(value1, value2, value3, ....) " +
"- Returns the greatest value in the list.", 
extended = "Example:\n" + " > SELECT _FUNC_(2, 5, 12, 3) FROM src;\n 12")
@HivePdkUnitTests(setup = "create table dual_data (i int); "
+ "insert overwrite table dual_data select 1 from dual limit 1;", 
cleanup = "drop table if exists dual_data;", 
cases = {
@HivePdkUnitTest(query = "SELECT nexr_greatest(2, 5, 12, 3) " +
"FROM dual_data;", result = "12"),
@HivePdkUnitTest(query = "SELECT nexr_greatest('2', '5', '12', '3') " +
"FROM dual_data;", result = "5"),
@HivePdkUnitTest(query = "SELECT nexr_greatest('apples', 'oranges', 'bananas') " +
"FROM dual_data;", result = "oranges") })
public class GenericUDFGreatest extends GenericUDF { private ObjectInspector[] argumentOIs;
private GenericUDFUtils.ReturnObjectInspectorResolver returnOIResolver; @Override
public ObjectInspector initialize(ObjectInspector[] arguments) throws UDFArgumentException { for (int i = 0; i < arguments.length; i++) {
if (arguments[i].getCategory() != ObjectInspector.Category.PRIMITIVE) {
throw new UDFArgumentTypeException(i, "Only primitive type arguments are accepted but "
+ arguments[i].getTypeName() + " is passed.");
}
} argumentOIs = arguments;
returnOIResolver = new GenericUDFUtils.ReturnObjectInspectorResolver(true);
for (int i = 0; i < arguments.length; i++) {
if (!returnOIResolver.update(arguments[i])) {
throw new UDFArgumentTypeException(i, "The value of return should have the same type: \""
+ returnOIResolver.get().getTypeName() + 
"\" is expected but \"" + arguments[i].getTypeName()
+ "\" is found");
}
} return returnOIResolver.get();
} @Override
public Object evaluate(DeferredObject[] fields) throws HiveException {
Object greatestObject = null;
ObjectInspector greatestOI = null; for (int i = 0; i < fields.length; i++) {
Object fieldObject = fields[i].get();
if (greatestObject == null) {
greatestObject = fieldObject;
greatestOI = argumentOIs[i];
continue;
} if (ObjectInspectorUtils.compare(greatestObject, greatestOI, fieldObject, argumentOIs[i]) <= 0) {
greatestObject = fieldObject;
greatestOI = argumentOIs[i];
}
} return returnOIResolver.convertIfNecessary(greatestObject, greatestOI);
} @Override
public String getDisplayString(String[] children) {
StringBuilder sb = new StringBuilder();
sb.append("greatest (");
for (int i = 0; i < children.length - 1; i++) {
sb.append(children[i]).append(", ");
}
sb.append(children[children.length - 1]).append(")");
return sb.toString();
}}