Merge pull request #93 from kudlam/master

Fixed partition key for batch
Azure · Feb 6, 2019 · b14f5f2 · b14f5f2
2 parents 3d4fed8 + f1747bd
commit b14f5f2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/event.go b/event.go
@@ -147,6 +147,11 @@ func (b *EventBatch) toEvent() (*Event, error) {
 		Format: batchMessageFormat,
 	}
 
+	if b.PartitionKey != nil {
+		msg.Annotations = make(amqp.Annotations)
+		msg.Annotations[partitionKeyAnnotationName] = b.PartitionKey
+	}
+
 	for idx, event := range b.Events {
 		innerMsg := event.toMsg()
 		bin, err := innerMsg.MarshalBinary()