Fix how evethub partitionkey is set

snowplow · Jul 6, 2021 · 905bfc3 · 905bfc3
1 parent 2e12c4b
commit 905bfc3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/target/eventhub.go b/pkg/target/eventhub.go
@@ -107,7 +107,7 @@ func (eht *EventHubTarget) process(messages []*models.Message) (*models.TargetWr
 	ehBatch := make([]*eventhub.Event, messageCount)
 	for i, msg := range messages {
 		ehEvent := eventhub.NewEvent(msg.Data)
-		ehEvent.Set("PartitionKey", msg.PartitionKey) // TODO: Should we be setting the partition key here? Is this the right way to do so?
+		ehEvent.PartitionKey = &msg.PartitionKey
 		ehBatch[i] = ehEvent
 	}