Golang官方限流器库实现限流示例详解

 更新时间:2022年08月18日 11:17:56   作者:jiaxwu  
这篇文章主要为大家介绍了Golang官方限流器库使用示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪

脚本之家 / 编程助手:解决程序员“几乎”所有问题!
脚本之家官方知识库 → 点击立即使用

前言

在翻Golang官方库的过程中,发现一个有趣的库golang.org/x/time ,里面只有一个类rate,研究了一下发现它是一个限流器,实现了很多的功能,当然它的核心原理并不复杂,也就是令牌桶算法。

令牌桶算法的原理是:令牌桶会不断地把令牌添加到桶里,而请求会从桶中获取令牌,只有拥有令牌地请求才能被接受。因为桶中可以提前保留一些令牌,所以它允许一定地突发流量通过。

例子

下面是限流算法常见的写法,首先判断是否有令牌,如果有就通过,否则直接失败。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
package main
import (
    "fmt"
    "time"
    "golang.org/x/time/rate"
)
func main() {
        // 每0.1秒生成一个令牌,也就是一秒10个令牌,最大保留令牌上限10
    l := rate.NewLimiter(rate.Every(time.Second/10), 10)
    for i := 0; i < 10; i++ {
        go func(i int) {
            for {
                                // 判断是否有令牌,如果有就输出
                if l.Allow() {
                    fmt.Printf("allow %d\n", i)
                }
                                // 每0.5秒请求一次
                time.Sleep(time.Second / 2)
            }
        }(i)
    }
    time.Sleep(time.Second * 10)
}

上面的rate.Every(time.Second/10)会返回一个Limit类型,代表每秒生成多少个令牌。

这个库还提供了另外一种写法,等待直到有令牌为止(或超时):

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
func main() {
    l := rate.NewLimiter(rate.Every(time.Second/10), 100)
    for i := 0; i < 10; i++ {
        go func(i int) {
            for {
                                // 等待直到有令牌
                if err := l.Wait(context.TODO()); err != nil {
                } else {
                    fmt.Printf("allow %d\n", i)
                }
                time.Sleep(time.Second / 2)
            }
        }(i)
    }
    time.Sleep(time.Second * 10)
}

这样在某些场景下我们可以让请求等待一会,而不是直接失败。

还有一个更加特殊的请求令牌方式,也就是先预留令牌,到指定时间不再需要去获取令牌,直接执行操作即可:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
func main() {
    l := rate.NewLimiter(rate.Every(time.Second/10), 10)
    for i := 0; i < 10; i++ {
        go func(i int) {
            for {
                                // 先预留令牌
                if r := l.Reserve(); r.OK() {
                                        // 休眠直到令牌生效
                    time.Sleep(r.Delay())
                    fmt.Printf("allow %d\n", i)
                }
                time.Sleep(time.Second / 2)
            }
        }(i)
    }
    time.Sleep(time.Second * 10)
}

当然,如果预留的令牌不想使用了,也可以使用r.Cancel()归还已预留的令牌。

上面的Allow()、Wait()、Reserve()都是一次消耗一个令牌,其实都有对应的AllowN()、WaitN()、ReserveN()方法,一次消耗N个令牌,这样就可以根据任务消耗的资源灵活的消耗令牌。

实现

不管我们从Allow()、Wait()还是Reserve()进去,最终都会进入到reserveN(now time.Time, n int, maxFutureReserve time.Duration) Reservation 方法:

首先在进入方法的时候,会先处理两种特殊情况:

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
    // 如果无限生成令牌,则直接返回
    if lim.limit == Inf {
    return Reservation{
        ok:        true,
        lim:       lim,
        tokens:    n,
        timeToAct: now,
    }
    // 如果不会生成令牌,则在初始令牌里面拿,直到拿完为止
} else if lim.limit == 0 {
    var ok bool
    if lim.burst >= n {
        ok = true
        lim.burst -= n
    }
    return Reservation{
        ok:        ok,
        lim:       lim,
        tokens:    lim.burst,
        timeToAct: now,
    }
}

然后重新计算当前有多少令牌,减去要消耗的令牌:

1
2
3
4
5
6
7
8
9
10
11
    // 计算当前有多少令牌
