Zenstruck\Porpaginas\Doctrine\RSMQueryResult

Complexity

Total Complexity

4

Size/Duplication

Total Lines	59
Duplicated Lines	0 %

Coupling/Cohesion

Components	2
Dependencies	5

Test Coverage

Coverage

100%

Importance

Changes	1
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	7	1
A	take()	0	15	1
A	count()	0	4	1
A	getIterator()	0	7	1

<?php

namespace Zenstruck\Porpaginas\Doctrine;

use Doctrine\DBAL\Query\QueryBuilder;
use Doctrine\ORM\EntityManagerInterface;
use Doctrine\ORM\Query\ResultSetMappingBuilder;
use Zenstruck\Porpaginas\Callback\CallbackPage;
use Zenstruck\Porpaginas\Result;

/**
 * @author Florian Klein <florian.klein@free.fr>
 * @author Kevin Bond <kevinbond@gmail.com>
 */
final class RSMQueryResult implements Result
{
    private $em;
    private $rsm;
    private $qb;
    private $queryBuilderResult;

    /**
     * @param EntityManagerInterface  $em
     * @param ResultSetMappingBuilder $rsm
     * @param QueryBuilder            $qb
     * @param callable|null           $countQueryBuilderModifier
     */
    public function __construct(EntityManagerInterface $em, ResultSetMappingBuilder $rsm, QueryBuilder $qb, callable $countQueryBuilderModifier = null)
    {
        $this->em = $em;
        $this->rsm = $rsm;
        $this->qb = $qb;
        $this->queryBuilderResult = new DBALQueryBuilderResult($qb, $countQueryBuilderModifier);
    }

    /**
     * {@inheritdoc}
     */
    public function take($offset, $limit)
    {
        $qb = clone $this->qb;
        $results = function ($offset, $limit) use ($qb) {
            $qb->setFirstResult($offset)
                ->setMaxResults($limit);

            $query = $this->em->createNativeQuery($qb->getSql(), $this->rsm);
            $query->setParameters($qb->getParameters());

            return $query->execute();
        };

        return new CallbackPage($results, [$this, 'count'], $offset, $limit);
    }

    /**
     * {@inheritdoc}
     */
    public function count()
    {
        return $this->queryBuilderResult->count();
    }

    /**
     * {@inheritdoc}
     */
    public function getIterator()
    {
        $query = $this->em->createNativeQuery($this->qb->getSql(), $this->rsm);
        $query->setParameters($this->qb->getParameters());

        return $query->iterate();
    }
}