Do not wait for listners if consumer got canceled

corvus-ch · Nov 21, 2018 · 7c9000e · 7c9000e
1 parent 2e639a6
commit 7c9000e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/consumer/consumer.go b/consumer/consumer.go
@@ -106,7 +106,9 @@ func (c *Consumer) consume(msgs <-chan amqp.Delivery, done chan error) {
 			return
 		}
 	}
-	c.wg.Wait()
+	if !c.canceled {
+		c.wg.Wait()
+	}
 	done <- nil
 }