Fix perf-produce cannot be closed (#255)

* Fix perf-produce cannot be closed Signed-off-by: xiaolong.ran <rxl@apache.org> * fix comments Signed-off-by: xiaolong.ran <rxl@apache.org> * fix comments Signed-off-by: xiaolong.ran <rxl@apache.org> * fix comments Signed-off-by: xiaolong.ran <rxl@apache.org>
apache · May 19, 2020 · e6d73af · e6d73af
1 parent 77f0806
commit e6d73af
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/perf/perf-producer.go b/perf/perf-producer.go
@@ -98,15 +98,15 @@ func produce(produceArgs *ProduceArgs, stop <-chan struct{}) {
 
 	ch := make(chan float64)
 
-	go func() {
+	go func(stopCh <-chan struct{}) {
 		var rateLimiter *rate.RateLimiter
 		if produceArgs.Rate > 0 {
 			rateLimiter = rate.New(produceArgs.Rate, time.Second)
 		}
 
 		for {
 			select {
-			case <-stop:
+			case <-stopCh:
 				return
 			default:
 			}
@@ -128,7 +128,7 @@ func produce(produceArgs *ProduceArgs, stop <-chan struct{}) {
 				ch <- latency
 			})
 		}
-	}()
+	}(stop)
 
 	// Print stats of the publish rate and latencies
 	tick := time.NewTicker(10 * time.Second)