如何在Golang中实现URL短链接服务-Golang-PHP中文网

如何在Golang中实现URL短链接服务

P粉602998670

发布： 2025-10-15 08:00:07

原创

630人浏览过

答案：实现URL短链接服务需生成短码、存储映射并提供重定向接口。使用Golang可借助自增ID转62进制生成唯一短码，通过SQLite或Redis持久化short_code与原URL的映射关系，HTTP接口处理中POST /shorten创建短链，GET /{code}查询并302跳转，辅以缓存、过期清理和点击统计优化性能与功能。

如何在golang中实现url短链接服务

实现一个URL短链接服务的核心是将长网址转换为简短、可访问的别名，并能通过该别名重定向到原始地址。Golang因其高性能和简洁的HTTP处理能力，非常适合构建这类服务。下面介绍关键步骤和代码结构。

1. 设计短码生成策略

短链接的核心是将长URL映射为短字符串（如 abc123），常见方式有：

自增ID转62进制：使用数据库自增主键，转换为 a-z, A-Z, 0-9 组成的字符串，保证唯一且无冲突。
哈希算法 + 冲突处理：对原URL做MD5或SHA1，取部分字符作为短码，但需检查是否已存在。
随机生成 + 唯一性校验：生成固定长度随机字符串，查库确认未被使用。

推荐使用自增ID转62进制，简单高效，避免碰撞。

示例：62进制转换函数

const charset = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789"

func toBase62(n int) string {
	if n == 0 {
		return string(charset[0])
	}
	var result []byte
	for n > 0 {
		result = append([]byte{charset[n%62]}, result...)
		n /= 62
	}
	return string(result)
}

登录后复制

2. 存储映射关系

需要持久化短码与原始URL的对应关系。可用数据库如SQLite、PostgreSQL，或内存存储如Redis（适合高并发）。

立即学习“go语言免费学习笔记（深入）”；

链企AI

专业的AI商业搜索和标讯服务平台，AI采集招投标信息，让您免费查看全网商业资讯，为您的商机之旅助力！

查看详情

表结构示例（SQLite）：

CREATE TABLE urls (
    id INTEGER PRIMARY KEY AUTOINCREMENT,
    short_code TEXT UNIQUE NOT NULL,
    original_url TEXT NOT NULL
);

登录后复制

插入时先查是否存在，不存在则生成新码并保存。

3. 实现HTTP接口

提供两个接口：创建短链和重定向访问。

POST /shorten：接收JSON中的long_url，返回短链地址。
GET /{code}：根据短码查询原URL，执行302跳转。

示例路由处理

http.HandleFunc("/shorten", func(w http.ResponseWriter, r *http.Request) {
    if r.Method != "POST" {
        http.Error(w, "只支持POST", 405)
        return
    }

    var data struct{ LongURL string }
    json.NewDecoder(r.Body).Decode(&data)

    // 查询是否已有相同URL
    // 若无，则生成新短码并插入数据库
    shortCode := toBase62(newID)

    // 插入数据库...
    
    w.Header().Set("Content-Type", "application/json")
    json.NewEncoder(w).Encode(map[string]string{
        "short_url": "http://localhost:8080/" + shortCode,
    })
})

http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
    path := strings.TrimPrefix(r.URL.Path, "/")
    if path == "" {
        http.Redirect(w, r, "/index.html", 302)
        return
    }

    // 根据 path (即 shortCode) 查询数据库
    originalURL := queryDB(path)
    if originalURL == "" {
        http.NotFound(w, r)
        return
    }

    http.Redirect(w, r, originalURL, 302)
})

登录后复制