vendor/ruflin/elastica/src/Connection/ConnectionPool.php line 16

Open in your IDE?
  1. <?php
  3. namespace Elastica\Connection;
  5. use Elastica\Client;
  6. use Elastica\Connection;
  7. use Elastica\Connection\Strategy\StrategyInterface;
  8. use Elastica\Exception\ClientException;
  9. use Exception;
  11. /**
  12.  * Description of ConnectionPool.
  13.  *
  14.  * @author chabior
  15.  */
  16. class ConnectionPool
  17. {
  18.     /**
  19.      * @var array|Connection[] Connections array
  20.      */
  21.     protected $_connections;
  23.     /**
  24.      * @var StrategyInterface Strategy for connection
  25.      */
  26.     protected $_strategy;
  28.     /**
  29.      * @var callable|null Function called on connection fail
  30.      */
  31.     protected $_callback;
  33.     public function __construct(array $connections, StrategyInterface $strategy, ?callable $callback = null)
  34.     {
  35.         $this->_connections = $connections;
  36.         $this->_strategy = $strategy;
  37.         $this->_callback = $callback;
  38.     }
  40.     /**
  41.      * @return $this
  42.      */
  43.     public function addConnection(Connection $connection): self
  44.     {
  45.         $this->_connections[] = $connection;
  47.         return $this;
  48.     }
  50.     /**
  51.      * @param Connection[] $connections
  52.      *
  53.      * @return $this
  54.      */
  55.     public function setConnections(array $connections): self
  56.     {
  57.         $this->_connections = $connections;
  59.         return $this;
  60.     }
  62.     public function hasConnection(): bool
  63.     {
  64.         foreach ($this->_connections as $connection) {
  65.             if ($connection->isEnabled()) {
  66.                 return true;
  67.             }
  68.         }
  70.         return false;
  71.     }
  73.     /**
  74.      * @return Connection[]
  75.      */
  76.     public function getConnections(): array
  77.     {
  78.         return $this->_connections;
  79.     }
  81.     /**
  82.      * @throws ClientException
  83.      */
  84.     public function getConnection(): Connection
  85.     {
  86.         return $this->_strategy->getConnection($this->getConnections());
  87.     }
  89.     public function onFail(Connection $connection, Exception $e, Client $client): void
  90.     {
  91.         $connection->setEnabled(false);
  93.         if ($this->_callback) {
  94.             ($this->_callback)($connection, $e, $client);
  95.         }
  96.     }
  98.     public function getStrategy(): StrategyInterface
  99.     {
  100.         return $this->_strategy;
  101.     }
  102. }