使用自增ID转62进制生成短码，通过HTTP接口实现长链接映射与重定向，结合sync.Mutex保证并发安全，内存存储用于原型，可扩展至Redis和URL校验。
用Golang实现一个URL短链接生成器，核心是把长网址转成短小可访问的链接，类似 bit.ly 这样的服务。虽然功能简单，但涉及哈希、存储、编码、并发安全等常见后端问题，适合练手项目。
1. 基本思路与设计
短链接系统本质是做映射：把长URL映射到一个短字符串（key），用户访问短链接时通过这个key查出原始URL并跳转。
关键点：
- 短码生成：可以用自增ID转62进制，或用哈希算法（如MD5取片段），也可用随机字符串
- 存储映射：内存用 map 或 sync.Map，生产可用 Redis 或数据库
- 冲突处理：生成的短码不能重复，需检查是否已存在
- 跳转逻辑：HTTP 302 重定向到原始URL
2. 使用自增ID + 62进制编码
最简单可靠的方式是使用全局自增ID，转换为62进制字符串作为短码（a-z, A-Z, 0-9）。
立即学习“go语言免费学习笔记（深入）”；
示例代码：
var (
idCounter int64 = 1000 // 模拟数据库自增ID
urlStore = make(map[string]string)
mu sync.Mutex
)
const chars = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"
func toBase62(n int64) string {
if n == 0 {
return string(chars[0])
}
result := ""
for n > 0 {
result = string(chars[n%62]) + result
n /= 62
}
return result
}
func generateShortKey() string {
mu.Lock()
defer mu.Unlock()
key := toBase62(idCounter)
idCounter++
return key
}
3. HTTP接口实现
提供两个接口：创建短链接 和 重定向访问。
func shortenHandler(w http.ResponseWriter, r *http.Request) {
if r.Method != "POST" {
http.Error(w, "只支持POST", http.StatusMethodNotAllowed)
return
}
longURL := r.FormValue("url")
if longURL == "" {
http.Error(w, "缺少url参数", http.StatusBadRequest)
return
}
// 检查是否已有相同长链
for k, v := range urlStore {
if v == longURL {
w.Write([]byte("短链: http://localhost:8080/" + k))
return
}
}
key := generateShortKey()
urlStore[key] = longURL
w.Write([]byte("短链: http://localhost:8080/" + key))
}
func redirectHandler(w http.ResponseWriter, r *http.Request) {
key := strings.TrimPrefix(r.URL.Path, "/")
if longURL, exists := urlStore[key]; exists {
http.Redirect(w, r, longURL, http.StatusFound)
} else {
http.Error(w, "链接不存在", http.StatusNotFound)
}
}
func main() {
http.HandleFunc("/shorten", shortenHandler)
http.HandleFunc("/", redirectHandler)
http.ListenAndServe(":8080", nil)
}
4. 可优化方向
当前版本是基础版，可用于学习。实际中可扩展：
- 用Redis替代内存map，支持持久化和分布式
- 加入校验：判断URL合法性
- 支持自定义短码
- 记录点击量、来源等统计信息
- 加缓存（如map[string]string做本地缓存）
- 使用更安全的随机生成方式防枚举
