AB测试路由必须在HTTP中间件层拦截,通过一致性哈希等策略前置分流并注入ab_group到context,避免handler内耦合判断、CDN缓存错乱及用户分组跳变。
AB测试路由必须在HTTP中间件层拦截,不能只靠handler分支
Go 的
本身不支持路径+权重/用户标签的动态分流,硬写两个 handler 并在其中 if-else 判断流量归属,会导致逻辑耦合、指标埋点分散、灰度策略无法统一管控。真实项目里 AB 测试路由的核心动作是:在请求进入业务逻辑前,就决定它该进 A 版本还是 B 版本,并把决策结果(如
)注入到
或
中。
推荐做法是用自定义中间件包裹所有待测试的路由,例如:
关键点:
分流逻辑(权重、用户ID哈希、cookie读取)必须前置,避免 handler 内重复解析
中间件应支持多种分流策略,但同一路径下不要混用「按 cookie」和「按 header」,否则难以复现问题
务必设置
响应头,防止 CDN 缓存导致 AB 流量错乱
用 consistent hash 实现用户级稳定分组,避免来回跳变
简单用
做随机分流,会导致同一用户刷新页面时在 A/B 间反复切换,破坏体验且无法做行为归因。正确做法是对用户标识(如
、
header 或加密后的 cookie)做一致性哈希,映射到固定分组。
立即学习
“
go语言免费学习笔记(深入)
”;
示例逻辑片段:
注意:
不要直接对原始用户邮箱或手机号哈希——可能含敏感信息且格式不一;建议先用服务端密钥签名再哈希,或统一转成
(整型)再算
如果使用 cookie 分流(如
),需校验签名,防止客户端伪造
新增 group 时,一致性哈希能保证大部分老用户分组不变;但若用纯取模(
),加减 group 会导致全局重分配
gorilla/mux 或 chi 路由器更适合 AB 场景,标准 net/http 不够用
只支持前缀匹配,无法处理
这类带 query 参数的显式分流调试,也不方便给某条路由单独挂载中间件。实际项目中基本都会换用更灵活的路由器。
go语言参考手册 中文CHM版
Go 是一个开源的编程语言,它能让构造简单、可靠且高效的软件变得容易。本文给大家带来Go参考手册,需要的可以来下载! Go是从2007年末由Robert Griesemer, Rob Pike, Ken Thompson主持开发,后来还加入了Ian Lance Taylor, Russ Cox等人,并最终于2009年11月开源,在2012年早些时候发布了Go 1稳定版本。现在Go的开发已经是完全开放的,并且拥有一个活跃的社区。 Go 语言特色 简洁、快速、安全 并行、有趣、开源 内存管理、v数组安全、编译
下载
比如用
实现可调试的 AB 路由:
常见选型对比:
:支持正则 path、query 匹配,
可提取
参数,适合需要精细控制匹配条件的场景
:中间件链天然支持嵌套,
可为子树统一加 AB 中间件,代码组织更清晰
别用
:不支持中间件,分流逻辑只能塞进 handler,后期难维护
AB 路由必须透传原始请求信息,否则下游服务无法做日志对齐
很多团队在中间件里直接
或
后转发,结果下游服务看到的
是修改过的,日志、监控、trace ID 都对不上。AB 路由不是反向代理,不该改请求主体。
正确做法是保留原始
不变,只通过 context 注入分流结果:
这样下游 handler 可以安全读取:
容易被忽略的细节:
如果用了 gRPC-gateway,HTTP 层的 AB 决策要透传到 gRPC metadata,否则后端服务拿不到分组信息
所有中间件(鉴权、限流、日志)都得在 AB 中间件之后注册,否则它们看到的 context 里没有
别在 AB 中间件里调用
—— 会消耗 body,后续 handler 读不到数据
AB 路由最难的不是怎么分,而是怎么让整个调用链(HTTP → service → DB → cache)都感知并尊重这个分组,同时不破坏原有可观测性体系。随便加个 if-else 很快就会变成线上排查黑洞。
http.ServeMuxab_groupcontext.Contexthttp.Requesthttp.Handle("/api/search", abMiddleware(
http.HandlerFunc(searchV1Handler),
http.HandlerFunc(searchV2Handler),
WithWeight(70, 30),
WithCookieKey("ab_search"),
))Cache-Control: privaterand.Intn(100)user_idX-User-IDfunc getUserGroup(userID string, groups []string) string {
h := fnv.New64a()
h.Write([]byte(userID))
hashVal := h.Sum64() % uint64(len(groups))
return groups[hashVal]
}user_idab_search=group_b% len(groups)net/http.ServeMux/api/search?ab=v2chir.Get("/api/search", abHandler(
searchV1,
searchV2,
chi.Middlewares{
// 允许 ?ab=v1 强制指定版本(仅 dev/staging)
forceABVersion(),
},
))gorilla/muxRoute.GetQueries()abchiWith()httprouterhttp.Redirecthttp.StripPrefixreq.URL.Pathreqctx := context.WithValue(r.Context(), ctxKeyABGroup, "group_b")
r = r.WithContext(ctx)
next.ServeHTTP(w, r) // 原样传下去group := r.Context().Value(ctxKeyABGroup).(string)
log.Printf("handling request in %s", group)ab_groupreq.Body.Read()