src/Extension/Doctrine/Connection/ConnectionWrapper.php line 62

Open in your IDE?
  1. <?php
  3. namespace App\Extension\Doctrine\Connection;
  5. use Doctrine\DBAL\Connection;
  6. use Doctrine\DBAL\Event\ConnectionEventArgs;
  7. use Doctrine\DBAL\Events;
  8. use Symfony\Component\HttpFoundation\Session\Session;
  10. /**
  11.  * @author Dawid zulus Pakula [zulus@w3des.net]
  12.  */
  13. class ConnectionWrapper extends Connection
  14. {
  15.     const SESSION_ACTIVE_DYNAMIC_CONN = 'cliente_connection';
  17.     /**
  18.      * @var Session
  19.      */
  20.     private $session;
  22.     /**
  23.      * @var bool
  24.      */
  25.     private $_isConnected = false;
  27.     /**
  28.      * @param Session $sess
  29.      * @return void
  30.      */
  31.     public function setSession(Session $sess)
  32.     {
  33.         $this->session = $sess;
  34.     }
  36.     /**
  37.      * @param string $dbname
  38.      * @param string|null $host
  39.      * @return void
  40.      */
  41.     public function forceSwitch(string $dbname, ?string $host = NULL)
  42.     {
  43.         if ( $this->session->has(self::SESSION_ACTIVE_DYNAMIC_CONN) ) {
  44.             $session = $this->session->get(self::SESSION_ACTIVE_DYNAMIC_CONN);
  45.             if ( $session['dbname'] === $dbname ) {
  46.                 return;
  47.             }
  48.         }
  50.         $this->session->set(self::SESSION_ACTIVE_DYNAMIC_CONN, ['dbname'=>$dbname,'host'=>$host]);
  52.         if ( $this->isConnected() ) {
  53.             $this->close();
  54.         }
  55.     }
  57.     /**
  58.      * {@inheritdoc}
  59.      */
  60.     public function connect()
  61.     {
  62.         if ( !$this->session->has(self::SESSION_ACTIVE_DYNAMIC_CONN) ) {
  63.             throw new \InvalidArgumentException('You have to inject into valid context first');
  64.         }
  66.         if ( $this->isConnected() ) {
  67.             return true;
  68.         }
  70.         $clienteParams = $this->session->get(self::SESSION_ACTIVE_DYNAMIC_CONN);
  71.         $params = $this->getParams();
  72.         $params['dbname'] = $clienteParams['dbname'];
  74.         if ( NULL !== $clienteParams['host'] ) {
  75.             $params['host'] = $clienteParams['host'];
  76.         }
  78.         $this->_conn = $this->_driver->connect(
  79.              $params
  80.             ,$params['user']
  81.             ,$params['password']
  82.         );
  84.         if ( $this->_eventManager->hasListeners(Events::postConnect) ) {
  85.             $eventArgs = new ConnectionEventArgs($this);
  86.             $this->_eventManager->dispatchEvent(Events::postConnect, $eventArgs);
  87.         }
  89.         $this->_isConnected = true;
  91.         return $this->_isConnected;
  92.     }
  94.     /**
  95.      * {@inheritdoc}
  96.      */
  97.     public function isConnected()
  98.     {
  99.         return $this->_isConnected;
  100.     }
  102.     /**
  103.      * {@inheritdoc}
  104.      */
  105.     public function close()
  106.     {
  107.         if ($this->isConnected()) {
  108.             parent::close();
  109.             $this->_isConnected = false;
  110.         }
  111.     }
  112. }