now, last, tokens := lim.advance(now)
// 减去要消耗的N个令牌
tokens -= float64(n)
    // 如果剩余令牌为负数,那么计算一下要等待多久才能拿到令牌
    var waitDuration time.Duration
if tokens < 0 {
    waitDuration = lim.limit.durationFromTokens(-tokens)
}
    // 判断请求是否成功,maxFutureReserve代表最大可以等待的时间,也就是请求能否接收拿到令牌需要等待的时间
    ok := n <= lim.burst && waitDuration <= maxFutureReserve

余下的代码就是更新限流器的状态。

小结

可以看到这个令牌桶限流器实现的功能非常的丰富,如果需要令牌桶限流器,可以优先考虑使用这个实现。

以上就是Golang官方限流器库使用示例详解的详细内容,更多关于Golang官方限流器库的资料请关注脚本之家其它相关文章!

蓄力AI

微信公众号搜索 “ 脚本之家 ” ,选择关注

程序猿的那些事、送书等活动等着你

原文链接:https://juejin.cn/post/7130890097947312158

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。
如若内容造成侵权/违法违规/事实不符,请将相关资料发送至 reterry123@163.com 进行投诉反馈,一经查实,立即处理!

相关文章

  • go语言中Timer和Ticker两种计时器的使用

    go语言中Timer和Ticker两种计时器的使用

    go语言中有Timer和Ticker这样的两种计时器,两种计时器分别实现了不同的计时功能,本文主要介绍了go语言中Timer和Ticker两种计时器的使用,感兴趣的可以了解一下
    2024-08-08
  • Go语言中的自定义类型你了解吗

    Go语言中的自定义类型你了解吗

    自定义类型是 Go 语言中非常重要的概念之一,通过自定义类型,我们可以更好地封装数据、组织代码,提高程序的可读性和可维护性。本文将从以下几个方面介绍 Go 自定义类型的相关知识,感兴趣的可以了解一下
    2023-04-04
  • go语言区块链学习调用智能合约

    go语言区块链学习调用智能合约

    这篇文章主要为大家介绍了go语言区块链学习中如何调用智能合约的实现示例,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步
    2021-10-10
  • golang redis中Pipeline通道的使用详解

    golang redis中Pipeline通道的使用详解

    本文主要介绍了golang redis中Pipeline通道的使用详解,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧
    2022-06-06
  • 一文带你搞懂Go如何读写Excel文件

    一文带你搞懂Go如何读写Excel文件

    Excelize是一个用纯Go语言编写的库,提供了一组函数,可以对XLAM / XLSM / XLSX / XLTM / XLTX文件进行读写。支持读写由Microsoft Excel™2007及以后版本生成的电子表格文档。本文就将用它实现读写Excel文件操作,感兴趣的可以学习一下
    2022-11-11
  • 源码解析gtoken替换jwt实现sso登录

    源码解析gtoken替换jwt实现sso登录

    这篇文章主要为大家介绍了源码解析gtoken替换jwt实现sso登录的示例,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2022-06-06
  • 三种Golang数组拷贝方式及性能分析详解

    三种Golang数组拷贝方式及性能分析详解

    在Go语言中,我们可以使用for、append()和copy()进行数组拷贝。这篇文章主要为大家详细介绍一下这三种方式的具体实现与性能分析,需要的可以参考一下
    2022-08-08
  • Golang中interface的基本用法详解

    Golang中interface的基本用法详解

    Go 中接口也是一个使用得非常频繁的特性,好的软件设计往往离不开接口的使用,比如依赖倒置原则(通过抽象出接口,分离了具体实现与实际使用的耦合)。 今天,就让我们来了解一下 Go 中接口的一些基本用法
    2023-01-01
  • Go语言中new()和 make()的区别详解

    Go语言中new()和 make()的区别详解

    这篇文章主要介绍了Go语言中new()和 make()的区别详解,本文讲解了new 的主要特性、make 的主要特性,并对它们的区别做了总结,需要的朋友可以参考下
    2014-10-10
  • golang recover函数使用中的一些坑解析

    golang recover函数使用中的一些坑解析

    这篇文章主要为大家介绍了golang recover函数使用中的一些坑解析,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
    2023-02-02

最新评论