make use of connection from the configuration (#267)

ecotoneframework · Nov 11, 2023 · 9a18615 · 9a18615
1 parent b72f2f4
commit 9a18615
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/packages/Dbal/src/DbalReconnectableConnectionFactory.php b/packages/Dbal/src/DbalReconnectableConnectionFactory.php
@@ -67,7 +67,7 @@ public function getConnection(): Connection
  public static function getWrappedConnection(object $connection): Connection
  {
  if ($connection instanceof EcotoneManagerRegistryConnectionFactory) {
- return $connection->getRegistry()->getConnection();
+ return $connection->getConnection();
  } else {
  $reflectionClass = new ReflectionClass($connection);
  $method = $reflectionClass->getMethod('establishConnection');

diff --git a/packages/Dbal/src/EcotoneManagerRegistryConnectionFactory.php b/packages/Dbal/src/EcotoneManagerRegistryConnectionFactory.php
@@ -4,22 +4,31 @@
 
 namespace Ecotone\Dbal;
 
+use Doctrine\DBAL\Connection;
 use Doctrine\Persistence\ManagerRegistry;
 use Enqueue\Dbal\ManagerRegistryConnectionFactory;
 
 final class EcotoneManagerRegistryConnectionFactory extends ManagerRegistryConnectionFactory
 {
  private ManagerRegistry $registry;
+ /** @var array<string, string> */
+ private array $config;
 
  public function __construct(ManagerRegistry $registry, array $config = [])
  {
  parent::__construct($registry, $config);
 
  $this->registry = $registry;
+ $this->config = $config;
  }
 
  public function getRegistry(): ManagerRegistry
  {
  return $this->registry;
  }
+
+ public function getConnection(): Connection
+ {
+ return $this->registry->getConnection($this->config['connection_name'] ?? null);
+ }
 }