【转】一个短URL生成算法

2015/11/24 17:07
阅读数 537
import java.security.MessageDigest;

public class UrlUtils {

	private final static String[] HEX_DIGITS = { "0", "1", "2", "3", "4", "5","6", "7", "8", "9", "a", "b", "c", "d", "e", "f" };
	private final static String[] SURL_CHARS = new String[] { // 要使用生成URL的字符
		"a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l", "m", "n",
		"o", "p", "q", "r", "s", "t", "u", "v", "w", "x", "y", "z",
		"0", "1", "2", "3", "4", "5", "6", "7", "8", "9", "A", "B",
		"C", "D", "E", "F", "G", "H", "I", "J", "K", "L", "M", "N",
		"O", "P", "Q", "R", "S", "T", "U", "V", "W", "X", "Y", "Z" };
	
	static String byteArrayToHexString(byte[] b) {
		StringBuilder resultSb = new StringBuilder();
		for (int i = 0; i < b.length; i++) {
			resultSb.append(byteToHexString(b[i]));
		}
		
		return resultSb.toString();
	}

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

	static String md5Encode(String origin) {
		String resultStr = null;
		try {
			resultStr = new String(origin).trim();
			MessageDigest md = MessageDigest.getInstance("MD5");
			resultStr = byteArrayToHexString(md.digest(resultStr.getBytes("UTF-8")));
		} catch (Exception ex) {
			ex.printStackTrace();
		}
		
		return resultStr;
	}

	/**
	 * @param key 自定义生成MD5加密字符串前的混合KEY
	 * @param url 需要转换短URL的地址
	 * @return 为url生成四个短URL字符
	 */
	public static String[] shortUrl(String key, String url) {
		if(key == null || url == null)
		{
			throw new IllegalArgumentException("shortUrl(String key, String url) need param `key` and `url`!");
		}
		
		String hex = md5Encode(key + url);
		int hexLen = hex.length();
		int subHexLen = hexLen / 8;
		
		String[] shortUrls = new String[4];
		StringBuilder outChars = new StringBuilder();
		
		for (int i = 0; i < subHexLen; i++) {
			outChars.delete(0, outChars.length());
			int j = i + 1;
			String subHex = hex.substring(i * 8, j * 8);
			long idx = Long.valueOf("3FFFFFFF", 16) & Long.valueOf(subHex, 16);

			for (int k = 0; k < 6; k++) {
				int index = (int) (Long.valueOf("0000003D", 16) & idx);
				outChars.append(SURL_CHARS[index]);
				idx = idx >> 5;
			}
			
			shortUrls[i] = outChars.toString();
		}

		return shortUrls;
	}

	public static void main(String[] args) {
	    String url = "http://www.oschina.net/question/59519_82623";   
            for (String string : shortUrl("test_key", url)) {   
               System.out.println(string); 
            }
            
            // maEJju
            // AjmIja
            // NNjQJz
            // uaUnYv
	}

}





展开阅读全文
打赏
0
13 收藏
分享
加载中
更多评论
打赏
0 评论
13 收藏
0
分享
返回顶部
顶部