Golang模拟令牌桶访问限流 Golang模拟令牌桶进行对访问的限流方式

时间:2021-04-27 鹿灏楷silves 人气:0

想了解Golang模拟令牌桶进行对访问的限流方式的相关内容吗，鹿灏楷silves在本文为您仔细讲解Golang模拟令牌桶访问限流的相关知识和一些Code实例，欢迎阅读和指正，我们先划重点：Golang,模拟令牌桶,访问限流,令牌桶限流，下面大家一起来学习吧。

利用channel进行模拟令牌桶对访问进行限流

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中有值，则向下层接口发送请求。

代码如下：

package main
import (
    "fmt"
    "time"
    "httpclient"
)
var LEN int = 10
func 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)
    }
}

以上为个人经验，希望能给大家一个参考，也希望大家多多支持。如有错误或未考虑完全的地方，望不吝赐教。

加载全部内容