hive 3.0.0自定義函式

hgs19921112發表於2018-09-06
參考: 
      
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/,如需轉載,請註明出處,否則將追究法律責任。

相關文章