[simple-client] Add ability to pass container builder.

makasim · makasim · commit 5c9af8de83c0 · 2018-02-15T11:07:38.000+02:00
diff --git a/pkg/simple-client/SimpleClient.php b/pkg/simple-client/SimpleClient.php
@@ -26,14 +26,20 @@
 use Interop\Queue\PsrContext;
 use Interop\Queue\PsrProcessor;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
+use Symfony\Component\DependencyInjection\ContainerInterface;
 
 final class SimpleClient
 {
     /**
-     * @var ContainerBuilder
+     * @var ContainerInterface
      */
     private $container;
 
+    /**
+     * @var array|string
+     */
+    private $config;
+
     /**
      * The config could be a transport DSN (string) or an array, here's an example of a few DSNs:.
      *
@@ -73,11 +79,13 @@ final class SimpleClient
      * ]
      *
      *
-     * @param string|array $config
+     * @param string|array          $config
+     * @param ContainerBuilder|null $container
      */
-    public function __construct($config)
+    public function __construct($config, ContainerBuilder $container = null)
     {
-        $this->container = $this->buildContainer($config);
+        $this->container = $this->buildContainer($config, $container ?: new ContainerBuilder());
+        $this->config = $config;
     }
 
     /**
@@ -252,16 +260,16 @@ public function getRouterProcessor()
     }
 
     /**
-     * @param array|string $config
+     * @param array|string     $config
+     * @param ContainerBuilder $container
      *
-     * @return ContainerBuilder
+     * @return ContainerInterface
      */
-    private function buildContainer($config)
+    private function buildContainer($config, ContainerBuilder $container)
     {
         $config = $this->buildConfig($config);
         $extension = $this->buildContainerExtension();
 
-        $container = new ContainerBuilder();
         $container->registerExtension($extension);
         $container->loadFromExtension($extension->getAlias(), $config);
 
