fix: Update rabbitmq to have close function

GarnBarn · Apr 18, 2023 · 20d8162 · 20d8162
1 parent 6f14c71
commit 20d8162
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/rabbitmq/rabbitmq.go b/rabbitmq/rabbitmq.go
@@ -28,6 +28,7 @@ type rabbitMQ struct {
 type RabbitMQ interface {
 	Consume(processor Processor, consumerConfig ConsumerConfig) (*rabbitmq.Consumer, error)
 	GetPublisher() *rabbitmq.Publisher
+	CloseConnection()
 }
 
 func NewRabbitMQ(connectionString string) (RabbitMQ, error) {
@@ -55,6 +56,11 @@ func NewRabbitMQ(connectionString string) (RabbitMQ, error) {
 	}, nil
 }
 
+func (r *rabbitMQ) CloseConnection() {
+	r.publisher.Close()
+	r.conn.Close()
+}
+
 func (r *rabbitMQ) GetPublisher() *rabbitmq.Publisher {
 	return r.publisher
 }