Cancel consuming a queue once done

boekkooi · Jun 17, 2015 · 89e2d24 · 89e2d24
1 parent f6db3d3
commit 89e2d24
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/Command/QueueConsumeCommand.php b/src/Command/QueueConsumeCommand.php
@@ -55,6 +55,12 @@ function (\AMQPEnvelope $envelope, \AMQPQueue $queue) use ($commandBus, &$limit)
                 return $limit > 0;
             }
         );
+
+        # Cancel consumption of the queue
+        $consumeQueue->cancel();
+        foreach ($queues as $queue) {
+            $queue->cancel();
+        }
     }
 
     /**

diff --git a/tests/Command/QueueConsumeCommandTest.php b/tests/Command/QueueConsumeCommandTest.php
@@ -151,6 +151,8 @@ function (callable $callback) {
                     } while (($callback($envelope, $queue)));
                 }
             );
+        $queue
+            ->shouldReceive('cancel');
         return $queue;
     }
 }