consumer.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	amqp "github.com/rabbitmq/amqp091-go"
 	"golang.org/x/time/rate"
+	"log"
 	"time"
 )
 
@@ -28,84 +29,78 @@ func runConsumer(ctx context.Context, client *Client, msgCh chan []byte) {
 
 	limiter := rate.NewLimiter(rate.Every(client.opts.consumerRateLimit), client.opts.consumerBurstSize)
 
-	reconnectSignal := make(chan struct{}, 1)
-	reconnectTrigger := func() {
-		select {
-		case reconnectSignal <- struct{}{}:
-		default:
-		}
-	}
-
-	// initial consume
 	deliveries, err := client.consume()
 	if err != nil {
-		client.logger.Printf("Could not start consuming: %s\n", err)
+		log.Printf("Could not start consuming: %s\n", err)
 		return
 	}
 
 	chClosedCh := make(chan *amqp.Error, 1)
 	client.Channel.NotifyClose(chClosedCh)
 
+	reconnectTimer := time.NewTimer(0)
+	defer reconnectTimer.Stop()
+	<-reconnectTimer.C
+
 	for {
+		if !reconnectTimer.Stop() {
+			select {
+			case <-reconnectTimer.C:
+			default:
+			}
+		}
+
 		select {
 
 		case <-runCtx.Done():
 			err = client.Close()
 			if err != nil {
-				client.logger.Printf("Close failed: %s\n", err)
+				log.Printf("Close failed: %s\n", err)
 			}
 			return
 
-		case <-reconnectSignal:
-			select {
-			case <-runCtx.Done():
-				return
-			case <-time.After(client.opts.reconnectDelay):
-			}
+		case amqErr := <-chClosedCh:
+			log.Printf("AMQP Channel closed due to: %s Reconnecting...\n", amqErr)
+			reconnectTimer.Reset(time.Second)
 
+		case <-reconnectTimer.C:
 			deliveries, err = client.consume()
 			if err != nil {
-				client.logger.Printf("Consume reconnect failed, retry: %s\n", err)
-				reconnectTrigger()
+				log.Println("Error trying to consume, will try again. Retry in 5 seconds.")
+				reconnectTimer.Reset(time.Second * 5)
 				continue
 			}
 
-			// re-create closing chan
 			chClosedCh = make(chan *amqp.Error, 1)
 			client.Channel.NotifyClose(chClosedCh)
-			client.logger.Println("Consume reconnect success")
-
-		case amqErr := <-chClosedCh:
-			client.logger.Printf("AMQP Channel closed due to: %s Reconnecting...\n", amqErr)
-			reconnectTrigger()
 
 		case delivery, ok := <-deliveries:
 			if !ok {
-				client.logger.Println("Deliveries channel closed unexpectedly")
-				reconnectTrigger()
+				log.Println("Deliveries channel closed unexpectedly")
+				reconnectTimer.Reset(time.Second)
 				continue
 			}
 
 			if err = limiter.Wait(runCtx); err != nil {
-				client.logger.Printf("Wait limiter failed: %s\n", err)
+				log.Printf("Wait limiter failed: %s\n", err)
 			}
 
 			select {
 			case <-runCtx.Done():
 				if err = delivery.Nack(false, true); err != nil {
-					client.logger.Printf("Error nacking message: %s\n", err)
+					log.Printf("Error nacking message: %s\n", err)
 				}
 
 				err = client.Close()
 				if err != nil {
-					client.logger.Printf("Close failed: %s\n", err)
+					log.Printf("Close failed: %s\n", err)
 				}
 				return
 
 			case msgCh <- delivery.Body:
-				client.logger.Printf("Received message: %s\n", delivery.Body)
+				log.Printf("Received message: %s\n", delivery.Body)
 				if err = delivery.Ack(false); err != nil {
-					client.logger.Printf("Error acknowledging message: %s\n", err)
+					log.Printf("Error acknowledging message: %s\n", err)
 				}
 			}
 		}

handler.go

@@ -6,14 +6,13 @@ import (
 	amqp "github.com/rabbitmq/amqp091-go"
 	"io"
 	"log"
-	"os"
 	"sync"
 	"time"
 )
 
 type Client struct {
 	mutex           *sync.Mutex
-	queueOptions    *QueueOpts
+	queueName       string
 	logger          *log.Logger
 	connection      *amqp.Connection
 	Channel         *amqp.Channel
@@ -36,33 +35,21 @@ type options struct {
 	logger            *log.Logger
 }
 
-type QueueOpts struct {
-	QueueName  string
-	Durable    bool
-	AutoDelete bool
-	Exclusive  bool
-	NoWait     bool
-	Args       amqp.Table
-}
-
-func NewClient(address Address, queueOpts QueueOpts, opts ...Option) (*Client, error) {
-	l := log.New(os.Stdout, "", log.LstdFlags)
-
+func NewClient(address Address, queueName string, opts ...Option) (*Client, error) {
 	addr, err := address.makeAddr()
 	if err != nil {
 		return nil, errors.Join(errBadAddr, err)
 	}
 
-	if queueOpts.QueueName == "" {
-		l.Fatal(errNoQueue)
+	if queueName == "" {
+		log.Fatal(errNoQueue)
 	}
 
 	client := Client{
-		mutex:        &sync.Mutex{},
-		queueOptions: &queueOpts,
-		done:         make(chan bool),
-		connected:    make(chan struct{}),
-		logger:       l,
+		mutex:     &sync.Mutex{},
+		queueName: queueName,
+		done:      make(chan bool),
+		connected: make(chan struct{}),
 	}
 
 	o := options{

publisher.go

@@ -3,6 +3,7 @@ package rabbit
 import (
 	"context"
 	"errors"
+	"log"
 	"time"
 )
 
@@ -22,15 +23,15 @@ func (p *pubHandler) Start(ctx context.Context, chanLen uint) chan<- []byte {
 			case <-ctx.Done():
 				for msg := range ch {
 					if err := p.push(msg); err != nil {
-						p.client.logger.Printf("Error publishing message (shutdown): %s", err)
+						log.Printf("Error publishing message (shutdown): %s", err)
 					}
 				}
-				p.client.logger.Println("Publisher stopped")
+				log.Println("Publisher stopped")
 				return
 
 			case msg := <-ch:
 				if err := p.push(msg); err != nil {
-					p.client.logger.Printf("Error publishing message: %s", err)
+					log.Printf("Error publishing message: %s", err)
 				}
 			}
 		}

service.go

@@ -78,16 +78,7 @@ func (c *Client) init(conn *amqp.Connection) error {
 		return err
 	}
 
-	//_, err = ch.QueueDeclare(c.queueName, false, false, false, false, nil)
-	_, err = ch.QueueDeclare(
-		c.queueOptions.QueueName,
-		c.queueOptions.Durable,
-		c.queueOptions.AutoDelete,
-		c.queueOptions.Exclusive,
-		c.queueOptions.NoWait,
-		c.queueOptions.Args,
-	)
-
+	_, err = ch.QueueDeclare(c.queueName, false, false, false, false, nil)
 	if err != nil {
 		return err
 	}
@@ -129,7 +120,7 @@ func (c *Client) unsafePush(data []byte) error {
 	return c.Channel.PublishWithContext(
 		ctx,
 		"",
-		c.queueOptions.QueueName,
+		c.queueName,
 		false,
 		false,
 		amqp.Publishing{
@@ -156,7 +147,7 @@ func (c *Client) consume() (<-chan amqp.Delivery, error) {
 	}
 
 	return c.Channel.Consume(
-		c.queueOptions.QueueName,
+		c.queueName,
 		"",
 		false,
 		false,