From 508fa22a918c88e88622d5f8a9b270e545f2ab1f Mon Sep 17 00:00:00 2001
From: SimFG <1142838399@qq.com>
Date: Tue, 12 Jul 2022 11:16:20 +0800
Subject: [PATCH] benchmark: Fix the panic when running `benchmark watch`

Signed-off-by: SimFG <1142838399@qq.com>
---
 tools/benchmark/cmd/watch.go | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/tools/benchmark/cmd/watch.go b/tools/benchmark/cmd/watch.go
index aab3108f5..110b372db 100644
--- a/tools/benchmark/cmd/watch.go
+++ b/tools/benchmark/cmd/watch.go
@@ -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 {