Przeglądaj źródła

Fix panic on producer performance tool

AJ Yoo 7 lat temu
rodzic
commit
05e0d8b086
1 zmienionych plików z 3 dodań i 0 usunięć
  1. 3 0
      tools/kafka-producer-performance/main.go

+ 3 - 0
tools/kafka-producer-performance/main.go

@@ -361,6 +361,9 @@ func runSyncProducer(topic string, partition, messageLoad, messageSize, routines
 }
 }
 
 
 func printMetrics(w io.Writer, r metrics.Registry) {
 func printMetrics(w io.Writer, r metrics.Registry) {
+	if r.Get("record-send-rate") == nil || r.Get("request-latency-in-ms") == nil {
+		return
+	}
 	recordSendRate := r.Get("record-send-rate").(metrics.Meter).Snapshot()
 	recordSendRate := r.Get("record-send-rate").(metrics.Meter).Snapshot()
 	requestLatency := r.Get("request-latency-in-ms").(metrics.Histogram).Snapshot()
 	requestLatency := r.Get("request-latency-in-ms").(metrics.Histogram).Snapshot()
 	requestLatencyPercentiles := requestLatency.Percentiles([]float64{0.5, 0.75, 0.95, 0.99, 0.999})
 	requestLatencyPercentiles := requestLatency.Percentiles([]float64{0.5, 0.75, 0.95, 0.99, 0.999})