0

0

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

P粉602998670

P粉602998670

发布时间:2026-01-12 16:18:01

|

153人浏览过

|

来源于php中文网

原创

不用自增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 会误触发跳转。

慧博商城系统HuiboShop2011
慧博商城系统HuiboShop2011

慧博商城系统HuiboShop2011系统特色:1、上百套模板随意下载切换、模板定时更新;2、csv数据导入、数据定向读取,一键导入商品信息,省时、省力; 3、多会员等级管理,一站搞定零售、批发、代销;4、可集成网站分销功能模块,缔造庞大的代理业务链,代理客户一键铺货;5、代码严谨,防SQL注入;前后台用户分开管理,密码不可逆加密;6、简单易操作、只需几分钟搞定一个商城网站;商城后台帐号admin

下载

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

相关专题

更多
golang如何定义变量
golang如何定义变量

golang定义变量的方法:1、声明变量并赋予初始值“var age int =值”;2、声明变量但不赋初始值“var age int”;3、使用短变量声明“age :=值”等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

178

2024.02.23

golang有哪些数据转换方法
golang有哪些数据转换方法

golang数据转换方法:1、类型转换操作符;2、类型断言;3、字符串和数字之间的转换;4、JSON序列化和反序列化;5、使用标准库进行数据转换;6、使用第三方库进行数据转换;7、自定义数据转换函数。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

226

2024.02.23

golang常用库有哪些
golang常用库有哪些

golang常用库有:1、标准库;2、字符串处理库;3、网络库;4、加密库;5、压缩库;6、xml和json解析库;7、日期和时间库;8、数据库操作库;9、文件操作库;10、图像处理库。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

337

2024.02.23

golang和python的区别是什么
golang和python的区别是什么

golang和python的区别是:1、golang是一种编译型语言,而python是一种解释型语言;2、golang天生支持并发编程,而python对并发与并行的支持相对较弱等等。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

208

2024.03.05

golang是免费的吗
golang是免费的吗

golang是免费的。golang是google开发的一种静态强类型、编译型、并发型,并具有垃圾回收功能的开源编程语言,采用bsd开源协议。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

388

2024.05.21

golang结构体相关大全
golang结构体相关大全

本专题整合了golang结构体相关大全,想了解更多内容,请阅读专题下面的文章。

194

2025.06.09

golang相关判断方法
golang相关判断方法

本专题整合了golang相关判断方法,想了解更详细的相关内容,请阅读下面的文章。

189

2025.06.10

golang数组使用方法
golang数组使用方法

本专题整合了golang数组用法,想了解更多的相关内容,请阅读专题下面的文章。

192

2025.06.17

Java 项目构建与依赖管理(Maven / Gradle)
Java 项目构建与依赖管理(Maven / Gradle)

本专题系统讲解 Java 项目构建与依赖管理的完整体系,重点覆盖 Maven 与 Gradle 的核心概念、项目生命周期、依赖冲突解决、多模块项目管理、构建加速与版本发布规范。通过真实项目结构示例,帮助学习者掌握 从零搭建、维护到发布 Java 工程的标准化流程,提升在实际团队开发中的工程能力与协作效率。

9

2026.01.12

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
MySQL 教程
MySQL 教程

共48课时 | 1.7万人学习

MySQL 初学入门(mosh老师)
MySQL 初学入门(mosh老师)

共3课时 | 0.3万人学习

简单聊聊mysql8与网络通信
简单聊聊mysql8与网络通信

共1课时 | 785人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号