return error if wrong proto type was given or loading was not possible

birdayz · Mar 26, 2020 · 71d26b3 · 71d26b3
1 parent 7925564
commit 71d26b3
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cmd/kaf/produce.go b/cmd/kaf/produce.go
@@ -70,6 +70,8 @@ var produceCmd = &cobra.Command{
 				}
 
 				data = pb
+			} else {
+				errorExit("Failed to load payload proto type")
 			}
 		}
 
@@ -86,7 +88,10 @@ var produceCmd = &cobra.Command{
 				}
 
 				key = sarama.ByteEncoder(pb)
+			} else {
+				errorExit("Failed to load key proto type")
 			}
+
 		}
 
 		var ts time.Time
@@ -110,7 +115,6 @@ var produceCmd = &cobra.Command{
 			}
 
 			fmt.Printf("Sent record to partition %v at offset %v.\n", partition, offset)
-
 		}
 
 	},