Spring Boot 短链接实现

短链接是将长URL转换为短字符串的服务，便于分享和传播。以下是使用Spring Boot实现短链接服务的几种方法：

1、基本实现方案

依赖配置
xml
<dependencies
<dependency
<groupIdorg.springframework.boot</groupId
<artifactIdspringbootstarterweb</artifactId
</dependency
<dependency
<groupIdorg.springframework.boot</groupId
<artifactIdspringbootstarterdatajpa</artifactId
</dependency
<dependency
<groupIdcom.h2database</groupId
<artifactIdh2</artifactId
<scoperuntime</scope
</dependency
</dependencies

实体类
java
@Entity
public class ShortUrl {
@Id
@GeneratedValue(strategy = GenerationType.IDENTITY)
private Long id;

@Column(nullable = false)
private String originalUrl;

@Column(unique = true)
private String shortCode;

// getters and setters
}

服务层
java
@Service
public class ShortUrlService {

@Autowired
private ShortUrlRepository repository;

private static final String BASE62 = 0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz;

public String createShortUrl(String originalUrl) {
ShortUrl shortUrl = new ShortUrl();
shortUrl.setOriginalUrl(originalUrl);
ShortUrl saved = repository.save(shortUrl);

String shortCode = encodeBase62(saved.getId());
saved.setShortCode(shortCode);
repository.save(saved);

return shortCode;
}

public String getOriginalUrl(String shortCode) {
ShortUrl shortUrl = repository.findByShortCode(shortCode)
.orElseThrow(() new RuntimeException(URL not found));
return shortUrl.getOriginalUrl();
}

private String encodeBase62(long num) {
StringBuilder sb = new StringBuilder();
while (num 0) {
sb.append(BASE62.charAt((int)(num % 62)));
num /= 62;
}
return sb.reverse().toString();
}
}

控制器
java
@RestController
@RequestMapping(/api/shorturl)
public class ShortUrlController {

@Autowired
private ShortUrlService service;

@PostMapping
public String create(@RequestBody String originalUrl) {
return service.createShortUrl(originalUrl);
}

@GetMapping(/{shortCode})
public ResponseEntity<Void redirect(@PathVariable String shortCode) {
String originalUrl = service.getOriginalUrl(shortCode);
return ResponseEntity.status(HttpStatus.FOUND)
.location(URI.create(originalUrl))
.build();
}
}

2、使用现成库

可以使用现成的库如 `urlshortener` 简化开发：

xml
<dependency
<groupIdcom.github.seratch</groupId
<artifactIdurlshortener</artifactId
<version1.0.0</version
</dependency

3、高级功能扩展

1、自定义短码：允许用户指定短码
2、过期时间：为短链接设置有效期
3、访问统计：记录点击次数、来源等
4、API限流：防止滥用
5、缓存：使用Redis提高性能

4、生产环境建议

1、使用分布式ID生成器（如雪花算法）代替自增ID
2、考虑使用Redis等缓存提高重定向速度
3、实现防重放攻击机制
4、添加监控和告警

5、替代方案

如果不想自己实现，可以考虑使用第三方服务：
Bitly API
Firebase Dynamic Links
新浪短链接API

以上是Spring Boot实现短链接服务的基本框架，可以根据实际需求进行扩展和优化。