Go 实现简单负载均衡器教程

本文深入讲解了如何使用 Go 语言从零实现一个具备生产可用性的简单负载均衡器，直击 http.ServeMux 无法胜任负载均衡的根本原因——它仅做路由分发，而真正的负载均衡必须在反向代理层控制后端选择、健康探测、请求透传与并发安全；文章手把手演示了基于 httputil.ReverseProxy 自定义 Director 函数实现轮询策略（强调 atomic 避免竞态）、正确透传 Host 头与真实客户端 IP（X-Forwarded-For/X-Forwarded-Proto）、以及不可或缺的 HTTP 级健康检查机制（含防抖动状态管理与线程安全缓存），帮你绕过常见坑点，构建健壮、可扩展的 Go 反向代理基础设施。

为什么不能直接用 net/http 的默认多路复用器做负载均衡

因为 http.ServeMux 只负责路由分发（path → handler），不涉及后端节点选择、健康检查或连接复用。真正的负载均衡必须在反向代理层介入，由你控制请求发给哪个上游地址。

Go 标准库的 httputil.NewSingleHostReverseProxy 是基础，但它是单目标；要支持多后端，得自己封装一个能轮询、随机或加权选择的代理逻辑。

• 直接修改 Director 函数是关键入口点，它决定 *http.Request 的 URL.Host 和 URL.Scheme
• 所有后端地址必须是完整 URL（如 "http://127.0.0.1:8081"），不能只写 host:port
• 若后端用 HTTPS，需显式配置 Transport.TLSClientConfig，否则会因证书校验失败返回 502

如何实现轮询（Round Robin）策略并避免并发安全问题

轮询看似简单，但多个 goroutine 同时调用时，index++ 会导致竞态。不能用普通 int 计数器，必须用原子操作或互斥锁。

推荐用 sync/atomic：声明一个 uint64 类型的计数器，每次用 atomic.AddUint64(&counter, 1) 获取递增值，再对后端列表长度取模。

• 初始化时把后端地址解析为 *url.URL，避免每次请求都重复解析
• 轮询索引用 atomic.LoadUint64 + atomic.StoreUint64 更安全，但 AddUint64 配合取模已足够
• 如果后端列表动态变更（如服务发现），需额外加读写锁，但静态配置下无需

var counter uint64
backends := []*url.URL{parse("http://127.0.0.1:8001"), parse("http://127.0.0.1:8002")}

director := func(req *http.Request) {
    i := atomic.AddUint64(&counter, 1) % uint64(len(backends))
    u := backends[i]
    req.URL.Scheme = u.Scheme
    req.URL.Host = u.Host
    req.URL.Path = u.Path // 注意：这里不覆盖 Path 可能导致路径拼接错误
}

如何让反向代理正确透传 Host 头和客户端真实 IP

默认的 httputil.NewSingleHostReverseProxy 会把 req.Host 改成后端地址，导致后端日志里看不到原始 Host；同时 X-Forwarded-For 也不会自动加。

必须在 Director 之后、发送请求前，通过 ReverseProxy.Transport.RoundTrip 的前置钩子（或自定义 Transport）补全这些头，但更简单的方式是在 Director 中直接设置：

• 保留原始 Host：req.Host = req.Header.Get("Host")（注意不是 req.Host 字段，那是解析后的）
• 追加 X-Forwarded-For：req.Header.Set("X-Forwarded-For", clientIP(req))，其中 clientIP 应从 X-Real-IP 或 RemoteAddr 提取并清洗
• 设置 X-Forwarded-Proto 为 req.URL.Scheme，方便后端判断是否 HTTPS

为什么健康检查不能只靠 TCP 连通性

TCP 握手成功不代表后端 HTTP 服务就绪——可能进程卡死、goroutine 泄漏、或路由未加载。真正在生产环境用，必须加 HTTP 级探测（如 GET /health），并缓存结果避免每请求都探活。

简单做法：启动时开一个 goroutine 定期轮询每个后端的 /health，维护一个 map[string]bool（host → isHealthy），在轮询选择时跳过不健康的节点。

• 探测间隔建议 5–10 秒，超时设为 2 秒以内，避免拖慢主流程
• 失败连续 3 次才标记为不可用，恢复需连续 2 次成功，防抖动
• 如果所有后端都不可用，代理应返回 503，而不是 panic 或阻塞

最易被忽略的是：健康状态缓存没加锁，或探测 goroutine 没用 time.AfterFunc 循环，导致内存泄漏或状态陈旧。

今天关于《Go 实现简单负载均衡器教程》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！