review fixes

Author: vyuldashev

src/VladimirYuldashev/LaravelQueueRabbitMQ/LaravelQueueRabbitMQServiceProvider.php

@@ -2,6 +2,7 @@
 
 namespace VladimirYuldashev\LaravelQueueRabbitMQ;
 
+use Illuminate\Queue\QueueManager;
 use Illuminate\Support\ServiceProvider;
 use VladimirYuldashev\LaravelQueueRabbitMQ\Queue\Connectors\RabbitMQConnector;
 
@@ -26,15 +27,16 @@ public function register()
      */
     public function boot()
     {
-        $appQueue = app('queue');
-        $rabbitMQConnector = new RabbitMQConnector();
+        /** @var QueueManager $queue */
+        $queue = $this->app['queue'];
+        $connector = new RabbitMQConnector;
 
-        $appQueue->stopping(function () use ($rabbitMQConnector) {
-            $rabbitMQConnector->getConnection()->close();
+        $queue->stopping(function () use ($connector) {
+            $connector->connection()->close();
         });
 
-        $appQueue->addConnector('rabbitmq', function () use ($rabbitMQConnector) {
-            return $rabbitMQConnector;
+        $queue->addConnector('rabbitmq', function () use ($connector) {
+            return $connector;
         });
     }
 }

src/VladimirYuldashev/LaravelQueueRabbitMQ/Queue/Connectors/RabbitMQConnector.php

@@ -8,6 +8,10 @@
 
 class RabbitMQConnector implements ConnectorInterface
 {
+
+    /** @var AMQPStreamConnection */
+    private $connection;
+
     /**
      * Establish a queue connection.
      *
@@ -18,7 +22,7 @@ class RabbitMQConnector implements ConnectorInterface
     public function connect(array $config)
     {
         // create connection with AMQP
-        $connection = new AMQPStreamConnection(
+        $this->connection = new AMQPStreamConnection(
             $config['host'],
             $config['port'],
             $config['login'],
@@ -27,8 +31,14 @@ public function connect(array $config)
         );
 
         return new RabbitMQQueue(
-            $connection,
+            $this->connection,
             $config
         );
     }
+
+    public function connection()
+    {
+        return $this->connection;
+    }
+
 }

src/VladimirYuldashev/LaravelQueueRabbitMQ/Queue/RabbitMQQueue.php

@@ -330,14 +330,4 @@ private function reportConnectionError($action, Exception $e)
         sleep($this->sleepOnError);
     }
 
-    /**
-     * Return rabbit connection layer
-     *
-     * @return AMQPStreamConnection
-     */
-    public function getConnection()
-    {
-        return $this->connection;
-    }
-
 }