使用令牌桶算法通过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），在每次写入前按字节数消耗令牌，即可实现写入带宽控制。
使用 uber-go/dig 在 Golang 中实现依赖注入——基于反射容器，提供 Provide/Invoke、dig.In/dig.Out 参数与结果对象，支持命名依赖等
package main
import (
"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/s
func 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），避免影响正常体验。
