meilisearch/Classes/System/Records/AbstractRepository.php
2021-04-17 21:20:54 +02:00

131 lines
4.2 KiB
PHP

<?php
namespace WapplerSystems\Meilisearch\System\Records;
/***************************************************************
* Copyright notice
*
* (c) 2017 Timo Hund <timo.hund@dkd.de
* 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 TYPO3\CMS\Core\Database\Connection;
use TYPO3\CMS\Core\Database\ConnectionPool;
use TYPO3\CMS\Core\Database\Query\QueryBuilder;
use TYPO3\CMS\Core\Utility\GeneralUtility;
/**
* Repository class to encapsulate the database access for records used in meilisearch.
*
* @author Timo Hund <timo.hund@dkd.de>
*/
abstract class AbstractRepository
{
/**
* @var string
*/
protected $table = '';
/**
* Retrieves a single row from the database by a given uid
*
* @param string $fields
* @param string $uid
* @return mixed
*/
protected function getOneRowByUid($fields, $uid)
{
$queryBuilder = $this->getQueryBuilder();
return $queryBuilder
->select($fields)
->from($this->table)
->where($queryBuilder->expr()->eq('uid', intval($uid)))
->execute()->fetch();
}
/**
* Returns QueryBuilder for Doctrine DBAL
*
* @return QueryBuilder
*/
protected function getQueryBuilder()
{
/** @var QueryBuilder $queryBuilder */
$queryBuilder = GeneralUtility::makeInstance(ConnectionPool::class)->getQueryBuilderForTable($this->table);
return $queryBuilder;
}
/**
* Returns current count of last searches
*
* @return int
*/
public function count() : int
{
return (int)$this->getQueryBuilder()
->count('*')
->from($this->table)
->execute()->fetchColumn(0);
}
/**
* Returns connection for all in transaction involved tables.
*
* Note: Rollback will not work in case of different connections.
*
* @param string[] ...$tableNames
* @return Connection
*/
public function getConnectionForAllInTransactionInvolvedTables(string ...$tableNames) : Connection
{
if (empty($tableNames) || count($tableNames) < 2) {
throw new \InvalidArgumentException(__METHOD__ . ' requires at least 2 table names.', 1504801512);
}
if (!$this->isConnectionForAllTablesTheSame(...$tableNames)) {
throw new \RuntimeException(
vsprintf('The tables "%s" using different database connections. Transaction needs same database connection ' .
'for all tables, please reconfigure the database settings for involved tables properly.', [implode('", "', $tableNames)]
), 1504866142
);
}
return GeneralUtility::makeInstance(ConnectionPool::class)->getConnectionForTable(array_shift($tableNames));
}
/**
* Checks whether all table involved in transaction using same connection.
*
* @param string[] ...$tableNames
* @return bool
*/
protected function isConnectionForAllTablesTheSame(string ...$tableNames) : bool
{
/** @var ConnectionPool $connectionPool */
$connectionPool = GeneralUtility::makeInstance(ConnectionPool::class);
$connection = $connectionPool->getConnectionForTable(array_shift($tableNames));
foreach ($tableNames as $tableName) {
$connectionForTable = $connectionPool->getConnectionForTable($tableName);
if ($connection !== $connectionForTable) {
return false;
}
}
return true;
}
}