Java后端短链接生成服务实现

无界猴

37天前更新

096970

在Java后端实现短链接生成功能是一个常见需求，尤其是在构建类似bit.ly或goo.gl这样的URL缩短服务时。这个功能通常涉及以下几个步骤：

生成唯一且较短的标识符：为了缩短URL，我们需要一个系统来生成短小的标识符（短链接的一部分）。这可以通过哈希函数、自增ID、UUID等方式实现，但通常需要对生成的ID进行Base62或其他进制编码，以确保其长度较短。
存储映射关系：我们需要一个数据库或内存存储来记录原始URL和短链接之间的映射关系。
提供API接口：前端或其他服务可以通过API接口请求生成短链接或解析短链接。

以下是一个简单的实现示例，使用Spring Boot框架和一个内存中的HashMap来存储映射关系：

1. 创建Spring Boot项目

首先，创建一个Spring Boot项目，添加必要的依赖项（如Spring Web）。

2. 定义短链接生成服务

import org.springframework.stereotype.Service;

import java.nio.charset.StandardCharsets;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.atomic.AtomicLong;

@Service
public class UrlShortenerService {

    private static final String CHARACTERS = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
    private static final int BASE = CHARACTERS.length();
    private static final AtomicLong counter = new AtomicLong(0);
    private final Map<String, String> shortToLongMap = new HashMap<>();
    private final Map<String, String> longToShortMap = new HashMap<>();

    // 生成唯一标识符
    private String generateShortId() {
        long id = counter.incrementAndGet();
        StringBuilder sb = new StringBuilder();
        while (id > 0) {
            sb.insert(0, CHARACTERS.charAt((int) (id % BASE)));
            id /= BASE;
        }
        return sb.toString();
    }

    // 生成短链接
    public String shortenUrl(String longUrl) {
        String shortId = generateShortId();
        while (longToShortMap.containsValue(shortId)) { // 避免冲突
            shortId = generateShortId();
        }
        shortToLongMap.put(shortId, longUrl);
        longToShortMap.put(longUrl, shortId);
        return shortId;
    }

    // 解析短链接
    public String resolveShortUrl(String shortId) {
        return shortToLongMap.getOrDefault(shortId, "Invalid short URL");
    }
}

3. 创建控制器

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

@RestController
@RequestMapping("/api")
public class UrlShortenerController {

    @Autowired
    private UrlShortenerService urlShortenerService;

    @PostMapping("/shorten")
    public String shortenUrl(@RequestBody String longUrl) {
        return urlShortenerService.shortenUrl(longUrl);
    }

    @GetMapping("/resolve/{shortId}")
    public String resolveUrl(@PathVariable String shortId) {
        return urlShortenerService.resolveShortUrl(shortId);
    }
}

4. 启动应用

创建一个Spring Boot主类并启动应用：

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;

@SpringBootApplication
public class UrlShortenerApplication {

    public static void main(String[] args) {
        SpringApplication.run(UrlShortenerApplication.class, args);
    }
}