博客

1、生成的是6位签名串，将此串可以储存到redis中，以签名后返回的串为key，原地址为value，待访问此地址时，直接从redis中获取，以此重定向到原地址中。

注：有存在重复的情况，需要做重复验证，验证redis中是否存在此签名。

package com.yt;

import java.security.MessageDigest;
import java.util.UUID;

/**
 * 生成短地址
 * @author tyg
 * @date   2019年1月24日下午2:27:24
 */
public class ShortUrlGenerator {

    /** 生成几个短连接地址，可自定义 */
    private final static int COUNT = 1;
    /** 生成几位短连接地址的签名串，可自定义 */
    public final static int LENGTH = 6;

    public static void main(String[] args) {
        String[] shortUrl = shortUrl("https://abc.com?name=Mj35eqpg4SI7UJ7CFp1K3z2UZVhpNYs0CQQzMv6sXD0SlS2EvJQ==&age=25");
        for (String s : shortUrl) {
            System.out.println(s);
        }
    }

    /**
     * 获取短连接地址
     * @param url 原字符串
     * @return String
     * @author tyg
     * @date   2019年1月24日下午2:27:24
     */
    public static String getShortUrl(String url) {
        return shortUrl(url)[0];
    }

    /**
     * 返回4组端地址
     * @param url 原串
     * @return String[]
     * @author tyg
     * @date   2019年1月24日下午2:28:36
     */
    private static String[] shortUrl(String url) {
        // 可以自定义生成 MD5 加密字符传前的混合 KEY
        // 要使用生成 URL 的字符
        String[] chars = new String[] { "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"

        };
        // 对传入网址进行 MD5 加密
        String hex = md5ByHex(UUID.randomUUID().toString() + url);

        String[] resUrl = new String[COUNT];
        for (int i = 0; i < COUNT; i++) {

            // 把加密字符按照 8 位一组 16 进制与 0x3FFFFFFF 进行位与运算
            String sTempSubString = hex.substring(i * 8, i * 8 + 8);

            // 这里需要使用 long 型来转换，因为 Inteper .parseInt() 只能处理 31 位 , 首位为符号位 , 如果不用long ，则会越界
            long lHexLong = 0x3FFFFFFF & Long.parseLong(sTempSubString, 16);
            StringBuilder outChars = new StringBuilder();
            for (int j = 0; j < LENGTH; j++) {
                // 把得到的值与 0x0000003D 进行位与运算，取得字符数组 chars 索引
                long index = 0x0000003D & lHexLong;
                // 把取得的字符相加
                outChars.append(chars[(int) index]);
                // 每次循环按位右移 5 位
                lHexLong = lHexLong >> 5;
            }
            // 把字符串存入对应索引的输出数组
            resUrl[i] = outChars.toString();
        }
        return resUrl;
    }

    /**
     * MD5加密(32位大写)
     * @param src
     * @return String
     * @author tyg
     * @date   2019年5月7日下午6:31:59
     */
    private static String md5ByHex(String src) {
        try {
            MessageDigest md = MessageDigest.getInstance("MD5");
            byte[] b = src.getBytes();
            md.reset();
            md.update(b);
            byte[] hash = md.digest();
            StringBuilder hs = new StringBuilder();
            String stmp = "";
            for (byte value : hash) {
                stmp = Integer.toHexString(value & 0xFF);
                if (stmp.length() == 1) {
                    hs.append("0").append(stmp);
                } else {
                    hs.append(stmp);
                }
            }
            return hs.toString().toUpperCase();
        } catch (Exception e) {
            return "";
        }
    }

}

2、调用示例。

package com.yt.controller;

import org.springframework.data.redis.core.BoundValueOperations;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RestController;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

/**
 * 测试类
 * @author tyg
 * @date   2020年8月5日下午2:33:40
 */
@RestController
public class TestController {

	@Resource
	protected RedisTemplate<String, String> redisTemplate;

	/**
	 * 重定向到原地址，请确保路径/st开头只在短地址使用
	 * @param response 
	 * @param sign 签名串
	 * @author tyg
	 * @date 2021-01-04 17:48
	 */
	@GetMapping("/st/{sign}")
	public void getUrl(HttpServletResponse response, @PathVariable String sign) throws IOException {
		BoundValueOperations<String, String> ops = redisTemplate.boundValueOps(sign);
		String toUrl = ops.get();
		if (toUrl != null) {
			// 重定向
			response.sendRedirect(toUrl);
		} else {
			// 处理不存在的情况

		}
	}
}