java实现短链接转换服务

类似上图这种短信，对应的就是一个短链接。

看到原理也不难，于是先写个最简易版的玩玩，以后有需求了再补充吧

下面是一个简易版的短链接生成代码

搭个spring-boot服务，复制这两个controller，就能看到效果。

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.HashMap;
import java.util.Random;

/**
 * @Description: 发号器随机生成
 */
@RestController
@RequestMapping("/shortUrl")
public class ShortUrlController {

    private String BASE62 = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
    // 短链url域名前缀
    private String shortUrlPrefix = "http://a.cn/";
    private HashMap<String, String> map = new HashMap<>();

    @RequestMapping("getShortUrl")
    public String getShortUrl(String longUrl) {
        String key = creatKey();
        while (map.containsKey(key)) {
            key = creatKey();
        }
        map.put(key, longUrl);
        return shortUrlPrefix + key;
    }

    @RequestMapping("getLongUrl")
    public String getLongUrl(String shortUrl) {
        return map.get(shortUrl.replace(shortUrlPrefix, ""));
    }

    private String creatKey() {
        Random rand = new Random();
        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < 6; i++) {
            sb.append(BASE62.charAt(rand.nextInt(62)));
        }
        return sb.toString();
    }
}

import com.zs.qunfx.util.Base62Util;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.HashMap;

/**
 * @Description: 发号器自增生成
 */
@RestController
@RequestMapping("/shortUrl2")
public class ShortUrl2Controller {

    // 短链url域名前缀
    private String shortUrlPrefix = "http://a.cn/";
    private HashMap<String, String> map = new HashMap<>();
    private Long num = 1000000L;

    @RequestMapping("getShortUrl")
    public String getShortUrl(String longUrl) {
        String key = creatKey();
        map.put(key, longUrl);
        return shortUrlPrefix + key;
    }

    @RequestMapping("getLongUrl")
    public String getLongUrl(String shortUrl) {
        return map.get(shortUrl.replace(shortUrlPrefix, ""));
    }

    private String creatKey() {
        String base62 = Base62Util.base62Encode(num);
        num++;
        return base62;
    }
}

最后

现在常用的还是第二种，用自增的发号器生成对应的短链接。

生产环境要用，可以用数据库的自增id来发号，或者分布式下生成id用类似雪花算法来发号。

同时，避免原始链接重复，可做重复判断，可用布隆过滤器或redis

长链接和短链接的对应关系，可保存在数据库，也可保存在类似redis中，顺带还可以设置过期时间。

也有开源的生成短链接的，比如yourls，PHP实现，安装只需要装PHP和mysql即可。