长URL转短连接的简单设计与实现

admin · 发表于 2017-3-21 09:19:28

非常多的时候，我们不想在分享的时候采用长长的链接，主要的原因有：

今天的理由不是重点，重点是如何用一点点代码实现这个短链接的生成。

我们的注意点有：

首先设计接口

复制代码

只是个示例，注释比较简陋，接下来是抽象类实现，把公共的内容放这里

import java.util.Random;
/**
* Created by luoguo on 2017/3/16.
*/
public abstract class AbstractUrlShortener implements UrlShortener {
public static char[] VALID_CHARS = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890".toCharArray();
private static Random random = new Random(System.currentTimeMillis());
protected int length = 4;
public AbstractUrlShortener() {
}
public AbstractUrlShortener(int length) {
this.length = length;
}
public void setLength(int length) {
this.length = length;
}
public int getLength() {
return length;
}
public String get(String url) {
String sortUrl = seek(url);
if (sortUrl == null) {
sortUrl = generate(url);
put(url, sortUrl);
}
return sortUrl;
}
public String generate(String url, int seed) {
char[] sortUrl = new char[length];
for (int i = 0; i < length; i++) {
sortUrl[i] = VALID_CHARS[seed % VALID_CHARS.length];
seed = random.nextInt(Integer.MAX_VALUE) % VALID_CHARS.length;
}
return new String(sortUrl);
}
public String generate(String url) {
String shortUrl;
shortUrl = generate(url, random.nextInt(Integer.MAX_VALUE));
while (seek(shortUrl) != null) {
shortUrl = generate(url, random.nextInt(Integer.MAX_VALUE));
}
put(url, shortUrl);
return shortUrl;
}
}

复制代码

实际的需要在数据库层实现，这里在内存里面存储的实现一个意思一下

import java.util.Date;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
/**
* 内存存储的实现
* <p>
* Created by luoguo on 2017/3/16.
*/
public class UrlShortenerMemory extends AbstractUrlShortener {
private Map<String, String> url2ShortUrl = new ConcurrentHashMap<String, String>();
private Map<String, String> shortUrl2Url = new ConcurrentHashMap<String, String>();
public UrlShortenerMemory() {
super();
}
public UrlShortenerMemory(int length) {
super(length);
}
public void put(String url, String shortUrl) {
url2ShortUrl.put(url, shortUrl);
shortUrl2Url.put(shortUrl, url);
}
public String seek(String shortUrl) {
return shortUrl2Url.get(shortUrl);
}
public void clean(String url) {
String sortUrl = url2ShortUrl.get(url);
if (sortUrl != null) {
url2ShortUrl.remove(url);
shortUrl2Url.remove(sortUrl);
}
}
public void clean(Date date) {
throw new UnsupportedOperationException();
}
}

复制代码

弄点测试代码，试试看效果怎么样

复制代码

这里是运行结果

复制代码

感觉还可以，当然里面还有一些不完善的地方

这个只是化几分钟展示一个思路，并没有经过系统的思考的验证，正式系统请慎重使用。

更多精彩技术博文，请看本人博客空间

00

		自动登录	找回密码
密码			立即注册