Use io.Closer interface instead of NotificationConnection

Author: bilaljaved

extension/amqp/amqp.go

@@ -1,11 +1,13 @@
 package amqp
 
 import (
+	"io"
+
 	"github.com/streadway/amqp"
 )
 
 // setup returns a connection and channel to be used for the Queue setup
-func setup(url, queue string) (NotificationConnection, NotificationChannel, error) {
+func setup(url, queue string) (io.Closer, NotificationChannel, error) {
 	conn, err := amqp.Dial(url)
 	if err != nil {
 		return nil, nil, err
@@ -27,8 +29,3 @@ func setup(url, queue string) (NotificationConnection, NotificationChannel, erro
 type NotificationChannel interface {
 	Publish(exchange, queue string, mandatory, immediate bool, msg amqp.Publishing) error
 }
-
-// NotificationConnection represents a connection to the notification queue
-type NotificationConnection interface {
-	Close() error
-}

extension/amqp/publisher.go

@@ -2,6 +2,7 @@ package amqp
 
 import (
 	"context"
+	"io"
 
 	"github.com/hellofresh/goengine"
 	"github.com/hellofresh/goengine/driver/sql"
@@ -17,14 +18,14 @@ type NotificationPublisher struct {
 	queue   string
 	logger  goengine.Logger
 
-	connection NotificationConnection
+	connection io.Closer
 	channel    NotificationChannel
 }
 
 // NewNotificationPublisher returns an instance of NotificationPublisher
 func NewNotificationPublisher(amqpDSN, queue string,
 	logger goengine.Logger,
-	connection NotificationConnection,
+	connection io.Closer,
 	channel NotificationChannel,
 ) (*NotificationPublisher, error) {