produce: tweaks

fgeller · Sep 9, 2023 · 5b7390c · 5b7390c
1 parent 5f40182
commit 5b7390c
Showing 1 changed file with 5 additions and 8 deletions.
diff --git a/produce.go b/produce.go
@@ -346,16 +346,13 @@ func (cmd *produceCmd) deserializeLines(in chan string, out chan message, partit
 			}
 
 			var part int32 = 0
-			if msg.Value != nil && cmd.partitioner == "hashCodeByValue" {
-				part = hashCodePartition(*msg.Value, partitionCount)
-				msg.Partition = &part
-			} else {
-				if msg.Key != nil && cmd.partitioner == "hashCode" {
+			if msg.Partition == nil {
+				if msg.Value != nil && cmd.partitioner == "hashCodeByValue" {
+					part = hashCodePartition(*msg.Value, partitionCount)
+				} else if msg.Key != nil && cmd.partitioner == "hashCode" {
 					part = hashCodePartition(*msg.Key, partitionCount)
 				}
-				if msg.Partition == nil {
-					msg.Partition = &part
-				}
+				msg.Partition = &part
 			}
 
 			out <- msg