Go 基于令牌桶的限流器实现

 更新时间:2021-11-15 07:12:15   作者:佚名   我要评论(0)

目录简介原理概述具体实现原理限流器如何限流简介


如果一般流量过大,下游系统反应不过来,这个时候就需要限流了,其实和上地铁是一样的,

简介

如果一般流量过大,下游系统反应不过来,这个时候就需要限流了,其实和上地铁是一样的,就是减慢上游访问下游的速度。
限制访问服务的频次或者频率,防止服务过载,被刷爆等。

Golang 官方扩展包 time(golang.org/x/time/rate) 中,提供了一个基于令牌桶等限流器实现。

原理概述

  • 令牌:每次拿到令牌,才可访问
  • 桶 ,桶的最大容量是固定的,以固定的频率向桶内增加令牌,直至加满
  • 每个请求消耗一个令牌。
  • 限流器初始化的时候,令牌桶一般是满的。

具体使用

package limiter

import (
 "fmt"
 "testing"
 "time"

 "golang.org/x/time/rate"
)

func TestLimter(t *testing.T) {
 limiter := rate.NewLimiter(rate.Every(time.Millisecond*31), 2)
 //time.Sleep(time.Second)
 for i := 0; i < 10; i++ {
  var ok bool
  if limiter.Allow() {
   ok = true
  }
  time.Sleep(time.Millisecond * 20)
  fmt.Println(ok, limiter.Burst())
 }
}

执行结果:
=== RUN   TestLimter
true 2
true 2
true 2
false 2
true 2
true 2
false 2
true 2
true 2
false 2
--- PASS: TestLimter (0.21s)

通过执行结果可以看到, 令牌桶开始是2个满的,由于令牌的间隔比请求的间隔多了11ms(31-20), 所以每两个请求会失败一次。

具体实现原理

先看下限流器的创建方法: NewLimiter

func NewLimiter(r Limit, b int) *Limiter {
 return &Limiter{
  limit: r,
  burst: b,
 }
}

查看限流器数据结构 Limiter

// The methods AllowN, ReserveN, and WaitN consume n tokens.
type Limiter struct {
 mu     sync.Mutex
 limit  Limit
 burst  int
 tokens float64
 // last is the last time the limiter's tokens field was updated
 last time.Time
 // lastEvent is the latest time of a rate-limited event (past or future)
 lastEvent time.Time
}
  • burst 表示了桶的大小
  • limit 表示放入桶的频率
  • tokens 表示剩余令牌个数
  • last 最近取走 token 的时间
  • lastEvent 最近限流事件的时间

当令牌桶发放后,会保留在 Reservation 对象中, 定义如下, Reservation 对象,描述了一个达到 timeToAct 时间后,可以获取到的令牌的数量 tokens 数。

type Reservation struct {
  ok        bool  // 是否满足条件分配到了tokens
  lim       *Limiter // 发送令牌的限流器
  tokens    int   // tokens 的数量
  timeToAct time.Time  //  满足令牌发放的时间
  limit Limit  // 令牌发放速度
}

限流器如何限流

官方提供的限流器有阻塞等待, 也有直接判断方式的, 还有提供维护预留式等。
如何实现限流的代码,在 reserveN 中。

使用时,每次都调用了 Allow() 方法

// Allow is shorthand for AllowN(time.Now(), 1).
func (lim *Limiter) Allow() bool {
 return lim.AllowN(time.Now(), 1)
}

// AllowN reports whether n events may happen at time now.
// Use this method if you intend to drop / skip events that exceed the rate limit.
// Otherwise use Reserve or Wait.
func (lim *Limiter) AllowN(now time.Time, n int) bool {
 return lim.reserveN(now, n, 0).ok
}

继续查看 reserverN  算法

方法说明:

  • 三个参数: now, n, maxFutureReserve
  • 在 now 时间需要拿到 n 个令牌,最多等待的时间为 maxFutureReserve
  • 结果将返回一个预留令牌的对象 Reservation
