Increse the heartbeat and fix middleware index

guilhermehubner · Mar 7, 2018 · 11b9a83 · 11b9a83
1 parent 9400ebb
commit 11b9a83
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 5 deletions.
diff --git a/broker/amqp.go b/broker/amqp.go
@@ -215,7 +215,11 @@ func (b *AMQPBroker) connect() {
 	log.Get().Info("CONNECTING...")
 
 	for {
-		connection, err := amqp.Dial(b.url)
+		connection, err := amqp.DialConfig(b.url, amqp.Config{
+			Heartbeat: 60 * time.Second,
+			Locale:    "en_US",
+		})
+
 		if err != nil {
 			log.Get().Error(fmt.Sprintf("broker/amqp: fail to connect: %v", err))
 			time.Sleep(b.backoff.Duration())

diff --git a/worker.go b/worker.go
@@ -87,10 +87,12 @@ func (w *worker) executeJob() {
 		index := i
 		oldWrapped := wrappedHandle
 
-		wrappedHandle = func(ctx context.Context) error {
-			return w.middlewares[index](injectJobInfo(ctx, *job, message),
-				oldWrapped)
-		}
+		wrappedHandle = func(index int) func(context.Context) error {
+			return func(ctx context.Context) error {
+				return w.middlewares[index](injectJobInfo(ctx, *job, message),
+					oldWrapped)
+			}
+		}(index)
 	}
 
 	ctx, cancelFn := context.WithCancel(context.Background())