Add Close method to transactional publisher

czeslavo · Mar 20, 2020 · 87b043a · 87b043a
1 parent 6b4ba83
commit 87b043a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pkg/firestore/transactional.go b/pkg/firestore/transactional.go
@@ -23,3 +23,7 @@ func NewTransactionalPublisher(config PublisherConfig, tx *firestore.Transaction
 func (p *TransactionalPublisher) Publish(topic string, messages ...*message.Message) error {
 	return p.publisher.PublishInTransaction(topic, p.tx, messages...)
 }
+
+func (p *TransactionalPublisher) Close() error {
+	return p.publisher.Close()
+}