Ver Fonte

Add missing error check on producer performance tool

AJ Yoo há 7 anos atrás
pai
commit
2ac32fdcf4
1 ficheiros alterados com 5 adições e 1 exclusões
  1. 5 1
      tools/kafka-producer-performance/main.go

+ 5 - 1
tools/kafka-producer-performance/main.go

@@ -201,7 +201,11 @@ func main() {
 	if err != nil {
 		printErrorAndExit(69, "Failed to create producer: %s", err)
 	}
-	defer producer.Close()
+	defer func() {
+		if err := producer.Close(); err != nil {
+			printErrorAndExit(69, "Failed to close producer: %s", err)
+		}
+	}()
 
 	// Construct -messageLoad messages of appoximately -messageSize random bytes.
 	messages := make([]*sarama.ProducerMessage, *messageLoad)