package com.billstudy.udf;
import java.util.HashMap;
import java.util.Map;
import org.apache.hadoop.hive.ql.exec.UDF;
import org.apache.hadoop.io.Text;
/**
 * 自定义UDF函数
 * @author Bill
 * @since V1.0 2015年6月14日 - 下午10:33:00
 */
public class NationUDF extends UDF{
	public static Map<String,String> nationMap = new HashMap<String,String>();
	static {
		
		nationMap.put("china", "中国");
		nationMap.put("japan", "日本");
		nationMap.put("usa", "美国");
		
	}
	
	private Text returnNation = new Text();
	
	/**
	 * 名称必须是evaluate 
	 * @author Bill
	 * @since V1.0 2015年6月14日 - 下午10:32:41
	 * @param nationKey
	 * @return
	 */
	public Text evaluate(Text nationKey){
		
		String nation = nationMap.get(nationKey.toString());
		
		if (nation == null) {
			nation = "未知";
		}
		
		returnNation.set(nation);
		
		return returnNation;
	}
	
}