312 lines
9.1 KiB
PHP
312 lines
9.1 KiB
PHP
|
<?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;
|
||
|
}
|
||
|
}
|