hive 3.0.0自定义函数

参考： 
      https://rramos.github.io/2017/09/27/hive-udfs/

package hello_hive;
import org.apache.hadoop.hive.ql.exec.UDFArgumentException;
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.ObjectInspectorFactory;
import org.apache.hadoop.hive.serde2.objectinspector.PrimitiveObjectInspector;
import org.apache.hadoop.hive.serde2.objectinspector.primitive.PrimitiveObjectInspectorFactory;
import org.apache.hadoop.io.Text;
import org.apache.hive.hplsql.Udf;
public class HiveTest extends Udf{
	StringBuilder builder = new StringBuilder();
	public String evaluate(String invalue) throws HiveException {
		int length = invalue.length();
		char[] cha = invalue.toCharArray();
		for(int i = 0 ;i<length;i++) {
			String tmp = String.valueOf(cha[i]);
			if(Math.floorMod(i, 2)==0) {
				builder.append(tmp.toUpperCase());
			}else {
				builder.append(tmp.toLowerCase());
			}
			
		}
		return builder.toString();	
	}
	
}
class HiveTest2 extends GenericUDF{
	StringBuilder builder = new StringBuilder();
	Text tx = new Text();
	@Override
	//主要逻辑函数
	public Object evaluate(DeferredObject[] defer) throws HiveException {
		builder.delete(0, builder.length());
		if(defer.length<=0) {
			return null;
		}
		else {
			int length = defer[0].get().toString().length();
			char[] cha = defer[0].get().toString().toCharArray();
			for(int i = 0 ;i<length;i++) {
				String tmp = String.valueOf(cha[i]);
				if(Math.floorMod(i, 2)==0) {
					builder.append(tmp.toUpperCase());
				}else {
					builder.append(tmp.toLowerCase());
				}
				
			}
			
		}
		tx.set(builder.toString());
		return tx;	
	}
	//获取函数帮助信息的函数
	@Override
	public String getDisplayString(String[] args) {
		return getStandardDisplayString("Usage: upd(strs)", args);
	}
	//判断函数返回值类型的函数
	@Override
	public ObjectInspector initialize(ObjectInspector[] arg0) throws UDFArgumentException {
		ObjectInspector outputOI =PrimitiveObjectInspectorFactory.writableStringObjectInspector;
		return  outputOI;
	}
	
}

来自 “ ITPUB博客 ” ，链接：http://blog.itpub.net/31506529/viewspace-2213683/，如需转载，请注明出处，否则将追究法律责任。

转载于:http://blog.itpub.net/31506529/viewspace-2213683/