Mirroristas/etcd
2
0
Fork 0
mirror of https://github.com/etcd-io/etcd.git synced 2024-09-27 06:25:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

benchmark: Fix the panic when running benchmark watch

Browse Source 
Signed-off-by: SimFG <1142838399@qq.com>
This commit is contained in:
SimFG 2022-07-12 11:16:20 +08:00
parent 9609a46629
commit 508fa22a91
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tools/benchmark/cmd/watch.go
View File

@ -23,7 +23,7 @@ import (
"sync/atomic"
"time"
"go.etcd.io/etcd/client/v3"
clientv3 "go.etcd.io/etcd/client/v3"
"go.etcd.io/etcd/pkg/v3/report"
"github.com/cheggaaa/pb/v3"
@ -209,7 +209,12 @@ func benchPutWatches(clients []*clientv3.Client, wk *watchedKeys) {
}
}()
limit := rate.NewLimiter(rate.Limit(watchPutRate), 1)
watchPutLimit := rate.Inf
if watchPutRate > 0 {
watchPutLimit = rate.Limit(watchPutRate)
}
limit := rate.NewLimiter(watchPutLimit, 1)
for _, cc := range clients {
go func(c *clientv3.Client) {
for op := range putreqc {