Fix how evethub partitionkey is set

colmsnowplow · colmsnowplow · commit 905bfc324d59 · 2021-07-06T16:42:49.000+01:00
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
 	}
 

Original file line number	Diff line number	Diff line change
`@@ -107,7 +107,7 @@ func (eht EventHubTarget) process(messages []models.Message) (*models.TargetWr`
`107`	`107`	`ehBatch := make([]*eventhub.Event, messageCount)`
`108`	`108`	`for i, msg := range messages {`
`109`	`109`	`ehEvent := eventhub.NewEvent(msg.Data)`
`110`		`- ehEvent.Set("PartitionKey", msg.PartitionKey) // TODO: Should we be setting the partition key here? Is this the right way to do so?`
	`110`	`+ ehEvent.PartitionKey = &msg.PartitionKey`
`111`	`111`	`ehBatch[i] = ehEvent`
`112`	`112`	`}`
`113`	`113`