有个平台的接口 QPS 和 并发数 限制都是 10,我理解的是他现在这个接口每秒最多 10 次 同时最多有 10 等待返回的请求。 但是我把 qps 设置成 9 去请求都会说我请求频率过快,一个请求需要处理的时间大概在 100ms 到 1s 范围内。麻烦大佬康康,我写了个大概的例子。
package main
import (
"context"
"fmt"
"log"
"math/rand"
"sync"
"time"
"golang.org/x/time/rate"
)
var qps = 10
var remoteChan = make(chan int, 100)
// 假装是个远程 qps 限制的接口
func remoteQPSFunc(i int) {
remoteChan <- i
log.Println(fmt.Sprintf("->Job: %d", len(remoteChan)))
t := rand.Intn(3000)
time.Sleep(time.Duration(t) * time.Millisecond)
<-remoteChan
log.Println(fmt.Sprintf("<-Job: %d", len(remoteChan)))
}
func main() {
jobChan := make(chan int, qps)
limiter := rate.NewLimiter(rate.Every(time.Duration(1000/qps)*time.Millisecond), 1)
wg := sync.WaitGroup{}
for i := range [100]int{} {
jobChan <- i
wg.Add(1)
go func(i int) {
defer wg.Done()
remoteQPSFunc(i)
limiter.Wait(context.Background())
<-jobChan
}(i)
}
wg.Wait()
}
@windyland 改了一下测试代码观察了一下,我没作对,正道的光这次没照在了大地上,原因就是网络的延迟,比如我请求在第1s发出,可能服务器在第2s收到请求
package main
import (
"context"
"fmt"
"log"
"math/rand"
"sync"
"time"
"golang.org/x/time/rate"
)
var qps = 10
var remoteChan = make(chan int, 100)
// 假装是个远程 qps 限制的接口
func remoteQPSFunc(i int) {
t1 := rand.Intn(1000)
time.Sleep(time.Duration(t1) * time.Millisecond)
remoteChan <- i
log.Println(fmt.Sprintf("Start Job: %d", len(remoteChan)))
t2 := rand.Intn(1000)
time.Sleep(time.Duration(t2) * time.Millisecond)
<-remoteChan
}
func main() {
jobChan := make(chan int, qps)
limiter := rate.NewLimiter(rate.Limit(qps), 1)
wg := sync.WaitGroup{}
for i := range [100]int{} {
jobChan <- i
limiter.Wait(context.Background())
wg.Add(1)
go func(i int) {
defer wg.Done()
remoteQPSFunc(i)
<-jobChan
}(i)
}
wg.Wait()
}
1
windyland 2020-07-26 12:27:16 +08:00 via Android
我怀疑它是偷懒直接限制了队列长度,而不是设置了 pqs 限制
|