Pagerfanta\Adapter\DoctrineDbalSingleTableAdapter

Complexity

Total Complexity

7

Size/Duplication

Total Lines	51
Duplicated Lines	0 %

Coupling/Cohesion

Components	0
Dependencies	3

Test Coverage

Coverage

100%

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	10	2
A	hasQueryBuilderJoins()	0	6	1
A	createCountQueryModifier()	0	10	1
A	createSelectForCountField()	0	8	2
A	countFieldHasNoAlias()	0	4	1

<?php

/*
 * This file is part of the Pagerfanta package.
 *
 * (c) Pablo Díez <pablodip@gmail.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Pagerfanta\Adapter;

use Doctrine\DBAL\Query\QueryBuilder;
use Pagerfanta\Exception\InvalidArgumentException;

/**
 * @author Michael Williams <michael@whizdevelopment.com>
 * @author Pablo Díez <pablodip@gmail.com>
 */
class DoctrineDbalSingleTableAdapter extends DoctrineDbalAdapter
{
    /**
     * Constructor.
     *
     * @param QueryBuilder $queryBuilder A DBAL query builder.
     * @param string       $countField   Primary key for the table in query. Used in count expression. Must include table alias
     */
    public function __construct(QueryBuilder $queryBuilder, $countField)
    {
        if ($this->hasQueryBuilderJoins($queryBuilder)) {
            throw new InvalidArgumentException('The query builder cannot have joins.');
        }

        $countQueryBuilderModifier = $this->createCountQueryModifier($countField);

        parent::__construct($queryBuilder, $countQueryBuilderModifier);
    }

    private function hasQueryBuilderJoins(QueryBuilder $queryBuilder)
    {
        $joins = $queryBuilder->getQueryPart('join');

        return !empty($joins);
    }

    private function createCountQueryModifier($countField)
    {
        $select = $this->createSelectForCountField($countField);

        return function (QueryBuilder $queryBuilder) use ($select) {
            $queryBuilder->select($select)
                         ->resetQueryPart('orderBy')
                         ->setMaxResults(1);
        };
    }

    private function createSelectForCountField($countField)
    {
        if ($this->countFieldHasNoAlias($countField)) {
            throw new InvalidArgumentException('The $countField must contain a table alias in the string.');
        }

        return sprintf('COUNT(DISTINCT %s) AS total_results', $countField);
    }

    private function countFieldHasNoAlias($countField)
    {
        return strpos($countField, '.') === false;
    }
}