DH\DoctrineAuditBundle\Reader\Paginator

Complexity

Total Complexity

5

Size/Duplication

Total Lines	58
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	3	1
A	getIterator()	0	13	1
A	count()	0	17	2
A	cloneQuery()	0	7	1

<?php

namespace DH\DoctrineAuditBundle\Reader;

use ArrayIterator;
use Countable;
use Doctrine\DBAL\Query\QueryBuilder;
use IteratorAggregate;

class Paginator implements Countable, IteratorAggregate
{
    /**
     * @var QueryBuilder
     */
    private $queryBuilder;

    /**
     * @var int
     */
    private $count;

    public function __construct(QueryBuilder $queryBuilder)
    {
        $this->queryBuilder = $queryBuilder;
    }

    public function count(): int
    {
        $queryBuilder = $this->cloneQuery($this->queryBuilder);

        $result = $queryBuilder
            ->resetQueryPart('select')
            ->resetQueryPart('orderBy')
            ->setMaxResults(null)
            ->setFirstResult(null)
            ->select('COUNT(id)')
            ->execute()
            ->fetchColumn(0)
        ;

        $this->count = false === $result ? 0 : $result;

        return $this->count;
    }

    public function getIterator()
    {
        $offset = $this->queryBuilder->getFirstResult();
        $length = $this->queryBuilder->getMaxResults();

        $result = $this->cloneQuery($this->queryBuilder)
            ->setMaxResults($length)
            ->setFirstResult($offset)
            ->execute()
            ->fetchAll()
        ;

        return new ArrayIterator($result);
    }

    private function cloneQuery(QueryBuilder $queryBuilder): QueryBuilder
    {
        /** @var QueryBuilder $cloneQuery */
        $cloneQuery = clone $queryBuilder;
        $cloneQuery->setParameters($queryBuilder->getParameters(), $queryBuilder->getParameterTypes());

        return $cloneQuery;
    }
}