meilisearch/Classes/System/Solr/SolrConnection.php

312 lines
9.1 KiB
PHP
Raw Normal View History

2021-04-17 00:26:33 +02:00
<?php
namespace WapplerSystems\Meilisearch\System\Solr;
/***************************************************************
* Copyright notice
*
* (c) 2009-2015 Ingo Renner <ingo@typo3.org>
* All rights reserved
*
* This script is part of the TYPO3 project. The TYPO3 project is
* free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 3 of the License, or
* (at your option) any later version.
*
* The GNU General Public License can be found at
* http://www.gnu.org/copyleft/gpl.html.
*
* This script is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* This copyright notice MUST APPEAR in all copies of the script!
***************************************************************/
use WapplerSystems\Meilisearch\System\Configuration\TypoScriptConfiguration;
use WapplerSystems\Meilisearch\System\Logging\SolrLogManager;
use WapplerSystems\Meilisearch\System\Solr\Parser\SchemaParser;
use WapplerSystems\Meilisearch\System\Solr\Parser\StopWordParser;
use WapplerSystems\Meilisearch\System\Solr\Parser\SynonymParser;
use WapplerSystems\Meilisearch\System\Solr\Service\SolrAdminService;
use WapplerSystems\Meilisearch\System\Solr\Service\SolrReadService;
use WapplerSystems\Meilisearch\System\Solr\Service\SolrWriteService;
use WapplerSystems\Meilisearch\Util;
use Psr\Container\ContainerExceptionInterface;
use Psr\Container\NotFoundExceptionInterface;
use Psr\EventDispatcher\EventDispatcherInterface;
use Psr\Http\Client\ClientInterface;
use Psr\Http\Message\RequestFactoryInterface;
use Psr\Http\Message\StreamFactoryInterface;
use Solarium\Client;
use Solarium\Core\Client\Adapter\Psr18Adapter;
use TYPO3\CMS\Core\Utility\GeneralUtility;
/**
* Solr Service Access
*
* @author Ingo Renner <ingo@typo3.org>
*/
class SolrConnection
{
/**
* @var SolrAdminService
*/
protected $adminService;
/**
* @var SolrReadService
*/
protected $readService;
/**
* @var SolrWriteService
*/
protected $writeService;
/**
* @var TypoScriptConfiguration
*/
protected $configuration;
/**
* @var SynonymParser
*/
protected $synonymParser = null;
/**
* @var StopWordParser
*/
protected $stopWordParser = null;
/**
* @var SchemaParser
*/
protected $schemaParser = null;
/**
* @var Node[]
*/
protected $nodes = [];
/**
* @var SolrLogManager
*/
protected $logger = null;
/**
* @var ClientInterface[]
*/
protected $clients = [];
/**
* @var ClientInterface
*/
protected $psr7Client;
/**
* @var RequestFactoryInterface
*/
protected $requestFactory;
/**
* @var StreamFactoryInterface
*/
protected $streamFactory;
/**
* @var EventDispatcherInterface
*/
protected $eventDispatcher;
/**
* Constructor
*
* @param Node $readNode
* @param Node $writeNode
* @param ?TypoScriptConfiguration $configuration
* @param ?SynonymParser $synonymParser
* @param ?StopWordParser $stopWordParser
* @param ?SchemaParser $schemaParser
* @param ?SolrLogManager $logManager
* @param ?ClientInterface $psr7Client
* @param ?RequestFactoryInterface $requestFactory
* @param ?StreamFactoryInterface $streamFactory
* @param ?EventDispatcherInterface $eventDispatcher
*
* @throws ContainerExceptionInterface
* @throws NotFoundExceptionInterface
*/
public function __construct(
Node $readNode,
Node $writeNode,
TypoScriptConfiguration $configuration = null,
SynonymParser $synonymParser = null,
StopWordParser $stopWordParser = null,
SchemaParser $schemaParser = null,
SolrLogManager $logManager = null,
ClientInterface $psr7Client = null,
RequestFactoryInterface $requestFactory = null,
StreamFactoryInterface $streamFactory = null,
EventDispatcherInterface $eventDispatcher = null
) {
$this->nodes['read'] = $readNode;
$this->nodes['write'] = $writeNode;
$this->nodes['admin'] = $writeNode;
$this->configuration = $configuration ?? Util::getSolrConfiguration();
$this->synonymParser = $synonymParser;
$this->stopWordParser = $stopWordParser;
$this->schemaParser = $schemaParser;
$this->logger = $logManager;
$this->psr7Client = $psr7Client ?? GeneralUtility::getContainer()->get(ClientInterface::class);
$this->requestFactory = $requestFactory ?? GeneralUtility::getContainer()->get(RequestFactoryInterface::class);
$this->streamFactory = $streamFactory ?? GeneralUtility::getContainer()->get(StreamFactoryInterface::class);
$this->eventDispatcher = $eventDispatcher ?? GeneralUtility::getContainer()->get(EventDispatcherInterface::class);
}
/**
* @param string $key
* @return Node
*/
public function getNode(string $key): Node
{
return $this->nodes[$key];
}
/**
* @return SolrAdminService
*/
public function getAdminService(): SolrAdminService
{
if ($this->adminService === null) {
$this->adminService = $this->buildAdminService();
}
return $this->adminService;
}
/**
* @return SolrAdminService
* @noinspection PhpIncompatibleReturnTypeInspection
*/
protected function buildAdminService(): SolrAdminService
{
$endpointKey = 'admin';
$client = $this->getClient($endpointKey);
$this->initializeClient($client, $endpointKey);
return GeneralUtility::makeInstance(SolrAdminService::class, $client, $this->configuration, $this->logger, $this->synonymParser, $this->stopWordParser, $this->schemaParser);
}
/**
* @return SolrReadService
*/
public function getReadService(): SolrReadService
{
if ($this->readService === null) {
$this->readService = $this->buildReadService();
}
return $this->readService;
}
/**
* @return SolrReadService
* @noinspection PhpIncompatibleReturnTypeInspection
*/
protected function buildReadService(): SolrReadService
{
$endpointKey = 'read';
$client = $this->getClient($endpointKey);
$this->initializeClient($client, $endpointKey);
return GeneralUtility::makeInstance(SolrReadService::class, $client);
}
/**
* @return SolrWriteService
*/
public function getWriteService(): SolrWriteService
{
if ($this->writeService === null) {
$this->writeService = $this->buildWriteService();
}
return $this->writeService;
}
/**
* @return SolrWriteService
* @noinspection PhpIncompatibleReturnTypeInspection
*/
protected function buildWriteService(): SolrWriteService
{
$endpointKey = 'write';
$client = $this->getClient($endpointKey);
$this->initializeClient($client, $endpointKey);
return GeneralUtility::makeInstance(SolrWriteService::class, $client);
}
/**
* @param Client $client
* @param string $endpointKey
* @return Client
*/
protected function initializeClient(Client $client, string $endpointKey): Client
{
if (trim($this->getNode($endpointKey)->getUsername()) === '') {
return $client;
}
$username = $this->getNode($endpointKey)->getUsername();
$password = $this->getNode($endpointKey)->getPassword();
$this->setAuthenticationOnAllEndpoints($client, $username, $password);
return $client;
}
/**
* @param Client $client
* @param string $username
* @param string $password
*/
protected function setAuthenticationOnAllEndpoints(Client $client, string $username, string $password)
{
foreach ($client->getEndpoints() as $endpoint) {
$endpoint->setAuthentication($username, $password);
}
}
/**
* @param string $endpointKey
* @return Client
*/
protected function getClient(string $endpointKey): Client
{
if ($this->clients[$endpointKey]) {
return $this->clients[$endpointKey];
}
$adapter = new Psr18Adapter($this->psr7Client, $this->requestFactory, $this->streamFactory);
$client = new Client($adapter, $this->eventDispatcher);
$client->getPlugin('postbigrequest');
$client->clearEndpoints();
$newEndpointOptions = $this->getNode($endpointKey)->getSolariumClientOptions();
$newEndpointOptions['key'] = $endpointKey;
$client->createEndpoint($newEndpointOptions, true);
$this->clients[$endpointKey] = $client;
return $client;
}
/**
* @param Client $client
* @param ?string $endpointKey
*/
public function setClient(Client $client, ?string $endpointKey = 'read')
{
$this->clients[$endpointKey] = $client;
}
}