Manticoresearch\Connection\ConnectionPool

Complexity

Total Complexity

13

Size/Duplication

Total Lines	87
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	getConnections()	0	3	1
A	getStrategy()	0	3	1
B	getConnection()	0	25	7
A	setStrategy()	0	3	1
A	hasConnections()	0	3	1
A	__construct()	0	5	1
A	setConnections()	0	3	1

<?php

namespace Manticoresearch\Connection;

use Manticoresearch\Connection;
use Manticoresearch\Connection\Strategy\SelectorInterface;
use Manticoresearch\Exceptions\ConnectionException;
use Manticoresearch\Exceptions\NoMoreNodesException;

/**
 * Class ConnectionPool
 * @package Manticoresearch\Connection
 */
class ConnectionPool
{
    /**
     * @var array
     */
    protected $connections;

    /**
     * @var SelectorInterface
     */
    public $strategy;

    public $retries;

    public $retries_attempts = 0;

    public $retries_info = [];

    public function __construct(array $connections, SelectorInterface $strategy, int $retries)
    {
        $this->connections = $connections;
        $this->strategy = $strategy;
        $this->retries = $retries;
    }

    /**
     * @return array
     */
    public function getConnections(): array
    {
        return $this->connections;
    }

    /**
     * @param array $connections
     */
    public function setConnections(array $connections)
    {
        $this->connections = $connections;
    }
    public function getConnection(): Connection
    {
        $this->retries_attempts++;
        $connection = $this->strategy->getConnection($this->connections);
        if ($this->retries_attempts <= $this->retries) {
            $this->retries_info[] = [
                'host' => $connection->getHost(),
                'port' => $connection->getPort(),
            ];
        }
        if ($connection->isAlive()) {
            return $connection;
        }
        if ($this->retries_attempts < $this->retries) {
            return $connection;
        }
        $exMsg = 'After %d retr%s to %d node%s, connection has failed. No more retries left.';
        $exMsg .= "\nRetries made:\n";
        foreach ($this->retries_info as $i => $info) {
            $i++;
            $exMsg .= " $i. to {$info['host']}:{$info['port']}\n";
        }
        $connCount = count($this->connections);
        throw new NoMoreNodesException(
            sprintf($exMsg, $this->retries, $this->retries > 1 ? 'ies' : 'y', $connCount, $connCount > 1 ? 's' : '')
        );
    }

    public function hasConnections(): bool
    {
        return $this->retries_attempts < $this->retries;
    }

    /**
     * @return SelectorInterface
     */
    public function getStrategy(): SelectorInterface
    {
        return $this->strategy;
    }

    /**
     * @param SelectorInterface $strategy
     */
    public function setStrategy(SelectorInterface $strategy)
    {
        $this->strategy = $strategy;
    }
}