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

8次阅读

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

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

为什么 不用自增 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 判断是否存在,那是竞态源头。

立即学习go 语言免费学习笔记(深入)”;

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.ServeMuxgorilla/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),否则浏览器会强缓存跳转目标,改链接后用户刷不出来。

text=ZqhQzanResources