使用令牌桶算法通过rate.Limiter实现带宽控制，封装LimitedWriter和LimitedReader可对网络读写进行限流，适用于TCP、HTTP等场景，结合io.Copy等操作透明限速，关键在于字节与令牌的映射及合理设置速率与突发容量。
在高并发服务中，控制网络带宽是防止系统过载、保障服务质量的重要手段。Golang 本身没有内置的带宽限流机制，但我们可以借助 token bucket（令牌桶） 算法结合 io.Reader/io.Writer 的包装方式，实现高效的带宽限制。下面介绍一种实用且易于集成的实现方式。
使用 golang.org/x/time/rate 进行速率控制
Golang 官方扩展包 golang.org/x/time/rate 提供了基于令牌桶的限流器，虽然它主要用于请求频率控制，但也可以用于控制数据传输速率。
核心思路是：在每次读取或写入数据前，先从限流器中“申请”相应字节数对应的通行权限，如果当前令牌不足，则等待直到有足够的令牌释放。
示例：封装一个带带宽限制的 io.Writer
立即学习“go语言免费学习笔记（深入）”；
通过包装底层的 io.Writer（如 net.Conn），在每次写入前按字节数消耗令牌，即可实现写入带宽控制。
package mainimport ( "golang.org/x/time/rate" "io" "net" "time")// LimitedWriter 包装原始 Writer 并添加带宽限制type LimitedWriter struct { Writer io.Writer Limiter *rate.Limiter}func (lw *LimitedWriter) Write(p []byte) (n int, err error) { // 计算需要等待的时间以写入 len(p) 字节 duration := lw.Limiter.ReserveN(time.Now(), len(p)).Delay() if duration > 0 { time.Sleep(duration) } return lw.Writer.Write(p)}// 使用示例：限制 TCP 写出速度为 1MB/sfunc startLimitedServer() { listener, _ := net.Listen("tcp", ":8080") defer listener.Close() limiter := rate.NewLimiter(rate.Limit(1<<20), 1<<17) // 1MB/s, burst 128KB for { conn, _ := listener.Accept() go func(c net.Conn) { defer c.Close() writer := &LimitedWriter{ Writer: c, Limiter: limiter, } // 模拟发送大量数据 data := make([]byte, 1<<20) // 1MB for i := 0; i < 10; i++ { writer.Write(data) } }(conn) }}
双向限流：同时控制上传和下载
实际场景中，可能需要分别控制连接的读（接收）和写（发送）带宽。可以分别为 io.Reader 和 io.Writer 实现限流包装器。
type LimitedReader struct { Reader io.Reader Limiter *rate.Limiter}func (lr *LimitedReader) Read(p []byte) (n int, err error) { n, err = lr.Reader.Read(p) if n > 0 { duration := lr.Limiter.ReserveN(time.Now(), n).Delay() if duration > 0 { time.Sleep(duration) } } return}
这样可以在建立网络连接后，将原始连接包装成：
LimitedReader + Original Writer：限制上传速度（客户端接收）Original Reader + LimitedWriter：限制下载速度（客户端发送）
应用到 HTTP 服务或文件传输
该方法不仅适用于 TCP，也可用于 HTTP 响应体传输、大文件下载等场景。
例如，在 HTTP 服务中限制每个响应的下载速度：
http.HandleFunc("/download", func(w http.ResponseWriter, r *http.Request) { limiter := rate.NewLimiter(100*1024, 50*1024) // 100KB/s writer := &LimitedWriter{Writer: w, Limiter: limiter} file, _ := os.Open("largefile.zip") defer file.Close() io.Copy(writer, file) // 数据流会被限速})
这种方式对业务逻辑透明，只需替换 Writer 即可生效。
基本上就这些。通过组合 rate.Limiter 与接口包装，Golang 能轻松实现细粒度的带宽控制，无需依赖第三方库，适合嵌入各类网络服务中。关键在于理解“每字节对应一个令牌”的映射关系，并合理设置限流速率与突发容量（burst），避免影响正常体验。
以上就是如何用Golang实现网络带宽限流_Golang 网络带宽限流实践的详细内容，更多请关注创想鸟其它相关文章！
版权声明：本文内容由互联网用户自发贡献，该文观点仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。
如发现本站有涉嫌抄袭侵权/违法违规的内容， 请发送邮件至 chuangxiangniao@163.com 举报，一经查实，本站将立刻删除。
发布者：程序猿，转转请注明出处：https://www.chuangxiangniao.com/p/1424432.html
微信扫一扫 
支付宝扫一扫 
