add method to get queue after build

contamobi · Jan 11, 2017 · 9def546 · 9def546
1 parent bb02de4
commit 9def546
Show file tree

Hide file tree

Showing 4 changed files with 35 additions and 0 deletions.
diff --git a/Queue/QueueBuilderInterface.php b/Queue/QueueBuilderInterface.php
@@ -12,4 +12,9 @@ interface QueueBuilderInterface
      * @return Queue
      */
     public function buildQueue($queueName, QueueServiceInterface $queueService, QueueBagInterface $queueBag);
+
+    /**
+     * @return Queue
+     */
+    public function getQueue();
 }
diff --git a/Transport/PubSub/SubscriberBuilder.php b/Transport/PubSub/SubscriberBuilder.php
@@ -11,6 +11,7 @@
 
 class SubscriberBuilder implements QueueBuilderInterface
 {
+    private $queue;
     private $connectionManager;
     private $logger;
 
@@ -36,6 +37,7 @@ public function buildQueue($queueName, QueueServiceInterface $queueService, Queu
         $queue = new Queue($this->getConnectionManager(), $queueBag, $this->logger);
         $queueCallback = new SubscriberQueueCallback($queueService);
         $queue->setCallback($queueCallback);
+        $this->queue = $queue;
 
         return $queue;
     }
@@ -47,4 +49,12 @@ public function getConnectionManager()
     {
         return $this->connectionManager;
     }
+
+    /**
+     * @return Queue
+     */
+    public function getQueue()
+    {
+        return $this->queue;
+    }
 }
diff --git a/Transport/Rpc/RpcServerBuilder.php b/Transport/Rpc/RpcServerBuilder.php
@@ -11,6 +11,7 @@
 
 class RpcServerBuilder implements QueueBuilderInterface
 {
+    private $queue;
     private $connectionManager;
     private $logger;
 
@@ -36,6 +37,7 @@ public function buildQueue($queueName, QueueServiceInterface $queueService, Queu
         $queue = new Queue($this->getConnectionManager(), $queueBag, $this->logger);
         $queueCallback = new RpcQueueCallback($queueService);
         $queue->setCallback($queueCallback);
+        $this->queue = $queue;
 
         return $queue;
     }
@@ -47,4 +49,12 @@ public function getConnectionManager()
     {
         return $this->connectionManager;
     }
+
+    /**
+     * @return Queue
+     */
+    public function getQueue()
+    {
+        return $this->queue;
+    }
 }
diff --git a/Transport/Worker/WorkerBuilder.php b/Transport/Worker/WorkerBuilder.php
@@ -12,6 +12,7 @@
 
 class WorkerBuilder implements QueueBuilderInterface
 {
+    private $queue;
     private $connectionManager;
     private $logger;
 
@@ -37,6 +38,7 @@ public function buildQueue($queueName, QueueServiceInterface $queueService, Queu
         $queue = new Queue($this->getConnectionManager(), $queueBag, $this->logger);
         $queueCallback = new WorkerQueueCallback($queueService);
         $queue->setCallback($queueCallback);
+        $this->queue = $queue;
 
         return $queue;
     }
@@ -48,4 +50,12 @@ public function getConnectionManager()
     {
         return $this->connectionManager;
     }
+
+    /**
+     * @return Queue
+     */
+    public function getQueue()
+    {
+        return $this->queue;
+    }
 }