Go负载均衡核心是客户端/网关层主动选节点,含轮询(原子计数器取模)、加权随机(总权重内采样)、健康检查(异步探测+动态剔除)及HTTP代理集成。

Go 语言实现负载均衡的核心在于:在客户端或网关层主动选择后端节点,而非依赖外部代理。关键不是“造轮子”,而是理解策略、连接管理与错误处理的结合点。
基础轮询(Round Robin)实现
最简单也最常用的策略,适合后端节点性能相近的场景。用一个原子计数器避免并发冲突:
- 维护一个服务地址切片,例如
[]string{"http://node1:8080", "http://node2:8080"} - 用
sync/atomic增加索引,取模确保循环:idx := atomic.AddUint64(&counter, 1) % uint64(len(nodes)) - 每次请求前获取当前节点,再发起 HTTP 调用(建议用
http.Client复用连接)
加权随机(Weighted Random)适配不均等节点
当某些节点配置更高、能承担更多流量时,可按权重分配。不需排序或累积数组,用“总权重内随机采样”更轻量:
- 定义结构体:
type Node struct { Addr string; Weight int } - 预计算总权重和;生成
[0, totalWeight)内随机整数 - 遍历节点,累加权重直到超过随机值,即为选中节点
- 注意:权重更新需加锁,但读多写少时可用
sync.RWMutex
健康检查 + 动态剔除失效节点
静态列表无法应对节点宕机。需异步探测并维护可用节点池:
立即学习“go语言免费学习笔记(深入)”;
- 启动 goroutine 定期对每个节点发 HEAD 或健康接口(如
/health) - 失败连续 N 次则标记为不可用,并从可用列表中移除;恢复后重新加入
- 请求分发时只从当前可用列表中选节点;若列表为空,可返回 503 或降级到本地缓存/默认响应
- 推荐使用
context.WithTimeout控制探测超时,避免阻塞
集成到 HTTP 代理服务(简易反向代理)
把负载均衡逻辑嵌入 http.Handler,对外暴露统一入口:
- 用
httputil.NewSingleHostReverseProxy创建代理,重写Director函数 - 在
Director中调用你的选择逻辑(如轮询),设置req.URL.Host和req.URL.Scheme - 添加日志、超时控制(
proxy.Transport设置ResponseHeaderTimeout等) - 可进一步支持路径前缀路由、Header 透传、熔断(如用
gobreaker库)










