栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 面试经验 > 面试问答

在golang中如何限制服务器上载和下载的速度?

面试问答 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

在golang中如何限制服务器上载和下载的速度?

有一个令牌桶算法可以帮助实现这样的速率限制。我找到了一个示例实现,您可以使用它:https
//github.com/juju/ratelimit
    package main    import (        "bytes"        "fmt"        "io"        "time"        "github.com/juju/ratelimit"    )    func main() {        // Source holding 1MB        src := bytes.NewReader(make([]byte, 1024*1024))        // Destination        dst := &bytes.Buffer{}        // Bucket adding 100KB every second, holding max 100KB        bucket := ratelimit.NewBucketWithRate(100*1024, 100*1024)        start := time.Now()        // Copy source to destination, but wrap our reader with rate limited one        io.Copy(dst, ratelimit.Reader(src, bucket))        fmt.Printf("Copied %d bytes in %sn", dst.Len(), time.Since(start))    }

运行它之后,输出为:

Copied 1048576 bytes in 9.239607694s

您可以使用不同的存储桶实现来提供所需的行为。在您的代码中,设置正确的令牌桶后,您将调用:

_, err = io.Copy(out, ratelimit.Reader(file, bucket))


转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/484150.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号