mirror/Part-DB.Part-DB-server
1
0
Fork 1
mirror of https://github.com/Part-DB/Part-DB-server.git synced 2025-06-27 04:08:57 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixed ResetAutoIncrementORMPurger with DBAL3

Browse source
This commit is contained in:
Jan Böhmer 2022-03-04 13:32:53 +01:00
parent 15a31433e1
commit 775a223ab2
1 changed files with 23 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

32
src/Doctrine/Purger/ResetAutoIncrementORMPurger.php
View file

@ -7,7 +7,9 @@ namespace App\Doctrine\Purger;
use Doctrine\Common\DataFixtures\Purger\ORMPurgerInterface; use Doctrine\Common\DataFixtures\Purger\ORMPurgerInterface;
use Doctrine\Common\DataFixtures\Purger\PurgerInterface; use Doctrine\Common\DataFixtures\Purger\PurgerInterface;
use Doctrine\Common\DataFixtures\Sorter\TopologicalSorter; use Doctrine\Common\DataFixtures\Sorter\TopologicalSorter;
use Doctrine\DBAL\Platforms\AbstractMySQLPlatform;
use Doctrine\DBAL\Platforms\AbstractPlatform; use Doctrine\DBAL\Platforms\AbstractPlatform;
use Doctrine\DBAL\Platforms\MySQLPlatform;
use Doctrine\DBAL\Schema\Identifier; use Doctrine\DBAL\Schema\Identifier;
use Doctrine\ORM\EntityManagerInterface; use Doctrine\ORM\EntityManagerInterface;
use Doctrine\ORM\Mapping\ClassMetadata; use Doctrine\ORM\Mapping\ClassMetadata;
@ -134,13 +136,19 @@ class ResetAutoIncrementORMPurger implements PurgerInterface, ORMPurgerInterface
} }
$connection = $this->em->getConnection(); $connection = $this->em->getConnection();
$filterExpr = $connection->getConfiguration()->getFilterSchemaAssetsExpression(); $filterExpr = method_exists(
$connection->getConfiguration(),
'getFilterSchemaAssetsExpression'
) ? $connection->getConfiguration()->getFilterSchemaAssetsExpression() : null;
$emptyFilterExpression = empty($filterExpr); $emptyFilterExpression = empty($filterExpr);
$schemaAssetsFilter = method_exists($connection->getConfiguration(), 'getSchemaAssetsFilter') ? $connection->getConfiguration()->getSchemaAssetsFilter() : null; $schemaAssetsFilter = method_exists(
$connection->getConfiguration(),
'getSchemaAssetsFilter'
) ? $connection->getConfiguration()->getSchemaAssetsFilter() : null;
//Disable foreign key checks //Disable foreign key checks
if($platform->getName() === 'mysql') { if($platform instanceof AbstractMySQLPlatform) {
$connection->executeQuery('SET foreign_key_checks = 0;'); $connection->executeQuery('SET foreign_key_checks = 0;');
} }
@ -161,20 +169,20 @@ class ResetAutoIncrementORMPurger implements PurgerInterface, ORMPurgerInterface
} }
if ($this->purgeMode === self::PURGE_MODE_DELETE) { if ($this->purgeMode === self::PURGE_MODE_DELETE) {
$connection->executeUpdate($this->getDeleteFromTableSQL($tbl, $platform)); $connection->executeStatement($this->getDeleteFromTableSQL($tbl, $platform));
} else { } else {
$connection->executeUpdate($platform->getTruncateTableSQL($tbl, true)); $connection->executeStatement($platform->getTruncateTableSQL($tbl, true));
} }
//Reseting autoincrement is only supported on MySQL platforms //Reseting autoincrement is only supported on MySQL platforms
if ($platform->getName() === 'mysql') { if ($platform instanceof AbstractMySQLPlatform) {
$connection->beginTransaction(); $connection->beginTransaction();
$connection->executeQuery($this->getResetAutoIncrementSQL($tbl, $platform)); $connection->executeQuery($this->getResetAutoIncrementSQL($tbl, $platform));
} }
} }
//Reenable foreign key checks //Reenable foreign key checks
if($platform->getName() === 'mysql') { if($platform instanceof AbstractMySQLPlatform) {
$connection->executeQuery('SET foreign_key_checks = 1;'); $connection->executeQuery('SET foreign_key_checks = 1;');
} }
} }
@ -270,7 +278,10 @@ class ResetAutoIncrementORMPurger implements PurgerInterface, ORMPurgerInterface
private function getTableName(ClassMetadata $class, AbstractPlatform $platform): string private function getTableName(ClassMetadata $class, AbstractPlatform $platform): string
{ {
if (isset($class->table['schema']) && ! method_exists($class, 'getSchemaName')) { if (isset($class->table['schema']) && ! method_exists($class, 'getSchemaName')) {
return $class->table['schema'] . '.' . $this->em->getConfiguration()->getQuoteStrategy()->getTableName($class, $platform); return $class->table['schema'] . '.' .
$this->em->getConfiguration()
->getQuoteStrategy()
->getTableName($class, $platform);
} }
return $this->em->getConfiguration()->getQuoteStrategy()->getTableName($class, $platform); return $this->em->getConfiguration()->getQuoteStrategy()->getTableName($class, $platform);
@ -285,7 +296,10 @@ class ResetAutoIncrementORMPurger implements PurgerInterface, ORMPurgerInterface
AbstractPlatform $platform AbstractPlatform $platform
): string { ): string {
if (isset($assoc['joinTable']['schema']) && ! method_exists($class, 'getSchemaName')) { if (isset($assoc['joinTable']['schema']) && ! method_exists($class, 'getSchemaName')) {
return $assoc['joinTable']['schema'] . '.' . $this->em->getConfiguration()->getQuoteStrategy()->getJoinTableName($assoc, $class, $platform); return $assoc['joinTable']['schema'] . '.' .
$this->em->getConfiguration()
->getQuoteStrategy()
->getJoinTableName($assoc, $class, $platform);
} }
return $this->em->getConfiguration()->getQuoteStrategy()->getJoinTableName($assoc, $class, $platform); return $this->em->getConfiguration()->getQuoteStrategy()->getJoinTableName($assoc, $class, $platform);