HIVE-UDF之MD5密钥算法实现

/**
 * Project: UDFTest
 *
 * File Created at 2012-11-28
 * $Id$
 *
 * Copyright 1999-2100 Alibaba.com Corporation Limited.
 * All rights reserved.
 *
 * This software is the confidential and proprietary information of
 * Alibaba Company. ("Confidential Information").  You shall not
 * disclose such Confidential Information and shall use it only in
 * accordance with the terms of the license agreement you entered into
 * with Alibaba.com.
 */

package com.alibaba.hive.udf.lpxuan;

import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;

import org.apache.hadoop.hive.ql.exec.UDF;
import org.apache.hadoop.io.DoubleWritable;
import org.apache.hadoop.io.IntWritable;
import org.apache.hadoop.io.LongWritable;

/**
 * TODO Comment of UDFMD5Hash
 *
 * @author pengxuan.lipx
 */

public class UDFMD5Hash extends UDF {

    private static final String[] hexDigits = { "0", "1", "2", "3", "4", "5", "6", "7", "8", "9",
            "a", "b", "c", "d", "e", "f"   };
    
    /**
     *
     * @param in  数据需要加密的值
     * @param seed  密钥
     * @return
     */
    public static String evaluate(LongWritable in, String seed){
        if (in == null ||seed == null) {
            return null;
        }
        String toEncrypt = in.toString() + seed;
        return getMd5(toEncrypt);
    }
    
    public static String evaluate(DoubleWritable in, String seed){
        if (in == null ||seed == null) {
            return null;
        }
        String toEncrypt = in.toString() + seed;
        return getMd5(toEncrypt);
    }
    
    public static String evaluate(IntWritable in, String seed){
        if (in == null ||seed == null) {
            return null;
        }
        String toEncrypt = in.toString() + seed;
        return getMd5(toEncrypt);
    }
    
    public static String evaluate(String in, String seed){
        if (in == null ||seed == null) {
            return null;
        }
        String toEncrypt = in.toString() + seed;
        return getMd5(toEncrypt);
    }

    private static String byteArrayToHexString(byte[] b) {
        StringBuffer resultSb = new StringBuffer();
        for (int i = 0; i < b.length; i++) {
            resultSb.append(byteToHexString(b[i]));
        }
        return resultSb.toString();
    }

    private static String byteToHexString(byte b) {
        int n = b;
        if (n < 0)
            n = 256 + n;
        int d1 = n / 16;
        int d2 = n % 16;
        return hexDigits[d1] + hexDigits[d2];
    }

    private static String getMd5(String toEncrypt) {
        try {
            byte[] results = MessageDigest.getInstance("MD5").digest(toEncrypt.getBytes());
            String resultString = byteArrayToHexString(results);
            return resultString;
        } catch (NoSuchAlgorithmException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
            return null;
        }
    }

    /**
     * @param args
     * @throws NoSuchAlgorithmException
     */
    public static void main(String[] args) throws NoSuchAlgorithmException {

        // TODO Auto-generated method stub
        UDFMD5Hash md5test = new UDFMD5Hash();
        LongWritable in1 = new LongWritable(12345678);
        String seed = "helloworld";
        DoubleWritable in2 = new DoubleWritable(12345678.0);
        String in3 = "12345678";
        System.out.println(md5test.evaluate(in1, seed));
        System.out.println(md5test.evaluate(in2, seed));
        System.out.println(md5test.evaluate(in3, seed));
    }

}