Cavemanon/piratepoet
8
Fork 0
Code Pull Requests Actions Releases 16 Activity

Add a helper method to detach all (or some) entities of a given class from entity manager

Browse Source 
[MAILPOET-5745]
This commit is contained in:
Jan Jakes
2023-11-28 15:59:27 +01:00
committed by Aschepikov
parent 39aa7ad27d
commit 076ee6d5de
2 changed files with 102 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
mailpoet/lib/Doctrine/Repository.php
View File

@@ -126,6 +126,24 @@ abstract class Repository {
$this->entityManager->detach($entity); $this->entityManager->detach($entity);
} }
/**
* @param callable(T): bool|null $filter
*/
public function detachAll(callable $filter = null): void {
$className = $this->getEntityClassName();
$rootClassName = $this->entityManager->getClassMetadata($className)->rootEntityName;
$entities = $this->entityManager->getUnitOfWork()->getIdentityMap()[$rootClassName] ?? [];
foreach ($entities as $entity) {
if (!($entity instanceof $className)) {
continue;
}
if ($filter && !$filter($entity)) {
continue;
}
$this->entityManager->detach($entity);
}
}
/** /**
* @return class-string<T> * @return class-string<T>
*/ */

84
mailpoet/tests/integration/Doctrine/RepositoryTest.php Normal file
View File

@@ -0,0 +1,84 @@
<?php declare(strict_types = 1);
namespace MailPoet\Test\Doctrine;
use MailPoet\Doctrine\Repository;
use MailPoet\Entities\SettingEntity;
class RepositoryTest extends \MailPoetTest {
public function testItCanPersistAndFlush(): void {
$repository = $this->createRepository();
$setting = new SettingEntity();
$setting->setName('name');
$setting->setValue('value');
$repository->persist($setting);
$repository->flush();
$this->assertSame($this->getEntityFromIdentityMap($setting->getId()), $setting);
}
public function testItCanDetachAll(): void {
$repository = $this->createRepository();
$setting1 = $this->createSetting('name-1', 'value-1');
$setting2 = $this->createSetting('name-2', 'value-2');
$setting3 = $this->createSetting('name-3', 'value-3');
$this->assertSame($this->getEntityFromIdentityMap($setting1->getId()), $setting1);
$this->assertSame($this->getEntityFromIdentityMap($setting2->getId()), $setting2);
$this->assertSame($this->getEntityFromIdentityMap($setting3->getId()), $setting3);
$repository->detachAll();
$this->assertNull($this->getEntityFromIdentityMap($setting1->getId()));
$this->assertNull($this->getEntityFromIdentityMap($setting2->getId()));
$this->assertNull($this->getEntityFromIdentityMap($setting3->getId()));
}
public function testItCanDetachSelectively(): void {
$repository = $this->createRepository();
$setting1 = $this->createSetting('name-1', 'value-1');
$setting2 = $this->createSetting('name-2', 'value-2');
$setting3 = $this->createSetting('name-3', 'value-3');
$this->assertSame($this->getEntityFromIdentityMap($setting1->getId()), $setting1);
$this->assertSame($this->getEntityFromIdentityMap($setting2->getId()), $setting2);
$this->assertSame($this->getEntityFromIdentityMap($setting3->getId()), $setting3);
$repository->detachAll(function (SettingEntity $setting) use ($setting1, $setting3) {
return !in_array($setting->getId(), [$setting1->getId(), $setting3->getId()], true);
});
$this->assertSame($this->getEntityFromIdentityMap($setting1->getId()), $setting1);
$this->assertNull($this->getEntityFromIdentityMap($setting2->getId()));
$this->assertSame($this->getEntityFromIdentityMap($setting3->getId()), $setting3);
}
private function createSetting(string $name, string $value): SettingEntity {
$setting = new SettingEntity();
$setting->setName($name);
$setting->setValue($value);
$this->entityManager->persist($setting);
$this->entityManager->flush();
return $setting;
}
/** @return Repository<SettingEntity> */
private function createRepository(): Repository {
/** @var Repository<SettingEntity> $repository */
$repository = new class($this->entityManager) extends Repository {
protected function getEntityClassName(): string {
return SettingEntity::class;
}
};
return $repository;
}
private function getEntityFromIdentityMap(?int $id): ?SettingEntity {
/** @var SettingEntity|false $entity */
$entity = $this->entityManager->getUnitOfWork()->tryGetById($id, SettingEntity::class);
return $entity ?: null;
}
}