From 905bfc324d599a7b529d095de88fb11677ef606d Mon Sep 17 00:00:00 2001
From: colmsnowplow <colm@snowplowanalytics.com>
Date: Tue, 6 Jul 2021 16:42:49 +0100
Subject: [PATCH] Fix how evethub partitionkey is set

---
 pkg/target/eventhub.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pkg/target/eventhub.go b/pkg/target/eventhub.go
index ff26c63c..56b03582 100644
--- 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
 	}