浏览代码

Add missing error check on producer performance tool

AJ Yoo 7 年之前
父节点
当前提交
2ac32fdcf4
共有 1 个文件被更改,包括 5 次插入1 次删除
  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)