facile-it / mongodb-bundle

src/Services/ClientRegistry.php

<?php

declare(strict_types=1);

namespace Facile\MongoDbBundle\Services;

use Facile\MongoDbBundle\Capsule\Client as BundleClient;
use Facile\MongoDbBundle\Event\ConnectionEvent;
use Facile\MongoDbBundle\Models\ClientConfiguration;
use Facile\MongoDbBundle\Services\DriverOptions\DriverOptionsInterface;
use MongoDB\Client;
use Symfony\Component\EventDispatcher\EventDispatcherInterface;
use Symfony\Component\EventDispatcher\LegacyEventDispatcherProxy;

/**
 * Class ClientRegistry.
 *
 * @internal
 */
final class ClientRegistry
{
    /** @var Client[] */
    private $clients;

    /** @var ClientConfiguration[] */
    private $configurations;

    /** @var bool */
    private $debug;

    /** @var EventDispatcherInterface */
    private $eventDispatcher;

    /** @var DriverOptionsInterface */
    private $driverOptionsService;

    public function __construct(
        EventDispatcherInterface $eventDispatcher,
        bool $debug,
        ?DriverOptionsInterface $driverOptionsService
    ) {
        $this->clients = [];
        $this->configurations = [];
        $this->debug = $debug;
        $this->eventDispatcher = $eventDispatcher;
        $this->driverOptionsService = $driverOptionsService;
    }

    public function addClientsConfigurations(array $configurations): void
    {
        foreach ($configurations as $name => $conf) {
            $this->configurations[$name] = $this->buildClientConfiguration($conf);
        }
    }

    /**
     * @param array $conf
     *
     * @return ClientConfiguration
     */
    private function buildClientConfiguration(array $conf): ClientConfiguration
    {
        if (! $conf['uri']) {
            $conf['uri'] = self::buildConnectionUri($conf['hosts']);
        }

        $conf['driverOptions'] = [];
        if ($this->driverOptionsService instanceof DriverOptionsInterface) {

$this->driverOptionsService is always a sub-type of Facile\MongoDbBundle\Services\DriverOptions\DriverOptionsInterface.

            $conf['driverOptions'] = $this->driverOptionsService->buildDriverOptions($conf);
        }

        return new ClientConfiguration(
            $conf['uri'],
            $conf['username'],
            $conf['password'],
            $conf['authSource'],
            [
                'replicaSet' => $conf['replicaSet'],
                'ssl' => $conf['ssl'],
                'connectTimeoutMS' => $conf['connectTimeoutMS'],
                'readPreference' => $conf['readPreference'],
            ],
            $conf['driverOptions']
        );
    }

    private static function buildConnectionUri(array $hosts): string
    {
        return 'mongodb://' . implode(
            ',',
            array_map(
                static function (array $host): string {
                    return sprintf('%s:%d', $host['host'], $host['port']);
                },
                $hosts
            )
        );
    }

    public function getClientForDatabase(string $name, string $databaseName): Client
    {
        return $this->getClient($name, $databaseName);
    }

    public function getClientNames(): array
    {
        return array_keys($this->clients);
    }

    public function getClient(string $name, ?string $databaseName = null): Client
    {
        $clientKey = null !== $databaseName ? $name . '.' . $databaseName : $name;

        if (! isset($this->clients[$clientKey])) {
            $conf = $this->configurations[$name];
            $options = array_merge(
                ['database' => $databaseName],
                $conf->getAuthSource() !== null ? ['authSource' => $conf->getAuthSource()] : [],
                $conf->getOptions()
            );
            $this->clients[$clientKey] = $this->buildClient($name, $conf->getUri(), $options, $conf->getDriverOptions());

            $event = new ConnectionEvent($clientKey);
            if (class_exists(LegacyEventDispatcherProxy::class)) {
                $this->eventDispatcher->dispatch($event, ConnectionEvent::CLIENT_CREATED);

The call to Symfony\Contracts\EventDispatcher\EventDispatcherInterface::dispatch() has too many arguments starting with Facile\MongoDbBundle\Event\ConnectionEvent::CLIENT_CREATED.

            } else {
                $this->eventDispatcher->dispatch(ConnectionEvent::CLIENT_CREATED, $event);

Facile\MongoDbBundle\Event\ConnectionEvent::CLIENT_CREATED of type string is incompatible with the type object expected by parameter $event of Symfony\Contracts\EventDispatcher\EventDispatcherInterface::dispatch().

            }
        }

        return $this->clients[$clientKey];
    }

    /**
     * @param string $clientName
     * @param string $uri
     * @param array  $options
     * @param array  $driverOptions
     *
     * @return Client
     */
    private function buildClient(string $clientName, string $uri, array $options, array $driverOptions): Client
    {
        if (true === $this->debug) {
            return new BundleClient($uri, $options, $driverOptions, $clientName, $this->eventDispatcher);
        }

        return new Client($uri, $options, $driverOptions);
    }
}