讲师中心 微信公众号
AI工具推荐 视频效率加速

如何在Golang中实现一个基础的负载均衡分发器

秋明君_8678

秋明君_8678

发布时间:2026-02-13 11:36:55

|

517人浏览过

|

来源于php中文网

原创

用net/http实现轮询分发器应避免魔改RoundTripper,而用原子计数器选后端、复用全局Client、重写Host和URL、清空RequestURI、健康检查发HEAD请求、流式转发请求体并关闭Body。

如何在golang中实现一个基础的负载均衡分发器

用 net/http 实现轮询分发器,别碰 http.RoundTripper 一开始

Go 标准库的 http.Client 默认不支持后端列表切换,直接写个简单轮询逻辑比魔改 RoundTripper 更快、更可控。真实场景里你大概率只需要在请求前选一个后端地址,而不是全程接管连接复用。

  • 先定义后端列表:[]string{"http://10.0.1.10:8080", "http://10.0.1.11:8080"},别存成 *http.Client 切片——每个 client 都带独立连接池,反而干扰复用
  • 用原子计数器做无锁轮询:atomic.AddUint64(&counter, 1) % uint64(len(backends)),避免加锁和 map 查找开销
  • 别在 handler 里每次 new http.Client,复用一个全局 http.Client,并设好 Timeout 和 Transport.MaxIdleConnsPerHost

转发请求时必须重写 Host 头和 URL,否则后端收不到正确路径

Go 的 httputil.NewSingleHostReverseProxy 看似省事,但它默认把原始 Host 头透传过去,而你的负载均衡器是中间层,后端服务通常依赖 Host 做路由或 TLS SNI 匹配。直接用它容易出现 404 或证书错误。

  • 手动构造 *http.Request:用 req.URL.Path 和 req.URL.RawQuery 拼出目标 URL,别用 req.URL.String() —— 它可能含原始 host 和 scheme
  • 显式设置 req.Host = backendHost(比如 "10.0.1.10:8080"),否则后端看到的是你 LB 的域名
  • 清空 req.RequestURI(设为 ""),否则 net/http 在写入底层连接时会优先用它,导致路径错乱

健康检查不能只 ping 端口,得发真实 HTTP 请求并设超时

只用 net.DialTimeout 检查端口通不通,会漏掉后端进程僵死、HTTP server 卡住但 TCP 连接仍存活的情况。真实故障中,「端口开着但返回不了 HTTP 响应」非常常见。

Golang Uber Dig
Golang Uber Dig

使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器,提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象,支持命名依赖等

下载
  • 用单独 goroutine 轮询,每 10 秒对每个后端发一次 HEAD /health,Client.Timeout 设为 2 秒以内
  • 失败连续 3 次才标记为 down,避免瞬时抖动误判;恢复需连续 2 次成功才重新加入轮询池
  • 状态存在 map[string]bool 里没问题,但读写要加 sync.RWMutex —— 检查协程写,分发逻辑只读,别全用 sync.Mutex

别让请求体读取阻塞整个 handler,req.Body 必须用 io.Copy 流式转发

如果用 io.ReadAll(req.Body) 把整个请求体读进内存再转发,大文件上传或长流式请求(如 SSE)会吃光内存,还可能触发 http: request body too large 错误。

立即学习“go语言免费学习笔记(深入)”;

  • 用 io.Copy 直接从 req.Body 写到下游 http.Post 的请求体,或用 req.Body 作为 http.NewRequest 的 body 参数(注意:此时不能重复读)
  • 务必调用 req.Body.Close(),否则连接不会归还给连接池,短时间内就耗尽 MaxIdleConnsPerHost
  • 如果需要修改请求头(比如加 X-Forwarded-For),在 io.Copy 前做,别等 copy 完再补——body 已被消费完

最麻烦的不是轮询逻辑,而是健康状态和请求体生命周期的耦合:一个后端刚被标为 down,但已有几个请求正在往它身上转发,这些请求得自然完成,不能强行中断。所以状态切换和请求分发之间要有明确的“快照”边界,比如用原子指针切换整个后端列表切片,而不是边遍历边过滤。

热门AI工具

更多
WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

小橡皮AI

一款AI工具,主要用于小橡皮一键去AI味,将机器味重的内容改得更有活人感,防止因为 AI 内容被平台打上 AI 标签,以及限流,适合需要提升相关任务效率的用户。

切问学术

切问学术是一款AI论文写作工具,复旦大学NLP团队推出的AI学术智能体。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

UpDream
UpDream Hot

一款AI视频创作工具,主要用于哔哩哔哩推出的自研AI视频创作工具,适合需要提升相关任务效率的用户。

音述AI
音述AI Hot

一款AI音频处理工具,主要用于音述AI是一个以“用声音述说故事”为核心的 AI 音乐创作与声音分享社区,适合需要提升相关任务效率的用户。

PixTV
PixTV Hot

PixTV是一款面向AIGC内容创作的AI视频生成工具。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

Loomy
Loomy Hot

一款AI工具,主要用于科大讯飞发布的桌面级 AI 助理,比 OpenClaw 更易用、更安全!,适合需要提升相关任务效率的用户。

相关专题

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

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

519

2024.02.23

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

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

616

2024.02.23

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

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

1216

2024.02.23

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

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

791

2024.03.05

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

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

1786

2024.05.21

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

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

5074

2025.06.09

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

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

2334

2025.06.10

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

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

4226

2025.06.17

Vite 项目如何打包和部署
Vite 项目如何打包和部署

Vite项目打包部署专题汇集生产构建、产物预览、输出目录、资源路径、Nginx与GitHub Pages部署及常见白屏和路由404问题的已审核文章,帮助读者完成上线流程。

100

2026.10.28

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn