src/Repository/BaseRepository.php line 66

Open in your IDE?
  1. <?php
  3. namespace App\Repository;
  5. use App\Entity\EntityInterface;
  6. use Doctrine\Bundle\DoctrineBundle\Repository\ServiceEntityRepositoryInterface;
  7. use Doctrine\Common\Persistence\ManagerRegistry;
  8. use Doctrine\ORM\EntityManager;
  9. use Doctrine\ORM\EntityManagerInterface;
  10. use Doctrine\ORM\EntityRepository;
  11. use Doctrine\ORM\Mapping\ClassMetadata;
  12. use Symfony\Component\HttpFoundation\Response;
  13. use Symfony\Component\HttpKernel\Exception\NotFoundHttpException;
  15. /**
  16.  * BaseRepository
  17.  *
  18.  * @author Zwer<ante@q-software.com>
  19.  */
  20. abstract class BaseRepository extends EntityRepository implements ServiceEntityRepositoryInterface
  21. {
  22.     public const ENTITY_CLASS_NAME = '';
  24.     public function __construct(EntityManagerInterface $entityManager, ?ClassMetadata $metadata = null, ?ManagerRegistry $registry = null)
  25.     {
  26.         if (static::ENTITY_CLASS_NAME === '') {
  27.             throw new \RuntimeException('Repository entity class name is empty');
  28.         }
  30.         if ($registry) {
  31.             /** @var EntityManager $manager */
  32.             $manager = $registry->getManagerForClass($this::ENTITY_CLASS_NAME);
  34.             parent::__construct($manager, $manager->getClassMetadata($this::ENTITY_CLASS_NAME));
  35.         } elseif ($entityManager instanceof EntityManager && $metadata instanceof ClassMetadata) {
  36.             parent::__construct($entityManager, $metadata);
  37.         } else {
  38.             throw new \RuntimeException('Failed to initialize repository.');
  39.         }
  40.     }
  42.     public function save(EntityInterface $entity, bool $flush = true): EntityInterface
  43.     {
  44.         $this->_em->persist($entity);
  45.         if ($flush) {
  46.             $this->_em->flush();
  47.         }
  49.         return $entity;
  50.     }
  52.     public function merge(EntityInterface $entity, bool $flush = true): EntityInterface
  53.     {
  54.         /** @var EntityInterface $entity */
  55.         $entity = $this->_em->merge($entity);
  56.         if ($flush) {
  57.             $this->_em->flush();
  58.             $this->_em->refresh($entity);
  59.         }
  61.         return $entity;
  62.     }
  64.     public function flush(EntityInterface $entity = null): void
  65.     {
  66.         $this->_em->flush($entity);
  67.     }
  69.     public function remove(EntityInterface $entity, bool $flush = true): void
  70.     {
  71.         $this->_em->remove($entity);
  72.         if ($flush) {
  73.             $this->_em->flush();
  74.         }
  75.     }
  77.     public function findOr404(int $id): EntityInterface
  78.     {
  79.         /** @var EntityInterface|null $entity */
  80.         $entity = $this->find($id);
  82.         if (null === $entity) {
  83.             $message = sprintf('Resource of type %s and ID %s could not be found!', $this::ENTITY_CLASS_NAME, $id);
  84.             throw new NotFoundHttpException($message, null, Response::HTTP_NOT_FOUND);
  85.         }
  87.         return $entity;
  88.     }
  89. }