码迷,mamicode.com
首页 > 编程语言 > 详细

005_针对于go语言中速率限制的思考

时间:2018-04-17 18:03:57      阅读:301      评论:0      收藏:0      [点我收藏+]

标签:传值   输出   代码   解决方案   怎么   需求   ++   条件   3.4   

在之前的go语言的速率限制这篇文章里,我们尝试了普通的速率限制,和脉冲型速率限制。其中,脉冲型速率限制是放开了限制,里面有3个请求是一次性到达,然后再按照200ms的速度限制的,之前的代码如下所示:

package main

import "fmt"
import "time"

func main() {
	requests := make(chan int, 5)
	for i := 1; i <= 5; i++ {
		requests <- i
	}
	close(requests)
	limiter := time.Tick(time.Millisecond * 200)

	for req := range requests {
		<-limiter
		fmt.Println("request", req, time.Now())
	}

	burstyLimiter := make(chan time.Time, 3)
	for i := 0; i < 3; i++ {
		burstyLimiter <- time.Now()
	}

	go func() {
		for t := range time.Tick(time.Millisecond * 200) {
			burstyLimiter <- t
		}
	}()

	burstyRequests := make(chan int, 5)
	for i := 1; i <= 5; i++ {
		burstyRequests <- i
	}
	close(burstyRequests)
	for req := range burstyRequests {
		<-burstyLimiter
		fmt.Println("request", req, time.Now())
	}
}

  

最终的输出是下面这样,可以看到下面那段输出的前三次输出,时间几乎没差,这是一次脉冲型速率限制:

request 1 2018-04-17 12:57:02.823975218 +0800 CST m=+0.205374957
request 2 2018-04-17 12:57:03.024067833 +0800 CST m=+0.405476106
request 3 2018-04-17 12:57:03.220187209 +0800 CST m=+0.601603847
request 4 2018-04-17 12:57:03.420175881 +0800 CST m=+0.801601050
request 5 2018-04-17 12:57:03.622105704 +0800 CST m=+1.003539485

request 1 2018-04-17 12:57:03.622191244 +0800 CST m=+1.003625029
request 2 2018-04-17 12:57:03.622210962 +0800 CST m=+1.003644748
request 3 2018-04-17 12:57:03.622223153 +0800 CST m=+1.003656939
request 4 2018-04-17 12:57:03.82356235 +0800 CST m=+1.205004724
request 5 2018-04-17 12:57:04.024178896 +0800 CST m=+1.405629826

  

那我们如果想实现另一种脉冲型速率限制怎么办,就是一开始,让速度变慢,然后再正常请求的,经过思考,代码改造如下:

package main

import "fmt"
import "time"

func main() {
	k := 0
	requests := make(chan int, 5)
	for i := 1; i <= 5; i++ {
		requests <- i
	}
	close(requests)
	limiter := time.Tick(time.Millisecond * 200)
	slow := time.Tick(time.Millisecond * 400)

	for req := range requests {
		if k < 3 {
			<-slow
			k++
		}
		<-limiter
		fmt.Println("request", req, time.Now())
	}
}

  

运行结果如下,可以看到,前三次间隔是400ms,第四次又还原回来间隔200ms:

request 1 2018-04-17 17:04:50.9986303 +0800 CST m=+0.405591505
request 2 2018-04-17 17:04:51.395919457 +0800 CST m=+0.802875205
request 3 2018-04-17 17:04:51.794626945 +0800 CST m=+1.201577217
request 4 2018-04-17 17:04:51.993709916 +0800 CST m=+1.400657453
request 5 2018-04-17 17:04:52.196450634 +0800 CST m=+1.603395386

  

我来解释下代码逻辑。为了实现需求,我在前面又设置了一个打点器,然后增加判断,如果k小于3,就从打点器中取值,然后k自增。这样的话保证了前三次取值慢,后面的取值快。

这时候有人会问了,那例子中是总共两个打点器,一个是400ms,一个是200ms,为什么前三次不是400+200总共600ms,而是间隔只有400ms呢?

我来解释一下,因为代码运行到<-slow时候是阻塞状态,通道内没有值,是需要打点器每隔400ms把值存入slow中,才能继续运行的。而<-limiter按理说也是阻塞状态,也需要打点器传值的。但是别忘了,limiter是200ms,在slow传值之前,limiter就已经存入值了。所以在if判断语句结束以后,limiter不用等待传值,直接取就行了。这样的话,在if条件执行完后,<-limiter执行是瞬间完成的,不用等待200ms的。

 

以上就是go语言中速率限制的一些思考和改造,在实际工作中,应该还有更加完美的解决方案,期待将来的改进。

 

005_针对于go语言中速率限制的思考

标签:传值   输出   代码   解决方案   怎么   需求   ++   条件   3.4   

原文地址:https://www.cnblogs.com/Joestar/p/8868069.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!