ORMRepository trait

ORMRepository.md

use Doctrine\ORM\EntityManager;
use Doctrine\ORM\EntityRepository;
use Doctrine\ORM\QueryBuilder;
use Zenstruck\Porpaginas\Doctrine\ORMQueryIterateResult;
use Zenstruck\Porpaginas\Result;

trait ORMRepository
{
    /** @var EntityManager */
    protected $em;

    /**
     * This annotation triggers setter injection when autowiring.
     *
     * @required
     */
    public function setEntityManager(EntityManager $em): void
    {
        $this->em = $em;
    }

    public function save($object): void
    {
        $this->em->persist($object);
        $this->em->flush();
    }

    public function count(): int
    {
        return $this->getIterator()->count();
    }

    public function getIterator(): Result
    {
        return new ORMQueryIterateResult($this->em, $this->qb());
    }

    protected function repo(): EntityRepository
    {
        return $this->em->getRepository($this->getEntityName());
    }

    protected function qb(string $alias = 'entity'): QueryBuilder
    {
        return $this->repo()->createQueryBuilder($alias);
    }

    abstract protected function getEntityName(): string;
}