不用自增 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 判断是否存在,那是竞态源头。
立即学习“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.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),否则浏览器会强缓存跳转目标,改链接后用户刷不出来。






























