如何通过 io.Copy 获得数据传输速度？

IT行业相对于一般传统行业，发展更新速度更快，一旦停止了学习，很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习，精进自己的技术，尤其是初学者。今天golang学习网给大家整理了《如何通过 io.Copy 获得数据传输速度？》，聊聊，我们一起来看看吧！

我正在使用 io.copy 将网络流复制到文件。我想提取传输运行的当前速度，最好以每秒字节为单位。

res, err := http.Get(url)

if err != nil {
    panic(err)
}

// Open output file
out, err := os.OpenFile("output", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
if err != nil {
    panic(err)
}

// Close output file as well as body    

defer out.Close()

defer func(Body io.ReadCloser) {
    err := Body.Close()
    if err != nil {
        panic(err)
    }
}(res.Body)

_, err := io.Copy(out, res.Body)

正确答案

正如评论中所指出的 - 整个传输速率很容易在事后计算 - 特别是在使用 io.copy 时。如果您想跟踪“实时”传输速率 - 并轮询长时间文件传输的结果 - 那么需要做更多的工作。

下面我概述了一个简单的 io.reader 包装器来跟踪总体传输速率。为了简洁起见，它不是 goroutine 安全的，但这样做是微不足道的。然后可以从另一个 goroutine 轮询进度，而主 goroutine 进行读取。

您可以创建一个 io.reader 包装器 - 并使用它来跟踪首次读取的时刻 - 然后跟踪未来读取的字节计数。最终结果可能如下所示：

r := newrater(resp.body) // io.reader wrapper

n, err := io.copy(out, r)

log.print(r) // stringer method shows human readable "b/s" output

要实现这一点，一种方法是：

type rate struct {
    r          io.reader
    count      int64 // may have large (2gb+) files - so don't use int
    start, end time.time
}

func newrater(r io.reader) *rate { return &rate{r: r} }

然后我们需要包装器 read 来跟踪底层 io.readers 进度：

func (r *rate) read(b []byte) (n int, err error) {

    if r.start.iszero() {
        r.start = time.now()
    }

    n, err = r.r.read(b) // underlying io.reader read

    r.count += int64(n)

    if err == io.eof {
        r.end = time.now()
    }

    return
}

可以像这样轮询任何时间的速率 - 甚至在 eof 之前：

func (r *rate) rate() (n int64, d time.duration) {
    end := r.rend
    if end.iszero() {
        end = time.now()
    }
    return r.count, end.sub(r.start)
}

和一个简单的 stringer 方法来显示 b/s：

func (r *rate) String() string {
    n, d := r.Rate()
    return fmt.Sprintf("%.0f b/s", float64(n)/(d.Seconds()))
}

注意：上面的 io.reader 包装器没有锁定到位，因此操作必须来自同一个 goroutine。由于问题涉及 io.copy - 那么这是一个安全的假设。

本篇关于《如何通过 io.Copy 获得数据传输速度？》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！