使用Golang构建短链接服务需实现长URL到短码的映射与重定向。1. 采用自增ID加Base62编码生成唯一短码，避免冲突且易于扩展；2. 数据存储推荐Redis，兼顾性能与持久化，支持快速读取和过期设置；3. 使用net/http或Gin框架处理POST /shorten生成短链和GET /{code}实现302跳转；4. 生产环境需校验URL格式、限制请求频率、支持自定义短码、统计点击量并启用HTTPS。核心是确保映射唯一性和跳转低延迟。
实现一个URL短链接服务的核心在于将长URL映射为简短、唯一的字符串，并提供跳转功能。Golang因其高性能和简洁语法，非常适合构建这类高并发Web服务。下面从设计思路到代码实现，一步步展示如何用Golang打造一个基础但实用的短链接服务。
1. 设计短链生成策略
短链的本质是将长URL映射为短字符串（如red">abc123），用户访问yourdomain.com/abc123时重定向到原始地址。
常见生成方式包括：
- 哈希法：使用MD5或SHA1对长URL哈希，取前几位并做Base62编码。优点是确定性映射，缺点是冲突可能，需校验是否已存在。
- 自增ID + 编码：使用数据库自增主键，转换为Base62字符串（a-z, A-Z, 0-9）。例如ID=12345 → "3d7"。唯一且无冲突，适合分布式场景可结合Snowflake生成唯一ID。
- 随机生成：生成固定长度随机字符串，检查是否冲突，若冲突则重试。控制长度（如6位）可支持海量组合。
推荐使用自增ID + Base62编码，逻辑清晰且易于扩展。
立即学习“go语言免费学习笔记（深入）”；
2. 数据存储与映射
需要持久化存储短码与原始URL的对应关系。可选方案：
- 内存存储（map）：开发调试阶段可用，性能极高，但重启丢失数据。
- Redis：适合缓存+持久化组合，SET key value EX 3600 可设置过期时间，GET快速读取，非常契合短链场景。
- MySQL/PostgreSQL：结构化存储，便于统计、管理、防重。表结构简单：
CREATE TABLE links (
id BIGINT AUTO_INCREMENT PRIMARY KEY,
code VARCHAR(10) UNIQUE NOT NULL,
url TEXT NOT NULL,
created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
);
3. Web服务路由与处理
使用net/http或Gin等框架快速搭建HTTP服务。
核心接口有两个：
- POST /shorten：接收JSON格式的长URL，返回短码或完整短链。
- GET /{code}：根据code查询原始URL，执行302跳转。
示例代码片段（使用标准库）：
http.HandleFunc("/shorten", func(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
http.Error(w, "only POST", http.StatusMethodNotAllowed)
return
}
var req struct{ URL string }
json.NewDecoder(r.Body).Decode(&req)
// 生成短码（简化版：使用自增+base62）
code := encodeBase62(generateID()) // 假设generateID返回递增数字
// 存入Redis或DB
err := redisClient.Set(ctx, code, req.URL, 0).Err()
if err != nil {
http.Error(w, "server error", http.StatusInternalServerError)
return
}
json.NewEncoder(w).Encode(map[string]string{
"short_url": "http://localhost:8080/" + code,
})
})
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
code := strings.TrimPrefix(r.URL.Path, "/")
if code == "" {
http.Redirect(w, r, "https://example.com", http.StatusFound)
return
}
originalURL, err := redisClient.Get(ctx, code).Result()
if err != nil {
http.Error(w, "not found", http.StatusNotFound)
return
}
http.Redirect(w, r, originalURL, http.StatusMovedPermanently)
})
4. 安全与优化建议
生产环境需考虑以下几点：
- 输入校验：确保传入的URL格式正确，可使用url.Parse验证。
- 防止恶意刷量：增加IP限流，可用token bucket或Redis记录请求频次。
- 自定义短码：允许用户指定短码，需检查是否已被占用。
- 统计点击量：每次跳转时更新Redis中的计数器，便于数据分析。
- HTTPS支持：使用Let's Encrypt免费证书启用TLS。
基本上就这些。一个轻量级短链服务可以很快跑起来，后续再根据需求加入缓存、队列、监控等功能。关键是把映射逻辑做稳，跳转做到低延迟。
