Chubbyphp\DoctrineDbServiceProvider\Registry\Psr\DoctrineDbalConnectionRegistry

Complexity

Total Complexity

9

Size/Duplication

Total Lines	84
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	getConnection()	0	11	2
A	__construct()	0	4	1
A	getDefaultConnectionName()	0	5	1
A	loadConnections()	0	5	2
A	getConnectionNames()	0	3	1
A	getConnections()	0	13	2

<?php

declare(strict_types=1);

namespace Chubbyphp\DoctrineDbServiceProvider\Registry\Psr;

use Doctrine\Common\Persistence\ConnectionRegistry;
use Doctrine\DBAL\Connection;
use Psr\Container\ContainerInterface;

final class DoctrineDbalConnectionRegistry implements ConnectionRegistry
{
    /**
     * @var ContainerInterface
     */
    private $container;

    /**
     * @var array<int, string>
     */
    private $connectionNames;

    /**
     * @var ContainerInterface
     */
    private $connections;

    /**
     * @var string
     */
    private $defaultConnectionName;

    public function __construct(ContainerInterface $container, array $connectionNames)
    {
        $this->container = $container;
        $this->connectionNames = $connectionNames;
    }

    public function getDefaultConnectionName(): string
    {
        $this->loadConnections();

        return $this->defaultConnectionName;
    }

    /**
     * @param string|null $name
     *
     * @throws \InvalidArgumentException
     */
    public function getConnection($name = null): Connection
    {
        $this->loadConnections();

        $name = $name ?? $this->getDefaultConnectionName();

        if (!$this->connections->has($name)) {
            throw new \InvalidArgumentException(sprintf('Missing connection with name "%s".', $name));
        }

        return $this->connections->get($name);
    }

    /**
     * @return array<string, Connection>
     */
    public function getConnections(): array
    {
        $this->loadConnections();

        $connections = [];
        /** @var string $name */
        foreach ($this->connectionNames as $name) {
            /** @var Connection $connection */
            $connection = $this->connections->get($name);
            $connections[$name] = $connection;
        }

        return $connections;
    }

    /**
     * @return array<string>
     */
    public function getConnectionNames(): array
    {
        return $this->connectionNames;
    }

    private function loadConnections(): void
    {
        if (null === $this->connections) {
            $this->connections = $this->container->get('doctrine.dbal.dbs');
            $this->defaultConnectionName = $this->container->get('doctrine.dbal.dbs.default');
        }
    }
}