feat(messenger/publisher): make ctx required

mgjules · Apr 11, 2022 · 4716739 · 4716739
1 parent 9fcae99
commit 4716739
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/transport/messenger/publisher.go b/transport/messenger/publisher.go
@@ -63,8 +63,12 @@ func NewTestPublisher(logger *logger.Logger) *Publisher {
 }
 
 // Publish is a wrapper for the MessagePublishr.Publish.
-func (p *Publisher) Publish(topic string, message ...*message.Message) error {
-	return p.MessagePublisher().Publish(topic, message...)
+func (p *Publisher) Publish(ctx context.Context, topic string, messages ...*message.Message) error {
+	for _, m := range messages {
+		m.SetContext(ctx)
+	}
+
+	return p.MessagePublisher().Publish(topic, messages...)
 }
 
 // MessagePublisher returns the message publisher.