// maxFutureReserve specifies the maximum reservation wait duration allowed.
// reserveN returns Reservation, not *Reservation, to avoid allocation in AllowN and WaitN.
func (lim *Limiter) reserveN(now time.Time, n int, maxFutureReserve time.Duration) Reservation {
 lim.mu.Lock()
 // 首先判断是否放入频次是否为无穷大,如果为无穷大,说明暂时不限流
 if lim.limit == Inf {
  lim.mu.Unlock()
  return Reservation{
   ok:        true,
   lim:       lim,
   tokens:    n,
   timeToAct: now,
  }
 }
   // 拿到截止 now 时间时,可以获取的令牌 tokens 数量,上一次拿走令牌的时间是last
 now, last, tokens := lim.advance(now)

 // Calculate the remaining number of tokens resulting from the request.
 //  更新 tokens数量,把需要拿走的去掉
 tokens -= float64(n)

 // Calculate the wait duration
 // 如果 tokens 数量为负数,说明需要等待,计算等待时间 WaitDuration
 var waitDuration time.Duration
 if tokens < 0 {
  waitDuration = lim.limit.durationFromTokens(-tokens)
 }

 // Decide result
 // 计算是否满足分配要求
 // 1. 需要分配的大小不超过桶容量
 // 2. 等待时间不超过设定的等待时长
 ok := n <= lim.burst && waitDuration <= maxFutureReserve

 // Prepare reservation
 // 最后构造一个 Resvervation 对象
 r := Reservation{
  ok:    ok,
  lim:   lim,
  limit: lim.limit,
 }
 if ok {
  r.tokens = n
  r.timeToAct = now.Add(waitDuration)
 }

 // Update state
   // 需要更新当前 limit 的值 
 if ok {
  lim.last = now
  lim.tokens = tokens
  lim.lastEvent = r.timeToAct
 } else {
  lim.last = last
 }

 lim.mu.Unlock()
 return r
}

从实现上看, limiter 并不是每隔一段时间更新当前桶的数量,而是记录了上次访问时和当前桶中令牌的数量,当再次访问时,通过上次访问时间计算出当前令牌的数量,决定是否可以发放令牌。

到此这篇关于Go 基于令牌桶的限流器实现的文章就介绍到这了,更多相关Go 令牌桶限流器内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家! 

您可能感兴趣的文章:
  • Golang 限流器的使用和实现示例

相关文章

  • Go 基于令牌桶的限流器实现

    Go 基于令牌桶的限流器实现

    目录简介原理概述具体实现原理限流器如何限流简介 如果一般流量过大,下游系统反应不过来,这个时候就需要限流了,其实和上地铁是一样的,
    2021-11-15
  • 超详细OpenMV与STM32单片机通信 附完整源码

    超详细OpenMV与STM32单片机通信 附完整源码

    目录1.前言(闲话)2.硬件连接3.软件代码———OpenMV端4.软件代码———STM32端5.利用PC端测试数据数据是否发送接收正常6.学习补充 (代码看不
    2021-11-15
  • GO 函数式选项模式(Functional Options Pattern)

    GO 函数式选项模式(Functional Options Pattern)

    Golang 开发者遇到的许多问题之一是尝试将一个函数的参数设置为可选. 这是一个非常常见的用例, 有些对象应该使用一些基本的默认设置来开箱即
    2021-11-15
  • 浅谈golang 的高效编码细节

    浅谈golang 的高效编码细节

    目录struct 和 map 用谁呢?字符串如何拼接是好?用 + 的方式使用 fmt.Sprintf() 的方式使用 strings.Join 的方式使用 buffer 的方式xdm,我
    2021-11-15
  • shell linux中如何用shell写一个占用CPU的脚本

    shell linux中如何用shell写一个占用CPU的脚本

    使用场景: 向公司申请的虚机资源自己工作用的比较方便,因占用较小basis要求回收掉,现写一个脚本,让CPU跑满一些。 首先看下共有几颗逻
    2021-11-15
  • shell脚本中 /dev/null 的用法小结

    shell脚本中 /dev/null 的用法小结

    /dev/null 是一个特殊的设备文件,它丢弃一切写入其中的数据 可以将它 视为一个黑洞, 它等效于只写文件, 写入其中的所有内容都会消失, 尝试从
    2021-11-15
  • Golang通脉之方法详情

    Golang通脉之方法详情

    目录方法和接收者指针类型的接收者 值类型的接收者 方法和函数任意类型添加方法方法继承方法重写结构体和方法补充方法和接收者 Go语言中
    2021-11-15
  • shell脚本多实例部署nginx的详细教程

    shell脚本多实例部署nginx的详细教程

    1. 创建一个目录,用来存放脚本和安装包 [root@localhost nginx]# tree . ├── install.sh └── packages └── nginx-1.20.1.ta
    2021-11-15
  • Python人工智能之混合高斯模型运动目标检测详解分析

    Python人工智能之混合高斯模型运动目标检测详解分析

    【人工智能项目】混合高斯模型运动目标检测 本次工作主要对视频中运动中的人或物的边缘背景进行检测。 那么走起来瓷!!! 原视频 高斯算
    2021-11-15
  • Golang通脉之流程控制详情

    Golang通脉之流程控制详情

    目录1、if else(分支结构)1.1 if条件判断基本写法1.2 if条件判断特殊写法2、for(循环结构)2.1 无限循环3、for range(键值循环)4、switch cas
    2021-11-15

最新评论