Golang实现短链接生成器项目解析

使用自增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）。

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做本地缓存）
• 使用更安全的随机生成方式防枚举