Golang开发一个简单的URL短链服务_网络技术_宿州市腾雀网络科技有限公司

Golang开发一个简单的URL短链服务

发布时间 - 2026-01-12 00:00:00 点击率：次

不用自增ID做短码因会暴露业务数据、难以并发预知ID，需解耦；推荐随机+重试或雪花ID派生，Go+SQLite用INSERT ON CONFLICT避免竞态。

为什么不用自增ID做短码

直接用数据库自增 id 转成62进制看似简单，但暴露了业务总量、增长节奏，还容易被遍历撞库。真实场景下，id 是内部标识，短码必须和它解耦。

更关键的是：并发插入时，你没法在写入前预知 id，除非先 INSERT ... RETURNING id（PostgreSQL）或用 LAST_INSERT_ID()（MySQL），但这会增加一次 round-trip，且难以原子化生成“未被占用的短码”。

短码生成必须是无状态、可并行、不依赖 DB 主键分配
推荐用随机 + 冲突重试，或雪花 ID 派生（需注意时间戳部分可能泄露）
线上服务建议加布隆过滤器预判冲突，但小规模服务直接 INSERT ... ON CONFLICT DO NOTHING 更直白

如何用 Go + SQLite 实现基础短链写入

SQLite 足够支撑日均百万以下请求，且免运维。重点不是选型，而是怎么让 INSERT 不因短码重复失败。

核心逻辑：生成候选短码 → 尝试插入 → 若已存在则重试（带上限）。不要用 SELECT 判断是否存在，那是竞态源头。

func (s *Service) CreateShortURL(longURL string) (string, error) {
    const maxRetries = 5
    for i := 0; i < maxRetries; i++ {
        code := randString(6) // 例如：aB3xK9
        _, err := s.db.Exec("INSERT INTO urls (code, long_url, created_at) VALUES (?, ?, ?)",
            code, longURL, time.Now().UTC())
        if err == nil {
            return code, nil
        }
        if sqlite.ErrConstraint(err) && strings.Contains(err.Error(), "UNIQUE") {
            continue // 短码冲突，重试
        }
        return "", err
    }
    return "", errors.New("failed to generate unique short code after retries")
}

randString(6) 建议用 math/rand + 字符集（a-z, A-Z, 0-9），避免 0/O/l/I 等易混淆字符
SQLite 的 UNIQUE 约束报错信息含 "UNIQUE constraint failed"，不能只靠 sqlite.ErrConstraint
生产环境应加 context.Context 控制超时，避免卡死在重试循环里

HTTP 路由怎么处理 302 跳转与 /api/shorten 分离

短链访问路径如 /abc123 必须 302 跳转，而创建接口是 POST /api/shorten。二者不能共用一个 handler，否则 GET /api/shorten 会误触发跳转。

用 http.ServeMux 或 gorilla/mux 都行，关键是路由顺序和通配符控制：

func main() {
    mux := http.NewServeMux()
    mux.HandleFunc("POST /api/shorten", handleShorten)
    mux.HandleFunc("GET /", handleRoot) // 可选首页
    mux.HandleFunc("GET /{code}", handleRedirect) // 注意：标准 net/http 不支持 {code}，需用 gorilla/mux 或 chi

    // 若坚持用 net/http，可：
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        if r.Method == "POST" && r.URL.Path == "/api/shorten" {
            handleShorten(w, r)
            return
        }
        if r.Method == "GET" && len(r.URL.Path) > 1 {
            code := strings.TrimPrefix(r.URL.Path, "/")
            if len(code) >= 4 && len(code) <= 8 && isAlnum(code) {
                handleRedirect(w, r, code)
                return
            }
        }
        http.NotFound(w, r)
    })
}

handleRedirect 中务必校验 code 长度和字符范围，防止路径遍历或 SQL 注入（即使用了参数化查询，也要防无效请求放大）
跳转前查 DB，命中则 http.Redirect(w, r, longURL, http.StatusFound)；未命中返回 404
别忘了设置 Cache-Control: no-cache，避免 CDN 缓存错误跳转

Redis 缓存怎么避免缓存穿透和雪崩

短链跳转是典型读多写少场景，加 Redis 能扛住突发流量，但两个坑必须填：

缓存穿透：恶意请求大量不存在的 code，每次都打到 DB。解决方案：对空结果也缓存（如 SET short:xyz "" EX 60），值为空字符串，过期时间设短（60s）
缓存雪崩：所有 key 同一时刻过期，DB 瞬间被打满。解决方案：写入时加随机偏移，比如 EX 3600 + rand.Intn(600)

Go 中用 github.com/go-redis/redis/v9 时，注意 GET 返回 redis.Nil 表示 key 不存在，不是 error：

val, err := rdb.Get(ctx, "short:"+code).Result()
if err == redis.Nil {
    // 查 DB，若存在则 SET，若不存在则 SET 空值
} else if err != nil {
    // 真实错误
} else {
    // 命中缓存，跳转
}

短码生成、存储、跳转三个环节里，最易被忽略的是「跳转响应头」：必须用 http.StatusFound（302），不是 http.StatusMovedPermanently（301），否则浏览器会强缓存跳转目标，改链接后用户刷不出来。

# mysql # go # golang # ai # 路由 # golang开发 # 为什么 # red # sql # select # Error # math # 字符串 # 循环 # 接口 # nil # 并发 # github # sqlite # redis # postgresql # 数据库 # http # 跳转 # 重试 # 的是 # 遍历 # 不存在 # 化生 # 加布 # 那是 # 也要 # 用了

相关栏目：【网站优化151355 】【网络推广146373 】【网络技术251813 】【 AI营销90571 】

上一篇：sublime怎么修改默认的新文件模板_sublime新文件

下一篇：sublime怎么处理csv文件_sublime CSV文件

相关栏目网站优化
网络推广
网络技术
AI营销

最新文章 Sublime怎么一键压缩JS代码 Su sublime如何在搜索中使用正则表达式 Sublime如何设置透明窗口效果 Su mysql如何设计商品表结构_mysql css属性背景图不显示怎么办_通过检查路如何使用Golang实现排序_Golan 农历闰月是怎么回事_为合回归年加一月调整塑造《刺客信条》艾吉奥传奇的编剧离开育碧 1英里等于多少公里 1mile和km的换 css grid布局中行和列是如何定义的 PS批量旋转和翻转图片，快速校正图片方向 C# Swagger UI自定义方法 C OPPO手机九宫格和全键盘怎么切换_OP Go语言如何实现用户登录注册_Golan 1节飞行速度多少公里每小时 1节是多少公纸嫁衣8千子树第五章庙门怎么开启庙门 Laravel 多行数据编辑表单中实现逐明日之后如何提升钓鱼等级明日之后提升钓支付宝怎样查年度账单_支付宝年度账单查看 C# 多线程UI更新Dispatcher

上一篇：sublime怎么修改默认的新文件模板_sublime新文件

下一篇：sublime怎么处理csv文件_sublime CSV文件