应使用自定义net/http.roundtripper实现可插拔负载均衡,因其能复用连接池、集成超时重试,且支持运行时更新后端与无锁健康检查;反向代理场景宜用httputil.newsinglehostreverseproxy配合gorilla/mux。

用 net/http.RoundTripper 做可插拔的负载均衡客户端
Go 标准库本身不提供负载均衡器组件,但你可以通过自定义 http.RoundTripper 实现——它才是实际发请求的底层接口,比在 http.Client 层做路由更干净、复用性更强。
常见错误是直接在 handler 里写随机选后端逻辑,结果每次请求都新建连接、无法复用连接池、也压根没考虑健康检查和重试。
- 必须把负载逻辑封装进
RoundTripper,才能让http.Client自动复用连接、走超时/重试机制 - 后端列表不能硬编码,建议用
sync.Map或atomic.Value支持运行时更新 - 别在
RoundTrip方法里做阻塞操作(比如 HTTP 健康探测),会卡住整个 client 连接池
// 示例:简单轮询 RoundTripper
type RoundRobinTransport struct {
backends []string
mu sync.RWMutex
idx uint64
}
func (r *RoundRobinTransport) RoundTrip(req *http.Request) (*http.Response, error) {
r.mu.RLock()
backend := r.backends[(int(atomic.AddUint64(&r.idx, 1)) % len(r.backends))]
r.mu.RUnlock()
u, _ := url.Parse(backend)
req.URL.Scheme = u.Scheme
req.URL.Host = u.Host
// 注意:这里没改 Path,适合反向代理场景;若需路径透传,要 clone req.URL
return http.DefaultTransport.RoundTrip(req)
}
用 gorilla/mux + httputil.NewSingleHostReverseProxy 快速搭反向代理型 LB
如果你要的是 HTTP 层的反向代理式负载均衡(比如网关),直接组合标准库 httputil.NewSingleHostReverseProxy 和路由库最省事,不用自己解析 Host/Path/Headers。
容易踩的坑是忽略 Director 函数里对 req.URL 的修改粒度:改错字段会导致 404 或循环代理。
-
Director中必须重写req.URL.Scheme和req.URL.Host,否则默认仍指向原始 Host - 如果后端是 HTTPS,
req.URL.Scheme要设为"https",且 Transport 需禁用证书校验(仅测试)或配置 CA - 不要在
Director里修改req.Header的Host字段,除非你明确要覆盖后端看到的 Host
proxy := httputil.NewSingleHostReverseProxy(&url.URL{Scheme: "http", Host: "127.0.0.1:8081"})
proxy.Director = func(req *http.Request) {
// 从轮询或一致性哈希选一个 backend
backend := pickBackend()
req.URL.Scheme = "http"
req.URL.Host = backend
// 不要 req.Header.Set("Host", backend) —— 默认已由 ReverseProxy 处理
}
健康检查失败时如何避免请求打到宕机节点
静态后端列表 + 无健康检查 = 请求必然失败。Go 没内置探活机制,得自己加定时任务+状态标记,而且状态读写必须无锁或低竞争。
典型错误是用全局 mutex 锁住整个 backend 列表做“标记下线”,结果所有请求排队等锁,延迟飙升。
- 每个 backend 独立维护
isHealthy状态,用atomic.Bool或sync/atomic操作 - 健康检查用 goroutine 定期发起轻量 HTTP HEAD 或 TCP Dial,失败连续 N 次再标记为 down
- 负载选择逻辑里跳过
!healthy.Load()的节点;若全 down,可 fallback 到本地兜底或 panic(视场景)
为什么别用 http.ServeMux 做负载路由
http.ServeMux 是路径匹配器,不是负载分发器。把它当 LB 用,本质是写一堆 if path == "/api" { proxyToA() },跟写 if-else 没区别,没法动态扩缩容,也不支持权重、故障转移。
真正需要 LB 时,要么用 RoundTripper(客户端侧),要么用 ReverseProxy(服务端侧)。ServeMux 只该负责“哪个路径交给哪个 handler”,不该掺和“这个 handler 具体连哪台机器”。
复杂点在于健康状态和路由策略(轮询/最小连接/一致性哈希)要解耦设计,否则加个新策略就得改遍所有地方。最容易被忽略的是:连接复用和健康状态必须联动——一个节点恢复后,它的连接池要能自动重建,而不是继续用旧的失效连接。
golang免费学习笔记(深入):立即使用
在学习笔记中,你将探索golang的核心概念和高级技巧!











