hive自定义函数UDF开发

package com.xxx.xxx;


import java.text.SimpleDateFormat;
import java.util.Calendar;
import java.util.Date;


import org.apache.hadoop.hive.ql.exec.UDF;


/**
 * 
 * @author 作 者：chenyunliang
 * @note 实现功能： 
 * @note 此类实现检查身份证和转换身份证功能
 */
public class ValidateIDCardUDF extends UDF {
  // 中国公民身份证号码最小长度
  public static final int  CHINA_ID_MIN_LENGTH    = 15;


  // 中国公民身份证号码最大长度
  public static final int  CHINA_ID_MAX_LENGTH    = 18;


  // 中国公民身份证号码最大长度
  public static final int  CHINA_ID_TWHKMK_LENGTH  = 10;


  // 每位加权因子
  public static final int  power[]          = { 7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2 };


  /**
   * 
   * @author 作 者：chenyunliang
   * @note 实现功能：
   * @note 此方法实现测试功能
   */
  public static void main(String[] args) {
    ValidateIDCardUDF validateIDCardUDF = new ValidateIDCardUDF();
    System.out.println(validateIDCardUDF.evaluate("43061119850113118X"));
    // 370725881105149, 37072519881105149x
  }


  /**
   * 
   * @author 作 者：chenyunliang
   * @note 实现功能：
   * @note 此方法实现将15位身份证号码转换为18位功能
   */
  public String evaluate(String cardID) {
    String cardID_18 = "";


    // 检查输入参数，必须是15位
    if (cardID.trim().length() == CHINA_ID_MAX_LENGTH) {
      return cardID.trim();
    } else if (cardID.trim().length() != CHINA_ID_MIN_LENGTH || !isNum(cardID)) {
      return null;
    }


    // 获取出生年月日
    String birthDayString = cardID.substring(6, 12);
    Date birthDate = null;
    try {
      birthDate = (Date) new SimpleDateFormat("yyMMdd").parse(birthDayString);
    } catch (Exception e) {
      e.printStackTrace();
    }


    // 通过日历来获取年份等
    Calendar calendar = Calendar.getInstance();
    if (null != birthDate) {
      calendar.setTime(birthDate);
    }


    String yearString = String.valueOf(calendar.get(Calendar.YEAR));
    cardID_18 = cardID.substring(0, 6) + yearString + cardID.substring(8);


    // 转换字符数组
    char[] cardID_18_chars = cardID_18.toCharArray();
    if (null != cardID_18_chars) {
      int[] iCard = convertCharToInt(cardID_18_chars);
      int iSum17 = getPowerSum(iCard);


      // 获取校验位
      String checksumString = getCheckCode18(iSum17);
      if (checksumString.length() > 0) {
        cardID_18 += checksumString;
      } else {
        return null;
      }
    }


    return cardID_18;


  }


  /**
   * 
   * @author 作 者：chenyunliang
   * @note 实现功能：
   * @note 此方法实现将power和值与11取模获得余数进行校验码判断功能
   */
  private String getCheckCode18(int iSum) {
    String sCode = "";
    switch (iSum % 11) {
    case 10:
      sCode = "2";
      break;
    case 9:
      sCode = "3";
      break;
    case 8:
      sCode = "4";
      break;
    case 7:
      sCode = "5";
      break;
    case 6:
      sCode = "6";
      break;
    case 5:
      sCode = "7";
      break;
    case 4:
      sCode = "8";
      break;
    case 3:
      sCode = "9";
      break;
    case 2:
      sCode = "x";
      break;
    case 1:
      sCode = "0";
      break;
    case 0:
      sCode = "1";
      break;
    }
    return sCode;
  }


  /**
   * 
   * @author 作 者：chenyunliang
   * @note 实现功能：
   * @note 此方法实现 将身份证的每位和对应位的加权因子相乘之后，再得到和值 功能
   */
  private int getPowerSum(int[] iArr) {
    int iSum = 0;
    if (power.length == iArr.length) {
      for (int i = 0; i < iArr.length; i++) {
        for (int j = 0; j < power.length; j++) {
          if (i == j) {
            iSum = iSum + iArr[i] * power[j];
          }
        }
      }
    }
    return iSum;
  }


  /**
   * 
   * @author 作 者：chenyunliang
   * @note 实现功能：
   * @note 此方法实现 将字符数组转换成数字数组 功能
   */
  private int[] convertCharToInt(char[] ca) {
    int len = ca.length;
    int[] iArr = new int[len];
    try {
      for (int i = 0; i < len; i++) {
        iArr[i] = Integer.parseInt(String.valueOf(ca[i]));
      }
    } catch (NumberFormatException e) {
      e.printStackTrace();
    }
    return iArr;
  }


  /**
   * 
   * 
   * @author 作 者：chenyunliang
   * @note 实现功能：
   * @note 此方法实现检查是否是数字功能
   */
  private boolean isNum(String val) {
    return val == null || "".equals(val) ? false : val.matches("^[0-9]*{1}");
  }
}

hdfs dfs -mkdir /user/hive/udflib

hdfs dfs -put hive-udf.jar  /user/hive/udflib

add jar hdfs:///user/hive/udflib/hive-udf.jar;
create function dcf.validateidcardudf as 'com.audaque.gldm.ValidateIDCardUDF' using jar 'hdfs:///user/hive/udflib/hive-udf.jar';

select dcf.validateIDCardUDF('370725881105149');

drop function if exists dcf.validateIDCardUDF;

list jars;

delete jar /opt/huawei/Bigdata/tmp/hivelocaltmp/session_resources/c03ed7ff-1b7b-43e3-b9cc-e76f45b7d397_resources/hive-udf.jar;