mirror of
https://github.com/nextcloud/server.git
synced 2025-01-30 22:37:01 +00:00
66 lines
2 KiB
PHP
66 lines
2 KiB
PHP
<?php
|
|
|
|
/**
|
|
* SPDX-FileCopyrightText: 2016-2024 Nextcloud GmbH and Nextcloud contributors
|
|
* SPDX-FileCopyrightText: 2016 ownCloud, Inc.
|
|
* SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
namespace OC\DB;
|
|
|
|
use Doctrine\DBAL\Schema\AbstractAsset;
|
|
use OCP\IConfig;
|
|
use function preg_match;
|
|
use function preg_quote;
|
|
|
|
/**
|
|
* Various PostgreSQL specific helper functions.
|
|
*/
|
|
class PgSqlTools {
|
|
/** @var \OCP\IConfig */
|
|
private $config;
|
|
|
|
/**
|
|
* @param \OCP\IConfig $config
|
|
*/
|
|
public function __construct(IConfig $config) {
|
|
$this->config = $config;
|
|
}
|
|
|
|
/**
|
|
* @brief Resynchronizes all sequences of a database after using INSERTs
|
|
* without leaving out the auto-incremented column.
|
|
* @param \OC\DB\Connection $conn
|
|
* @return null
|
|
*/
|
|
public function resynchronizeDatabaseSequences(Connection $conn) {
|
|
$databaseName = $conn->getDatabase();
|
|
$conn->getConfiguration()->setSchemaAssetsFilter(function ($asset) {
|
|
/** @var string|AbstractAsset $asset */
|
|
$filterExpression = '/^' . preg_quote($this->config->getSystemValueString('dbtableprefix', 'oc_')) . '/';
|
|
if ($asset instanceof AbstractAsset) {
|
|
return preg_match($filterExpression, $asset->getName()) !== false;
|
|
}
|
|
return preg_match($filterExpression, $asset) !== false;
|
|
});
|
|
|
|
foreach ($conn->createSchemaManager()->listSequences() as $sequence) {
|
|
$sequenceName = $sequence->getName();
|
|
$sqlInfo = 'SELECT table_schema, table_name, column_name
|
|
FROM information_schema.columns
|
|
WHERE column_default = ? AND table_catalog = ?';
|
|
$result = $conn->executeQuery($sqlInfo, [
|
|
"nextval('$sequenceName'::regclass)",
|
|
$databaseName
|
|
]);
|
|
$sequenceInfo = $result->fetchAssociative();
|
|
$result->free();
|
|
/** @var string $tableName */
|
|
$tableName = $sequenceInfo['table_name'];
|
|
/** @var string $columnName */
|
|
$columnName = $sequenceInfo['column_name'];
|
|
$sqlMaxId = "SELECT MAX($columnName) FROM $tableName";
|
|
$sqlSetval = "SELECT setval('$sequenceName', ($sqlMaxId))";
|
|
$conn->executeQuery($sqlSetval);
|
|
}
|
|
}
|
|
}
|