summaryrefslogtreecommitdiff
path: root/plugins/jobs/brokers/amqp/redial.go
blob: bfb1fbff5fbc137a25a047f10e89abff4072fdd9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
package amqp

import (
	"fmt"
	"time"

	"github.com/cenkalti/backoff/v4"
	"github.com/streadway/amqp"
)

// redialer used to redial to the rabbitmq in case of the connection interrupts
func (j *JobsConsumer) redialer() {
	go func() {
		for err := range j.conn.NotifyClose(make(chan *amqp.Error)) {
			if err != nil {
				j.logger.Error("connection closed, reconnecting", "error", err)

				expb := backoff.NewExponentialBackOff()
				// set the retry timeout (minutes)
				expb.MaxElapsedTime = time.Minute * j.retryTimeout
				op := func() error {
					j.logger.Warn("rabbitmq reconnecting, caused by", "error", err)

					j.Lock()
					var dialErr error
					j.conn, dialErr = amqp.Dial(j.connStr)
					if dialErr != nil {
						j.Unlock()
						return fmt.Errorf("fail to dial server endpoint: %v", dialErr)
					}
					j.Unlock()

					j.logger.Info("rabbitmq dial succeed. trying to redeclare queues and subscribers")

					// re-init connection
					deliv, errInit := j.initRabbitMQ()
					if errInit != nil {
						j.Unlock()
						j.logger.Error("error while redialing", "error", errInit)
						return errInit
					}

					// restart listener
					j.listener(deliv)

					j.logger.Info("queues and subscribers redeclare succeed")
					return nil
				}

				retryErr := backoff.Retry(op, expb)
				if retryErr != nil {
					j.logger.Error("backoff failed", "error", retryErr)
					return
				}
			}
		}
	}()
}