|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
func FW(max int,duration time.Duration){//定义一个channel ，进行初始化 contain := make(chan bool , max) for i := 0 ; i < max ; i ++{ contain <- true//写入channel } go func() {//开启一个线程 for { contain <- true time.Sleep(duration) } }() for <- contain {//如果上一个线程写入一个true，就会运行这个代码块 fmt.Println("helllo world") }} |
补充：golang简易令牌桶算法实现
基本思路：
定义一个chan，chan大小为需要限制的qps大小，go一个协程启动tick，每1000/qps时间在tick中写入数值，启动另一个协程，读取chan中的值，如果读取到chan中有值，则向下层接口发送请求。
代码如下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
package mainimport ( "fmt" "time" "httpclient")var LEN int = 10func tickStoreCh(arrlen int, ch chan int) { len := 1000/arrlen fmt.Println(len) tickTime := time.NewTicker(time.Duration(len)*time.Millisecond) var i int for { fmt.Println(len) i++ <-tickTime.C ch<- i }}func OrganReq(org string, qps int) { ch := make(chan int, qps) go tickStoreCh(qps, ch) time.Sleep(1000*time.Millisecond) for { //收客户请求，发送http请求给RE client := httpclient.NewHttpClient(time.Duration(1000)*time.Millisecond, time.Duration(2000)*time.Millisecond) header := make(map[string]string) header["Content-Type"] = "application/json;charset=utf-8" code, err := client.ResponseCode("http://127.0.0.1:19988", header, "llltest") value := <- ch fmt.Println(code, value, err, "lenchan:", len(ch)) //time.Sleep(time.Second) }} |
以上为个人经验，希望能给大家一个参考，也希望大家多多支持服务器之家。如有错误或未考虑完全的地方，望不吝赐教。
原文链接：https://blog.csdn.net/Xiang_lhh/article/details/114990448
