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

Add a helper method to refresh all (or some) entities of a given class

Browse Source 
[MAILPOET-5845]
This commit is contained in:
Jan Jakes
2024-01-18 17:55:15 +01:00
committed by Aschepikov
parent aa13c8956d
commit 682527cd1b
2 changed files with 92 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -111,6 +111,24 @@ abstract class Repository {
$this->entityManager->refresh($entity);
}
/**
* @param callable(T): bool|null $filter
*/
public function refreshAll(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->refresh($entity);
}
}
public function flush() {
$this->entityManager->flush();
}

74
mailpoet/tests/integration/Doctrine/RepositoryTest.php
View File

@ -56,6 +56,80 @@ class RepositoryTest extends \MailPoetTest {
$this->assertSame($this->getEntityFromIdentityMap($setting3->getId()), $setting3);
}
public function testItCanRefreshAll(): 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->entityManager->createQueryBuilder()
->update(SettingEntity::class, 's')
->set('s.value', ':value')
->where('s.name = :name')
->setParameter('value', 'new-value-1')
->setParameter('name', 'name-1')
->getQuery()
->execute();
$this->entityManager->createQueryBuilder()
->update(SettingEntity::class, 's')
->set('s.value', ':value')
->where('s.name = :name')
->setParameter('value', 'new-value-2')
->setParameter('name', 'name-2')
->getQuery()
->execute();
$this->assertSame($setting1->getValue(), 'value-1');
$this->assertSame($setting2->getValue(), 'value-2');
$this->assertSame($setting3->getValue(), 'value-3');
$repository->refreshAll();
$this->assertSame($setting1->getValue(), 'new-value-1');
$this->assertSame($setting2->getValue(), 'new-value-2');
$this->assertSame($setting3->getValue(), 'value-3');
}
public function testItCanRefreshSelectively(): 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->entityManager->createQueryBuilder()
->update(SettingEntity::class, 's')
->set('s.value', ':value')
->where('s.name = :name')
->setParameter('value', 'new-value-1')
->setParameter('name', 'name-1')
->getQuery()
->execute();
$this->entityManager->createQueryBuilder()
->update(SettingEntity::class, 's')
->set('s.value', ':value')
->where('s.name = :name')
->setParameter('value', 'new-value-2')
->setParameter('name', 'name-2')
->getQuery()
->execute();
$this->assertSame($setting1->getValue(), 'value-1');
$this->assertSame($setting2->getValue(), 'value-2');
$this->assertSame($setting3->getValue(), 'value-3');
$repository->refreshAll(function (SettingEntity $setting) {
return in_array($setting->getName(), ['name-1', 'name-3'], true);
});
$this->assertSame($setting1->getValue(), 'new-value-1');
$this->assertSame($setting2->getValue(), 'value-2');
$this->assertSame($setting3->getValue(), 'value-3');
}
private function createSetting(string $name, string $value): SettingEntity {
$setting = new SettingEntity();
$setting->setName($name);