如何用Golang实现网络带宽限流_Golang 网络带宽限流实践-Golang-PHP中文网

如何用Golang实现网络带宽限流_Golang 网络带宽限流实践

P粉602998670

发布： 2025-11-17 16:11:02

原创

741人浏览过

使用令牌桶算法通过rate.Limiter实现带宽控制，封装LimitedWriter和LimitedReader可对网络读写进行限流，适用于TCP、HTTP等场景，结合io.Copy等操作透明限速，关键在于字节与令牌的映射及合理设置速率与突发容量。

如何用golang实现网络带宽限流_golang 网络带宽限流实践

在高并发服务中，控制网络带宽是防止系统过载、保障服务质量的重要手段。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 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 实现限流包装器。

无限画

千库网旗下AI绘画创作平台

查看详情

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 网络带宽限流实践的详细内容，更多请关注php中文网其它相关文章！