感謝您的閱讀艘包，本文由 楊斌的博客 版權(quán)所有。
如若轉(zhuǎn)載，請(qǐng)注明出處：楊斌的博客（https://y0ngb1n.github.io/a/build-a-custom-url-shortener-service.html?utm_source=jianshu）

項(xiàng)目已托管于 GitHub：y0ngb1n/spring-boot-samples，歡迎 Star, Fork :kissing_heart:

準(zhǔn)備工作

• Spring Boot 2.1.0+
• Redis
• Lombok
• Guava 28.0
• Common Validator 1.6

添加依賴項(xiàng)

pom.xml

<dependencies>
    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-web</artifactId>
    </dependency>

    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-data-redis</artifactId>
    </dependency>

    <dependency>
      <groupId>commons-validator</groupId>
      <artifactId>commons-validator</artifactId>
    </dependency>

    <dependency>
      <groupId>com.google.guava</groupId>
      <artifactId>guava</artifactId>
    </dependency>
</dependencies>

application.yml

spring:
  # Redis Config
  redis:
    url: 127.0.0.1
    port: 6379
    password: your_password

logging:
  level:
    io.github.y0ngb1n.*: debug

核心代碼

/**
 * URL Shortener Resource
 *
 * @author yangbin
 */
@Slf4j
@RestController
@RequestMapping(path = "/v1")
public class UrlShortenerController {

  @Autowired
  StringRedisTemplate redisTemplate;

  @GetMapping(path = "/{id}")
  public String getUrl(@PathVariable String id) {
+   String url = redisTemplate.opsForValue().get(id);
    log.debug("URL Retrieved: {}", url);
    return url;
  }

  @PostMapping
  public String create(@RequestBody String url) {
    UrlValidator urlValidator = new UrlValidator(
      new String[]{"http", "https"}
    );
    if (urlValidator.isValid(url)) {
-     String id = Hashing.murmur3_32().hashString(url, StandardCharsets.UTF_8).toString();
      log.debug("URL Id generated: {}", id);
+     redisTemplate.opsForValue().set(id, url);
      return id;
    }
    throw new RuntimeException("URL Invalid: " + url);
  }
}

使用方式

Step 0: 安裝并啟動(dòng) Redis

# on Windows
scoop install redis
redis-server

# on Mac
brew install redis
redis-server

Step 1: 啟動(dòng) url-shortener 服務(wù)

$ mvn install
...
[INFO] BUILD SUCCESS
...
$ mvn spring-boot:run
...
2019-08-21 21:03:50.215  INFO 10244 --- [ main] o.s.b.w.embedded.tomcat.TomcatWebServer  : Tomcat started on port(s): 8080 (http) with context path ''
2019-08-21 21:03:50.219  INFO 10244 --- [ main] i.g.y.s.u.UrlShortenerApplication        : Started UrlShortenerApplication in 6.01 seconds (JVM running for 12.165)

Step 2: 生成短鏈

$ curl -X POST http://127.0.0.1:8080/v1 \
  -H 'Content-Type: text/plain' \
  -d https://y0ngb1n.github.io
515bbe2b

Step 3: 還原短鏈

$ curl -X GET http://127.0.0.1:8080/v1/515bbe2b
https://y0ngb1n.github.io

查看日志

...
2019-08-21 21:42:26.788 DEBUG 10244 --- [nio-8080-exec-2] i.g.y.s.u.c.UrlShortenerController       : URL Id generated: 515bbe2b
2019-08-21 21:42:40.748 DEBUG 10244 --- [nio-8080-exec-3] i.g.y.s.u.c.UrlShortenerController       : URL Retrieved: https://y0ngb1n.github.io

參考資料

• https://youtu.be/Zr0E2VP24w8
• https://en.wikipedia.org/wiki/MurmurHash
• https://github.com/google/guava/wiki/HashingExplained
• https://www.flyml.net/2016/09/05/cassandra-tutorial-murmurhash/
• 短網(wǎng)址（Short URL）系統(tǒng)的原理及其實(shí)現(xiàn)倘要，by 胡方運(yùn)
• 如何快速判斷某 URL 是否在一個(gè) 20 億的網(wǎng)址 URL 集合中？十拣，by 張振偉
  • 應(yīng)用場(chǎng)景：黑名單封拧、URL 去重、單詞拼寫檢查夭问、Key-Value 緩存系統(tǒng)的 Key 校驗(yàn)哮缺、ID 校驗(yàn)，比如訂單系統(tǒng)查詢某個(gè)訂單 ID 是否存在甲喝，如果不存在就直接返回
• 談?wù)勅治ㄒ?ID 生成方法
• Leaf — 美團(tuán)點(diǎn)評(píng)分布式 ID 生成系統(tǒng)