diff --git a/lib/Doctrine/ORM/Tools/Pagination/LimitSubqueryOutputWalker.php b/lib/Doctrine/ORM/Tools/Pagination/LimitSubqueryOutputWalker.php index 338920b2ef0..0c17a13579e 100644 --- a/lib/Doctrine/ORM/Tools/Pagination/LimitSubqueryOutputWalker.php +++ b/lib/Doctrine/ORM/Tools/Pagination/LimitSubqueryOutputWalker.php @@ -13,10 +13,15 @@ namespace Doctrine\ORM\Tools\Pagination; +use Doctrine\ORM\Query\AST\ArithmeticExpression; +use Doctrine\ORM\Query\AST\ArithmeticTerm; +use Doctrine\ORM\Query\AST\OrderByClause; +use Doctrine\ORM\Query\AST\PartialObjectExpression; use Doctrine\ORM\Query\AST\PathExpression; +use Doctrine\ORM\Query\AST\SelectExpression; +use Doctrine\ORM\Query\Expr\OrderBy; use Doctrine\ORM\Query\SqlWalker; use Doctrine\ORM\Query\AST\SelectStatement; -use Doctrine\DBAL\Platforms\PostgreSqlPlatform; /** * Wraps the query in order to select root entity IDs for pagination. @@ -56,6 +61,23 @@ class LimitSubqueryOutputWalker extends SqlWalker */ private $maxResults; + /** + * @var \Doctrine\ORM\EntityManager + */ + private $em; + + /** + * @var array + */ + private $orderByPathExpressions = []; + + /** + * The quote strategy. + * + * @var \Doctrine\ORM\Mapping\QuoteStrategy + */ + private $quoteStrategy; + /** * Constructor. * @@ -78,6 +100,9 @@ public function __construct($query, $parserResult, array $queryComponents) $this->maxResults = $query->getMaxResults(); $query->setFirstResult(null)->setMaxResults(null); + $this->em = $query->getEntityManager(); + $this->quoteStrategy = $this->em->getConfiguration()->getQuoteStrategy(); + parent::__construct($query, $parserResult, $queryComponents); } @@ -85,13 +110,27 @@ public function __construct($query, $parserResult, array $queryComponents) * Walks down a SelectStatement AST node, wrapping it in a SELECT DISTINCT. * * @param SelectStatement $AST + * @param bool $addMissingItemsFromOrderByToSelect * * @return string * * @throws \RuntimeException */ - public function walkSelectStatement(SelectStatement $AST) + public function walkSelectStatement(SelectStatement $AST, $addMissingItemsFromOrderByToSelect = true) { + // We don't want to call this recursively! + if ($AST->orderByClause instanceof OrderByClause && $addMissingItemsFromOrderByToSelect) { + // In the case of ordering a query by columns from joined tables, we + // must add those columns to the select clause of the query BEFORE + // the SQL is generated. + $this->addMissingItemsFromOrderByToSelect($AST); + } + + // Remove order by clause from the inner query + // It will be re-appended in the outer select generated by this method + $orderByClause = $AST->orderByClause; + $AST->orderByClause = null; + // Set every select expression as visible(hidden = false) to // make $AST have scalar mappings properly - this is relevant for referencing selected // fields from outside the subquery, for example in the ORDER BY segment @@ -104,6 +143,9 @@ public function walkSelectStatement(SelectStatement $AST) $innerSql = parent::walkSelectStatement($AST); + // Restore orderByClause + $AST->orderByClause = $orderByClause; + // Restore hiddens foreach ($AST->selectClause->selectExpressions as $idx => $expr) { $expr->hiddenAliasResultVariable = $hiddens[$idx]; @@ -163,7 +205,7 @@ public function walkSelectStatement(SelectStatement $AST) implode(', ', $sqlIdentifier), $innerSql); // http://www.doctrine-project.org/jira/browse/DDC-1958 - $sql = $this->preserveSqlOrdering($AST, $sqlIdentifier, $innerSql, $sql); + $sql = $this->preserveSqlOrdering($sqlIdentifier, $innerSql, $sql, $orderByClause); // Apply the limit and offset. $sql = $this->platform->modifyLimitQuery( @@ -182,49 +224,183 @@ public function walkSelectStatement(SelectStatement $AST) } /** - * Generates new SQL for Postgresql or Oracle if necessary. + * Finds all PathExpressions in an AST's OrderByClause, and ensures that + * the referenced fields are present in the SelectClause of the passed AST. * * @param SelectStatement $AST + */ + private function addMissingItemsFromOrderByToSelect(SelectStatement $AST) + { + $this->orderByPathExpressions = []; + + // We need to do this in another walker because otherwise we'll end up + // polluting the state of this one. + $walker = clone $this; + + // This will populate $orderByPathExpressions via + // LimitSubqueryOutputWalker::walkPathExpression, which will be called + // as the select statement is walked. We'll end up with an array of all + // path expressions referenced in the query. + $walker->walkSelectStatement($AST, false); + $orderByPathExpressions = $walker->getOrderByPathExpressions(); + + // Get a map of referenced identifiers to field names. + $selects = []; + foreach ($orderByPathExpressions as $pathExpression) { + $idVar = $pathExpression->identificationVariable; + $field = $pathExpression->field; + if (!isset($selects[$idVar])) { + $selects[$idVar] = []; + } + $selects[$idVar][$field] = true; + } + + // Loop the select clause of the AST and exclude items from $select + // that are already being selected in the query. + foreach ($AST->selectClause->selectExpressions as $selectExpression) { + if ($selectExpression instanceof SelectExpression) { + $idVar = $selectExpression->expression; + if (!is_string($idVar)) { + continue; + } + $field = $selectExpression->fieldIdentificationVariable; + if ($field === null) { + // No need to add this select, as we're already fetching the whole object. + unset($selects[$idVar]); + } else { + unset($selects[$idVar][$field]); + } + } + } + + // Add select items which were not excluded to the AST's select clause. + foreach ($selects as $idVar => $fields) { + $AST->selectClause->selectExpressions[] = new SelectExpression(new PartialObjectExpression($idVar, array_keys($fields)), null, true); + } + } + + /** + * Generates new SQL for statements with an order by clause + * * @param array $sqlIdentifier * @param string $innerSql * @param string $sql + * @param OrderByClause $orderByClause * - * @return void + * @return string */ - public function preserveSqlOrdering(SelectStatement $AST, array $sqlIdentifier, $innerSql, $sql) + private function preserveSqlOrdering(array $sqlIdentifier, $innerSql, $sql, $orderByClause) { - // For every order by, find out the SQL alias by inspecting the ResultSetMapping. - $sqlOrderColumns = array(); - $orderBy = array(); - if (isset($AST->orderByClause)) { - foreach ($AST->orderByClause->orderByItems as $item) { - $expression = $item->expression; - - $possibleAliases = $expression instanceof PathExpression - ? array_keys($this->rsm->fieldMappings, $expression->field) - : array_keys($this->rsm->scalarMappings, $expression); - - foreach ($possibleAliases as $alias) { - if (!is_object($expression) || $this->rsm->columnOwnerMap[$alias] == $expression->identificationVariable) { - $sqlOrderColumns[] = $alias; - $orderBy[] = $alias . ' ' . $item->type; - break; - } - } - } - // remove identifier aliases - $sqlOrderColumns = array_diff($sqlOrderColumns, $sqlIdentifier); + // If the sql statement has an order by clause, we need to wrap it in a new select distinct + // statement + if (! $orderByClause instanceof OrderByClause) { + return $sql; } - if (count($orderBy)) { - $sql = sprintf( - 'SELECT DISTINCT %s FROM (%s) dctrn_result ORDER BY %s', - implode(', ', array_merge($sqlIdentifier, $sqlOrderColumns)), - $innerSql, - implode(', ', $orderBy) + // Rebuild the order by clause to work in the scope of the new select statement + /* @var array $sqlOrderColumns an array of items that need to be included in the select list */ + /* @var array $orderBy an array of rebuilt order by items */ + list($sqlOrderColumns, $orderBy) = $this->rebuildOrderByClauseForOuterScope($orderByClause); + + // Identifiers are always included in the select list, so there's no need to include them twice + $sqlOrderColumns = array_diff($sqlOrderColumns, $sqlIdentifier); + + // Build the select distinct statement + $sql = sprintf( + 'SELECT DISTINCT %s FROM (%s) dctrn_result ORDER BY %s', + implode(', ', array_merge($sqlIdentifier, $sqlOrderColumns)), + $innerSql, + implode(', ', $orderBy) + ); + + return $sql; + } + + /** + * Generates a new order by clause that works in the scope of a select query wrapping the original + * + * @param OrderByClause $orderByClause + * @return array + */ + private function rebuildOrderByClauseForOuterScope(OrderByClause $orderByClause) + { + $dqlAliasToSqlTableAliasMap + = $searchPatterns + = $replacements + = $dqlAliasToClassMap + = $selectListAdditions + = $orderByItems + = []; + + // Generate DQL alias -> SQL table alias mapping + foreach(array_keys($this->rsm->aliasMap) as $dqlAlias) { + $dqlAliasToClassMap[$dqlAlias] = $class = $this->queryComponents[$dqlAlias]['metadata']; + $dqlAliasToSqlTableAliasMap[$dqlAlias] = $this->getSQLTableAlias($class->getTableName(), $dqlAlias); + } + + // Pattern to find table path expressions in the order by clause + $fieldSearchPattern = '/(?rsm->fieldMappings as $fieldAlias => $columnName) { + $dqlAliasForFieldAlias = $this->rsm->columnOwnerMap[$fieldAlias]; + $columnName = $this->quoteStrategy->getColumnName( + $columnName, + $dqlAliasToClassMap[$dqlAliasForFieldAlias], + $this->em->getConnection()->getDatabasePlatform() ); + + $sqlTableAliasForFieldAlias = $dqlAliasToSqlTableAliasMap[$dqlAliasForFieldAlias]; + + $searchPatterns[] = sprintf($fieldSearchPattern, $sqlTableAliasForFieldAlias, $columnName); + $replacements[] = $fieldAlias; } - return $sql; + $complexAddedOrderByAliases = 0; + foreach($orderByClause->orderByItems as $orderByItem) { + // Walk order by item to get string representation of it + $orderByItemString = $this->walkOrderByItem($orderByItem); + + // Replace path expressions in the order by clause with their column alias + $orderByItemString = preg_replace($searchPatterns, $replacements, $orderByItemString); + + // The order by items are not required to be in the select list on Oracle and PostgreSQL, but + // for the sake of simplicity, order by items will be included in the select list on all platforms. + // This doesn't impact functionality. + $selectListAddition = trim(preg_replace('/([^ ]+) (?:asc|desc)/i', '$1', $orderByItemString)); + + // If the expression is an arithmetic expression, we need to create an alias for it. + if ($orderByItem->expression instanceof ArithmeticTerm) { + $orderByAlias = "ordr_" . $complexAddedOrderByAliases++; + $orderByItemString = $orderByAlias . " " . $orderByItem->type; + $selectListAddition .= " AS $orderByAlias"; + } + $selectListAdditions[] = $selectListAddition; + $orderByItems[] = $orderByItemString; + } + + return array($selectListAdditions, $orderByItems); + } + + /** + * {@inheritdoc} + */ + public function walkPathExpression($pathExpr) + { + if (!in_array($pathExpr, $this->orderByPathExpressions)) { + $this->orderByPathExpressions[] = $pathExpr; + } + + return parent::walkPathExpression($pathExpr); + } + + /** + * getter for $orderByPathExpressions + * + * @return array + */ + public function getOrderByPathExpressions() + { + return $this->orderByPathExpressions; } } diff --git a/tests/Doctrine/Tests/Models/Pagination/Company.php b/tests/Doctrine/Tests/Models/Pagination/Company.php new file mode 100644 index 00000000000..f14d1b91edf --- /dev/null +++ b/tests/Doctrine/Tests/Models/Pagination/Company.php @@ -0,0 +1,30 @@ +useModelSet('cms'); + $this->useModelSet('pagination'); parent::setUp(); $this->populate(); } @@ -30,12 +29,12 @@ protected function setUp() */ public function testCountSimpleWithoutJoin($useOutputWalkers) { - $dql = "SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u"; + $dql = 'SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u'; $query = $this->_em->createQuery($dql); $paginator = new Paginator($query); $paginator->setUseOutputWalkers($useOutputWalkers); - $this->assertCount(3, $paginator); + $this->assertCount(9, $paginator); } /** @@ -43,77 +42,417 @@ public function testCountSimpleWithoutJoin($useOutputWalkers) */ public function testCountWithFetchJoin($useOutputWalkers) { - $dql = "SELECT u,g FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.groups g"; + $dql = 'SELECT u,g FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.groups g'; $query = $this->_em->createQuery($dql); $paginator = new Paginator($query); $paginator->setUseOutputWalkers($useOutputWalkers); - $this->assertCount(3, $paginator); + $this->assertCount(9, $paginator); } public function testCountComplexWithOutputWalker() { - $dql = "SELECT g, COUNT(u.id) AS userCount FROM Doctrine\Tests\Models\CMS\CmsGroup g LEFT JOIN g.users u GROUP BY g HAVING COUNT(u.id) > 0"; + $dql = 'SELECT g, COUNT(u.id) AS userCount FROM Doctrine\Tests\Models\CMS\CmsGroup g LEFT JOIN g.users u GROUP BY g HAVING COUNT(u.id) > 0'; $query = $this->_em->createQuery($dql); $paginator = new Paginator($query); $paginator->setUseOutputWalkers(true); - $this->assertCount(9, $paginator); + $this->assertCount(3, $paginator); + } + + public function testCountComplexWithoutOutputWalker() + { + $dql = 'SELECT g, COUNT(u.id) AS userCount FROM Doctrine\Tests\Models\CMS\CmsGroup g LEFT JOIN g.users u GROUP BY g HAVING COUNT(u.id) > 0'; + $query = $this->_em->createQuery($dql); + + $paginator = new Paginator($query); + $paginator->setUseOutputWalkers(false); + + $this->setExpectedException( + 'RuntimeException', + 'Cannot count query that uses a HAVING clause. Use the output walkers for pagination' + ); + $this->assertCount(3, $paginator); } /** * @dataProvider useOutputWalkers */ - public function testIterateSimpleWithoutJoinFetchJoinHandlingOff($useOutputWalkers) + public function testCountWithComplexScalarOrderBy($useOutputWalkers) { - $dql = "SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u"; + $dql = 'SELECT l FROM Doctrine\Tests\Models\Pagination\Logo l ORDER BY l.image_width * l.image_height DESC'; $query = $this->_em->createQuery($dql); - $paginator = new Paginator($query, false); + $paginator = new Paginator($query); $paginator->setUseOutputWalkers($useOutputWalkers); - $this->assertCount(3, $paginator->getIterator()); + $this->assertCount(9, $paginator); } /** - * @dataProvider useOutputWalkers + * @dataProvider useOutputWalkersAndFetchJoinCollection */ - public function testIterateSimpleWithoutJoinFetchJoinHandlingOn($useOutputWalkers) + public function testIterateSimpleWithoutJoin($useOutputWalkers, $fetchJoinCollection) { - $dql = "SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u"; + $dql = 'SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u'; $query = $this->_em->createQuery($dql); - $paginator = new Paginator($query, true); + $paginator = new Paginator($query, $fetchJoinCollection); + $paginator->setUseOutputWalkers($useOutputWalkers); + $this->assertCount(9, $paginator->getIterator()); + + // Test with limit + $query->setMaxResults(3); + $paginator = new Paginator($query, $fetchJoinCollection); + $paginator->setUseOutputWalkers($useOutputWalkers); + $this->assertCount(3, $paginator->getIterator()); + + // Test with limit and offset + $query->setMaxResults(3)->setFirstResult(4); + $paginator = new Paginator($query, $fetchJoinCollection); $paginator->setUseOutputWalkers($useOutputWalkers); $this->assertCount(3, $paginator->getIterator()); } + private function iterateWithOrderAsc($useOutputWalkers, $fetchJoinCollection, $baseDql, $checkField) + { + + // Ascending + $dql = "$baseDql ASC"; + $query = $this->_em->createQuery($dql); + + $paginator = new Paginator($query, $fetchJoinCollection); + $paginator->setUseOutputWalkers($useOutputWalkers); + $iter = $paginator->getIterator(); + $this->assertCount(9, $iter); + $result = iterator_to_array($iter); + $this->assertEquals($checkField . "0", $result[0]->$checkField); + } + + private function iterateWithOrderAscWithLimit($useOutputWalkers, $fetchJoinCollection, $baseDql, $checkField) + { + + // Ascending + $dql = "$baseDql ASC"; + $query = $this->_em->createQuery($dql); + + // With limit + $query->setMaxResults(3); + $paginator = new Paginator($query, $fetchJoinCollection); + $paginator->setUseOutputWalkers($useOutputWalkers); + $iter = $paginator->getIterator(); + $this->assertCount(3, $iter); + $result = iterator_to_array($iter); + $this->assertEquals($checkField . "0", $result[0]->$checkField); + } + + private function iterateWithOrderAscWithLimitAndOffset($useOutputWalkers, $fetchJoinCollection, $baseDql, $checkField) + { + // Ascending + $dql = "$baseDql ASC"; + $query = $this->_em->createQuery($dql); + + // With offset + $query->setMaxResults(3)->setFirstResult(3); + $paginator = new Paginator($query, $fetchJoinCollection); + $paginator->setUseOutputWalkers($useOutputWalkers); + $iter = $paginator->getIterator(); + $this->assertCount(3, $iter); + $result = iterator_to_array($iter); + $this->assertEquals($checkField . "3", $result[0]->$checkField); + } + + private function iterateWithOrderDesc($useOutputWalkers, $fetchJoinCollection, $baseDql, $checkField) + { + $dql = "$baseDql DESC"; + $query = $this->_em->createQuery($dql); + + $paginator = new Paginator($query, $fetchJoinCollection); + $paginator->setUseOutputWalkers($useOutputWalkers); + $iter = $paginator->getIterator(); + $this->assertCount(9, $iter); + $result = iterator_to_array($iter); + $this->assertEquals($checkField . "8", $result[0]->$checkField); + } + + private function iterateWithOrderDescWithLimit($useOutputWalkers, $fetchJoinCollection, $baseDql, $checkField) + { + $dql = "$baseDql DESC"; + $query = $this->_em->createQuery($dql); + + // With limit + $query->setMaxResults(3); + $paginator = new Paginator($query, $fetchJoinCollection); + $paginator->setUseOutputWalkers($useOutputWalkers); + $iter = $paginator->getIterator(); + $this->assertCount(3, $iter); + $result = iterator_to_array($iter); + $this->assertEquals($checkField . "8", $result[0]->$checkField); + } + + private function iterateWithOrderDescWithLimitAndOffset($useOutputWalkers, $fetchJoinCollection, $baseDql, $checkField) + { + $dql = "$baseDql DESC"; + $query = $this->_em->createQuery($dql); + + // With offset + $query->setMaxResults(3)->setFirstResult(3); + $paginator = new Paginator($query, $fetchJoinCollection); + $paginator->setUseOutputWalkers($useOutputWalkers); + $iter = $paginator->getIterator(); + $this->assertCount(3, $iter); + $result = iterator_to_array($iter); + $this->assertEquals($checkField . "5", $result[0]->$checkField); + } + + /** + * @dataProvider useOutputWalkersAndFetchJoinCollection + */ + public function testIterateSimpleWithoutJoinWithOrder($useOutputWalkers, $fetchJoinCollection) + { + // Ascending + $dql = 'SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u ORDER BY u.username'; + $this->iterateWithOrderAsc($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + $this->iterateWithOrderDesc($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + } + + /** + * @dataProvider useOutputWalkersAndFetchJoinCollection + */ + public function testIterateSimpleWithoutJoinWithOrderAndLimit($useOutputWalkers, $fetchJoinCollection) + { + // Ascending + $dql = 'SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u ORDER BY u.username'; + $this->iterateWithOrderAscWithLimit($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + $this->iterateWithOrderDescWithLimit($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + } + + /** + * @dataProvider useOutputWalkersAndFetchJoinCollection + */ + public function testIterateSimpleWithoutJoinWithOrderAndLimitAndOffset($useOutputWalkers, $fetchJoinCollection) + { + // Ascending + $dql = 'SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u ORDER BY u.username'; + $this->iterateWithOrderAscWithLimitAndOffset($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + $this->iterateWithOrderDescWithLimitAndOffset($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + } + + /** + * @dataProvider fetchJoinCollection + */ + public function testIterateSimpleWithOutputWalkerWithoutJoinWithComplexOrder($fetchJoinCollection) + { + // Ascending + $dql = 'SELECT l FROM Doctrine\Tests\Models\Pagination\Logo l ORDER BY l.image_width * l.image_height'; + $this->iterateWithOrderAsc(true, $fetchJoinCollection, $dql, "image"); + $this->iterateWithOrderDesc(true, $fetchJoinCollection, $dql, "image"); + } + + /** + * @dataProvider fetchJoinCollection + */ + public function testIterateSimpleWithOutputWalkerWithoutJoinWithComplexOrderAndLimit($fetchJoinCollection) + { + // Ascending + $dql = 'SELECT l FROM Doctrine\Tests\Models\Pagination\Logo l ORDER BY l.image_width * l.image_height'; + $this->iterateWithOrderAscWithLimit(true, $fetchJoinCollection, $dql, "image"); + $this->iterateWithOrderDescWithLimit(true, $fetchJoinCollection, $dql, "image"); + } + + /** + * @dataProvider fetchJoinCollection + */ + public function testIterateSimpleWithOutputWalkerWithoutJoinWithComplexOrderAndLimitAndOffset($fetchJoinCollection) + { + // Ascending + $dql = 'SELECT l FROM Doctrine\Tests\Models\Pagination\Logo l ORDER BY l.image_width * l.image_height'; + $this->iterateWithOrderAscWithLimitAndOffset(true, $fetchJoinCollection, $dql, "image"); + $this->iterateWithOrderDescWithLimitAndOffset(true, $fetchJoinCollection, $dql, "image"); + } + /** * @dataProvider useOutputWalkers */ public function testIterateWithFetchJoin($useOutputWalkers) { - $dql = "SELECT u,g FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.groups g"; + $dql = 'SELECT u,g FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.groups g'; $query = $this->_em->createQuery($dql); $paginator = new Paginator($query, true); $paginator->setUseOutputWalkers($useOutputWalkers); - $this->assertCount(3, $paginator->getIterator()); + $this->assertCount(9, $paginator->getIterator()); + } + + /** + * @dataProvider useOutputWalkers + */ + public function testIterateWithFetchJoinWithOrder($useOutputWalkers) + { + $dql = 'SELECT u,g FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.groups g ORDER BY u.username'; + $this->iterateWithOrderAsc($useOutputWalkers, true, $dql, "username"); + $this->iterateWithOrderDesc($useOutputWalkers, true, $dql, "username"); + } + + /** + * @dataProvider useOutputWalkers + */ + public function testIterateWithFetchJoinWithOrderAndLimit($useOutputWalkers) + { + $dql = 'SELECT u,g FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.groups g ORDER BY u.username'; + $this->iterateWithOrderAscWithLimit($useOutputWalkers, true, $dql, "username"); + $this->iterateWithOrderDescWithLimit($useOutputWalkers, true, $dql, "username"); + } + + /** + * @dataProvider useOutputWalkers + */ + public function testIterateWithFetchJoinWithOrderAndLimitAndOffset($useOutputWalkers) + { + $dql = 'SELECT u,g FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.groups g ORDER BY u.username'; + $this->iterateWithOrderAscWithLimitAndOffset($useOutputWalkers, true, $dql, "username"); + $this->iterateWithOrderDescWithLimitAndOffset($useOutputWalkers, true, $dql, "username"); + } + + /** + * @dataProvider useOutputWalkersAndFetchJoinCollection + */ + public function testIterateWithRegularJoinWithOrderByColumnFromJoined($useOutputWalkers, $fetchJoinCollection) + { + $dql = 'SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.email e ORDER BY e.email'; + $this->iterateWithOrderAsc($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + $this->iterateWithOrderDesc($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + } + + /** + * @dataProvider useOutputWalkersAndFetchJoinCollection + */ + public function testIterateWithRegularJoinWithOrderByColumnFromJoinedWithLimit($useOutputWalkers, $fetchJoinCollection) + { + $dql = 'SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.email e ORDER BY e.email'; + $this->iterateWithOrderAscWithLimit($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + $this->iterateWithOrderDescWithLimit($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + } + + /** + * @dataProvider useOutputWalkersAndFetchJoinCollection + */ + public function testIterateWithRegularJoinWithOrderByColumnFromJoinedWithLimitAndOffset($useOutputWalkers, $fetchJoinCollection) + { + $dql = 'SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.email e ORDER BY e.email'; + $this->iterateWithOrderAscWithLimitAndOffset($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + $this->iterateWithOrderDescWithLimitAndOffset($useOutputWalkers, $fetchJoinCollection, $dql, "username"); + } + + /** + * @dataProvider fetchJoinCollection + */ + public function testIterateWithOutputWalkersWithRegularJoinWithComplexOrderByReferencingJoined($fetchJoinCollection) + { + // long function name is loooooooooooong + + $dql = 'SELECT c FROM Doctrine\Tests\Models\Pagination\Company c JOIN c.logo l ORDER BY l.image_height * l.image_width'; + $this->iterateWithOrderAsc(true, $fetchJoinCollection, $dql, "name"); + $this->iterateWithOrderDesc(true, $fetchJoinCollection, $dql, "name"); + } + + /** + * @dataProvider fetchJoinCollection + */ + public function testIterateWithOutputWalkersWithRegularJoinWithComplexOrderByReferencingJoinedWithLimit($fetchJoinCollection) + { + // long function name is loooooooooooong + + $dql = 'SELECT c FROM Doctrine\Tests\Models\Pagination\Company c JOIN c.logo l ORDER BY l.image_height * l.image_width'; + $this->iterateWithOrderAscWithLimit(true, $fetchJoinCollection, $dql, "name"); + $this->iterateWithOrderDescWithLimit(true, $fetchJoinCollection, $dql, "name"); + } + + /** + * @dataProvider fetchJoinCollection + */ + public function testIterateWithOutputWalkersWithRegularJoinWithComplexOrderByReferencingJoinedWithLimitAndOffset($fetchJoinCollection) + { + // long function name is loooooooooooong + + $dql = 'SELECT c FROM Doctrine\Tests\Models\Pagination\Company c JOIN c.logo l ORDER BY l.image_height * l.image_width'; + $this->iterateWithOrderAscWithLimitAndOffset(true, $fetchJoinCollection, $dql, "name"); + $this->iterateWithOrderDescWithLimitAndOffset(true, $fetchJoinCollection, $dql, "name"); + } + + /** + * @dataProvider useOutputWalkers + */ + public function testIterateWithFetchJoinWithOrderByColumnFromJoined($useOutputWalkers) + { + $dql = 'SELECT u,g,e FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.groups g JOIN u.email e ORDER BY e.email'; + $this->iterateWithOrderAsc($useOutputWalkers, true, $dql, "username"); + $this->iterateWithOrderDesc($useOutputWalkers, true, $dql, "username"); + } + + /** + * @dataProvider useOutputWalkers + */ + public function testIterateWithFetchJoinWithOrderByColumnFromJoinedWithLimit($useOutputWalkers) + { + $dql = 'SELECT u,g,e FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.groups g JOIN u.email e ORDER BY e.email'; + $this->iterateWithOrderAscWithLimit($useOutputWalkers, true, $dql, "username"); + $this->iterateWithOrderDescWithLimit($useOutputWalkers, true, $dql, "username"); + } + + /** + * @dataProvider useOutputWalkers + */ + public function testIterateWithFetchJoinWithOrderByColumnFromJoinedWithLimitAndOffset($useOutputWalkers) + { + $dql = 'SELECT u,g,e FROM Doctrine\Tests\Models\CMS\CmsUser u JOIN u.groups g JOIN u.email e ORDER BY e.email'; + $this->iterateWithOrderAscWithLimitAndOffset($useOutputWalkers, true, $dql, "username"); + $this->iterateWithOrderDescWithLimitAndOffset($useOutputWalkers, true, $dql, "username"); + } + + /** + * @dataProvider fetchJoinCollection + */ + public function testIterateWithOutputWalkersWithFetchJoinWithComplexOrderByReferencingJoined($fetchJoinCollection) + { + $dql = 'SELECT c,l FROM Doctrine\Tests\Models\Pagination\Company c JOIN c.logo l ORDER BY l.image_width * l.image_height'; + $this->iterateWithOrderAsc(true, $fetchJoinCollection, $dql, "name"); + $this->iterateWithOrderDesc(true, $fetchJoinCollection, $dql, "name"); + } + + /** + * @dataProvider fetchJoinCollection + */ + public function testIterateWithOutputWalkersWithFetchJoinWithComplexOrderByReferencingJoinedWithLimit($fetchJoinCollection) + { + $dql = 'SELECT c,l FROM Doctrine\Tests\Models\Pagination\Company c JOIN c.logo l ORDER BY l.image_width * l.image_height'; + $this->iterateWithOrderAscWithLimit(true, $fetchJoinCollection, $dql, "name"); + $this->iterateWithOrderDescWithLimit(true, $fetchJoinCollection, $dql, "name"); + } + + /** + * @dataProvider fetchJoinCollection + */ + public function testIterateWithOutputWalkersWithFetchJoinWithComplexOrderByReferencingJoinedWithLimitAndOffset($fetchJoinCollection) + { + $dql = 'SELECT c,l FROM Doctrine\Tests\Models\Pagination\Company c JOIN c.logo l ORDER BY l.image_width * l.image_height'; + $this->iterateWithOrderAscWithLimitAndOffset(true, $fetchJoinCollection, $dql, "name"); + $this->iterateWithOrderDescWithLimitAndOffset(true, $fetchJoinCollection, $dql, "name"); } public function testIterateComplexWithOutputWalker() { - $dql = "SELECT g, COUNT(u.id) AS userCount FROM Doctrine\Tests\Models\CMS\CmsGroup g LEFT JOIN g.users u GROUP BY g HAVING COUNT(u.id) > 0"; + $dql = 'SELECT g, COUNT(u.id) AS userCount FROM Doctrine\Tests\Models\CMS\CmsGroup g LEFT JOIN g.users u GROUP BY g HAVING COUNT(u.id) > 0'; $query = $this->_em->createQuery($dql); $paginator = new Paginator($query); $paginator->setUseOutputWalkers(true); - $this->assertCount(9, $paginator->getIterator()); + $this->assertCount(3, $paginator->getIterator()); } public function testDetectOutputWalker() { // This query works using the output walkers but causes an exception using the TreeWalker - $dql = "SELECT g, COUNT(u.id) AS userCount FROM Doctrine\Tests\Models\CMS\CmsGroup g LEFT JOIN g.users u GROUP BY g HAVING COUNT(u.id) > 0"; + $dql = 'SELECT g, COUNT(u.id) AS userCount FROM Doctrine\Tests\Models\CMS\CmsGroup g LEFT JOIN g.users u GROUP BY g HAVING COUNT(u.id) > 0'; $query = $this->_em->createQuery($dql); // If the Paginator detects the custom output walker it should fall back to using the @@ -131,7 +470,7 @@ public function testDetectOutputWalker() public function testCloneQuery() { - $dql = "SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u"; + $dql = 'SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u'; $query = $this->_em->createQuery($dql); $paginator = new Paginator($query); @@ -142,7 +481,7 @@ public function testCloneQuery() public function testQueryWalkerIsKept() { - $dql = "SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u"; + $dql = 'SELECT u FROM Doctrine\Tests\Models\CMS\CmsUser u'; $query = $this->_em->createQuery($dql); $query->setHint(Query::HINT_CUSTOM_TREE_WALKERS, array('Doctrine\Tests\ORM\Functional\CustomPaginationTestTreeWalker')); @@ -170,7 +509,7 @@ public function testCountQueryStripsParametersInSelect() $getCountQuery->setAccessible(true); $this->assertCount(2, $getCountQuery->invoke($paginator)->getParameters()); - $this->assertCount(3, $paginator); + $this->assertCount(9, $paginator); $query->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Query\SqlWalker'); @@ -179,25 +518,44 @@ public function testCountQueryStripsParametersInSelect() // if select part of query is replaced with count(...) paginator should remove // parameters from query object not used in new query. $this->assertCount(1, $getCountQuery->invoke($paginator)->getParameters()); - $this->assertCount(3, $paginator); + $this->assertCount(9, $paginator); } public function populate() { - for ($i = 0; $i < 3; $i++) { + $groups = array(); + for ($j = 0; $j < 3; $j++) {; + $group = new CmsGroup(); + $group->name = "group$j"; + $groups[] = $group; + $this->_em->persist($group); + } + + for ($i = 0; $i < 9; $i++) { $user = new CmsUser(); $user->name = "Name$i"; $user->username = "username$i"; $user->status = "active"; + $user->email = new CmsEmail(); + $user->email->user = $user; + $user->email->email = "email$i"; + for ($j = 0; $j < 3; $j++) { + $user->addGroup($groups[$j]); + } $this->_em->persist($user); + } - for ($j = 0; $j < 3; $j++) {; - $group = new CmsGroup(); - $group->name = "group$j"; - $user->addGroup($group); - $this->_em->persist($group); - } + for ($i = 0; $i < 9; $i++) { + $company = new Company(); + $company->name = "name$i"; + $company->logo = new Logo(); + $company->logo->image = "image$i"; + $company->logo->image_width = 100 + $i; + $company->logo->image_height = 100 + $i; + $company->logo->company = $company; + $this->_em->persist($company); } + $this->_em->flush(); } @@ -208,6 +566,24 @@ public function useOutputWalkers() array(false), ); } + + public function fetchJoinCollection() + { + return array( + array(true), + array(false), + ); + } + + public function useOutputWalkersAndFetchJoinCollection() + { + return array( + array(true, false), + array(true, true), + array(false, false), + array(false, true), + ); + } } class CustomPaginationTestTreeWalker extends Query\TreeWalkerAdapter diff --git a/tests/Doctrine/Tests/ORM/Tools/Pagination/LimitSubqueryOutputWalkerTest.php b/tests/Doctrine/Tests/ORM/Tools/Pagination/LimitSubqueryOutputWalkerTest.php index e8d7910b34d..c733c228bed 100644 --- a/tests/Doctrine/Tests/ORM/Tools/Pagination/LimitSubqueryOutputWalkerTest.php +++ b/tests/Doctrine/Tests/ORM/Tools/Pagination/LimitSubqueryOutputWalkerTest.php @@ -5,6 +5,7 @@ use Doctrine\DBAL\Platforms\MySqlPlatform; use Doctrine\DBAL\Platforms\OraclePlatform; use Doctrine\DBAL\Platforms\PostgreSqlPlatform; +use Doctrine\DBAL\Platforms\SQLServerPlatform; use Doctrine\ORM\Query; class LimitSubqueryOutputWalkerTest extends PaginationTestCase @@ -33,7 +34,7 @@ public function testLimitSubqueryWithSortPg() $limitQuery->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); $this->assertEquals( - "SELECT DISTINCT id_0, title_1 FROM (SELECT m0_.id AS id_0, m0_.title AS title_1, c1_.id AS id_2, a2_.id AS id_3, a2_.name AS name_4, m0_.author_id AS author_id_5, m0_.category_id AS category_id_6 FROM MyBlogPost m0_ INNER JOIN Category c1_ ON m0_.category_id = c1_.id INNER JOIN Author a2_ ON m0_.author_id = a2_.id ORDER BY m0_.title ASC) dctrn_result ORDER BY title_1 ASC", $limitQuery->getSql() + "SELECT DISTINCT id_0, title_1 FROM (SELECT m0_.id AS id_0, m0_.title AS title_1, c1_.id AS id_2, a2_.id AS id_3, a2_.name AS name_4, m0_.author_id AS author_id_5, m0_.category_id AS category_id_6 FROM MyBlogPost m0_ INNER JOIN Category c1_ ON m0_.category_id = c1_.id INNER JOIN Author a2_ ON m0_.author_id = a2_.id) dctrn_result ORDER BY title_1 ASC", $limitQuery->getSql() ); $this->entityManager->getConnection()->setDatabasePlatform($odp); @@ -51,7 +52,7 @@ public function testLimitSubqueryWithScalarSortPg() $limitQuery->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); $this->assertEquals( - "SELECT DISTINCT id_1, sclr_0 FROM (SELECT COUNT(g0_.id) AS sclr_0, u1_.id AS id_1, g0_.id AS id_2 FROM User u1_ INNER JOIN user_group u2_ ON u1_.id = u2_.user_id INNER JOIN groups g0_ ON g0_.id = u2_.group_id ORDER BY sclr_0 ASC) dctrn_result ORDER BY sclr_0 ASC", + "SELECT DISTINCT id_1, sclr_0 FROM (SELECT COUNT(g0_.id) AS sclr_0, u1_.id AS id_1, g0_.id AS id_2 FROM User u1_ INNER JOIN user_group u2_ ON u1_.id = u2_.user_id INNER JOIN groups g0_ ON g0_.id = u2_.group_id) dctrn_result ORDER BY sclr_0 ASC", $limitQuery->getSql() ); @@ -70,7 +71,7 @@ public function testLimitSubqueryWithMixedSortPg() $limitQuery->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); $this->assertEquals( - "SELECT DISTINCT id_1, sclr_0 FROM (SELECT COUNT(g0_.id) AS sclr_0, u1_.id AS id_1, g0_.id AS id_2 FROM User u1_ INNER JOIN user_group u2_ ON u1_.id = u2_.user_id INNER JOIN groups g0_ ON g0_.id = u2_.group_id ORDER BY sclr_0 ASC, u1_.id DESC) dctrn_result ORDER BY sclr_0 ASC, id_1 DESC", + "SELECT DISTINCT id_1, sclr_0 FROM (SELECT COUNT(g0_.id) AS sclr_0, u1_.id AS id_1, g0_.id AS id_2 FROM User u1_ INNER JOIN user_group u2_ ON u1_.id = u2_.user_id INNER JOIN groups g0_ ON g0_.id = u2_.group_id) dctrn_result ORDER BY sclr_0 ASC, id_1 DESC", $limitQuery->getSql() ); @@ -89,7 +90,7 @@ public function testLimitSubqueryWithHiddenScalarSortPg() $limitQuery->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); $this->assertEquals( - "SELECT DISTINCT id_1, sclr_0 FROM (SELECT COUNT(g0_.id) AS sclr_0, u1_.id AS id_1, g0_.id AS id_2 FROM User u1_ INNER JOIN user_group u2_ ON u1_.id = u2_.user_id INNER JOIN groups g0_ ON g0_.id = u2_.group_id ORDER BY sclr_0 ASC, u1_.id DESC) dctrn_result ORDER BY sclr_0 ASC, id_1 DESC", + "SELECT DISTINCT id_1, sclr_0 FROM (SELECT COUNT(g0_.id) AS sclr_0, u1_.id AS id_1, g0_.id AS id_2 FROM User u1_ INNER JOIN user_group u2_ ON u1_.id = u2_.user_id INNER JOIN groups g0_ ON g0_.id = u2_.group_id) dctrn_result ORDER BY sclr_0 ASC, id_1 DESC", $limitQuery->getSql() ); @@ -118,7 +119,7 @@ public function testLimitSubqueryWithSortOracle() $limitQuery->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); $this->assertEquals( - "SELECT DISTINCT ID_0, TITLE_1 FROM (SELECT m0_.id AS ID_0, m0_.title AS TITLE_1, c1_.id AS ID_2, a2_.id AS ID_3, a2_.name AS NAME_4, m0_.author_id AS AUTHOR_ID_5, m0_.category_id AS CATEGORY_ID_6 FROM MyBlogPost m0_ INNER JOIN Category c1_ ON m0_.category_id = c1_.id INNER JOIN Author a2_ ON m0_.author_id = a2_.id ORDER BY m0_.title ASC) dctrn_result ORDER BY TITLE_1 ASC", $limitQuery->getSql() + "SELECT DISTINCT ID_0, TITLE_1 FROM (SELECT m0_.id AS ID_0, m0_.title AS TITLE_1, c1_.id AS ID_2, a2_.id AS ID_3, a2_.name AS NAME_4, m0_.author_id AS AUTHOR_ID_5, m0_.category_id AS CATEGORY_ID_6 FROM MyBlogPost m0_ INNER JOIN Category c1_ ON m0_.category_id = c1_.id INNER JOIN Author a2_ ON m0_.author_id = a2_.id) dctrn_result ORDER BY TITLE_1 ASC", $limitQuery->getSql() ); $this->entityManager->getConnection()->setDatabasePlatform($odp); @@ -137,7 +138,7 @@ public function testLimitSubqueryWithScalarSortOracle() $limitQuery->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); $this->assertEquals( - "SELECT DISTINCT ID_1, SCLR_0 FROM (SELECT COUNT(g0_.id) AS SCLR_0, u1_.id AS ID_1, g0_.id AS ID_2 FROM User u1_ INNER JOIN user_group u2_ ON u1_.id = u2_.user_id INNER JOIN groups g0_ ON g0_.id = u2_.group_id ORDER BY SCLR_0 ASC) dctrn_result ORDER BY SCLR_0 ASC", + "SELECT DISTINCT ID_1, SCLR_0 FROM (SELECT COUNT(g0_.id) AS SCLR_0, u1_.id AS ID_1, g0_.id AS ID_2 FROM User u1_ INNER JOIN user_group u2_ ON u1_.id = u2_.user_id INNER JOIN groups g0_ ON g0_.id = u2_.group_id) dctrn_result ORDER BY SCLR_0 ASC", $limitQuery->getSql() ); @@ -157,7 +158,7 @@ public function testLimitSubqueryWithMixedSortOracle() $limitQuery->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); $this->assertEquals( - "SELECT DISTINCT ID_1, SCLR_0 FROM (SELECT COUNT(g0_.id) AS SCLR_0, u1_.id AS ID_1, g0_.id AS ID_2 FROM User u1_ INNER JOIN user_group u2_ ON u1_.id = u2_.user_id INNER JOIN groups g0_ ON g0_.id = u2_.group_id ORDER BY SCLR_0 ASC, u1_.id DESC) dctrn_result ORDER BY SCLR_0 ASC, ID_1 DESC", + "SELECT DISTINCT ID_1, SCLR_0 FROM (SELECT COUNT(g0_.id) AS SCLR_0, u1_.id AS ID_1, g0_.id AS ID_2 FROM User u1_ INNER JOIN user_group u2_ ON u1_.id = u2_.user_id INNER JOIN groups g0_ ON g0_.id = u2_.group_id) dctrn_result ORDER BY SCLR_0 ASC, ID_1 DESC", $limitQuery->getSql() ); @@ -207,7 +208,67 @@ public function testCountQueryWithArithmeticOrderByCondition() $query->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); $this->assertSame( - 'SELECT DISTINCT id_0 FROM (SELECT a0_.id AS id_0, a0_.name AS name_1 FROM Author a0_ ORDER BY (1 - 1000) * 1 DESC) dctrn_result', + 'SELECT DISTINCT id_0, (1 - 1000) * 1 AS ordr_0 FROM (SELECT a0_.id AS id_0, a0_.name AS name_1 FROM Author a0_) dctrn_result ORDER BY ordr_0 DESC', + $query->getSQL() + ); + } + + public function testCountQueryWithComplexScalarOrderByItem() + { + $query = $this->entityManager->createQuery( + 'SELECT a FROM Doctrine\Tests\ORM\Tools\Pagination\Avatar a ORDER BY a.image_height * a.image_width DESC' + ); + $this->entityManager->getConnection()->setDatabasePlatform(new MySqlPlatform()); + + $query->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); + + $this->assertSame( + 'SELECT DISTINCT id_0, image_height_2 * image_width_3 AS ordr_0 FROM (SELECT a0_.id AS id_0, a0_.image AS image_1, a0_.image_height AS image_height_2, a0_.image_width AS image_width_3, a0_.image_alt_desc AS image_alt_desc_4, a0_.user_id AS user_id_5 FROM Avatar a0_) dctrn_result ORDER BY ordr_0 DESC', + $query->getSQL() + ); + } + + public function testCountQueryWithComplexScalarOrderByItemJoined() + { + $query = $this->entityManager->createQuery( + 'SELECT u FROM Doctrine\Tests\ORM\Tools\Pagination\User u JOIN u.avatar a ORDER BY a.image_height * a.image_width DESC' + ); + $this->entityManager->getConnection()->setDatabasePlatform(new MySqlPlatform()); + + $query->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); + + $this->assertSame( + 'SELECT DISTINCT id_0, image_height_1 * image_width_2 AS ordr_0 FROM (SELECT u0_.id AS id_0, a1_.image_height AS image_height_1, a1_.image_width AS image_width_2, a1_.user_id AS user_id_3 FROM User u0_ INNER JOIN Avatar a1_ ON u0_.id = a1_.user_id) dctrn_result ORDER BY ordr_0 DESC', + $query->getSQL() + ); + } + + public function testCountQueryWithComplexScalarOrderByItemJoinedWithPartial() + { + $query = $this->entityManager->createQuery( + 'SELECT u, partial a.{id, image_alt_desc} FROM Doctrine\Tests\ORM\Tools\Pagination\User u JOIN u.avatar a ORDER BY a.image_height * a.image_width DESC' + ); + $this->entityManager->getConnection()->setDatabasePlatform(new MySqlPlatform()); + + $query->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); + + $this->assertSame( + 'SELECT DISTINCT id_0, image_height_3 * image_width_4 AS ordr_0 FROM (SELECT u0_.id AS id_0, a1_.id AS id_1, a1_.image_alt_desc AS image_alt_desc_2, a1_.image_height AS image_height_3, a1_.image_width AS image_width_4, a1_.user_id AS user_id_5 FROM User u0_ INNER JOIN Avatar a1_ ON u0_.id = a1_.user_id) dctrn_result ORDER BY ordr_0 DESC', + $query->getSQL() + ); + } + + public function testCountQueryWithComplexScalarOrderByItemOracle() + { + $query = $this->entityManager->createQuery( + 'SELECT a FROM Doctrine\Tests\ORM\Tools\Pagination\Avatar a ORDER BY a.image_height * a.image_width DESC' + ); + $this->entityManager->getConnection()->setDatabasePlatform(new OraclePlatform()); + + $query->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); + + $this->assertSame( + 'SELECT DISTINCT ID_0, IMAGE_HEIGHT_2 * IMAGE_WIDTH_3 AS ordr_0 FROM (SELECT a0_.id AS ID_0, a0_.image AS IMAGE_1, a0_.image_height AS IMAGE_HEIGHT_2, a0_.image_width AS IMAGE_WIDTH_3, a0_.image_alt_desc AS IMAGE_ALT_DESC_4, a0_.user_id AS USER_ID_5 FROM Avatar a0_) dctrn_result ORDER BY ordr_0 DESC', $query->getSQL() ); } @@ -224,11 +285,26 @@ public function testLimitSubqueryWithHiddenSelectionInOrderBy() $query->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); $this->assertEquals( - 'SELECT DISTINCT id_0, name_2 FROM (SELECT a0_.id AS id_0, a0_.name AS name_1, a0_.name AS name_2 FROM Author a0_ ORDER BY name_2 DESC) dctrn_result ORDER BY name_2 DESC', + 'SELECT DISTINCT id_0, name_2 FROM (SELECT a0_.id AS id_0, a0_.name AS name_1, a0_.name AS name_2 FROM Author a0_) dctrn_result ORDER BY name_2 DESC', $query->getSql() ); } + public function testLimitSubqueryWithColumnWithSortDirectionInName() + { + $query = $this->entityManager->createQuery( + 'SELECT a FROM Doctrine\Tests\ORM\Tools\Pagination\Avatar a ORDER BY a.image_alt_desc DESC' + ); + $this->entityManager->getConnection()->setDatabasePlatform(new MySqlPlatform()); + + $query->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); + + $this->assertSame( + 'SELECT DISTINCT id_0, image_alt_desc_4 FROM (SELECT a0_.id AS id_0, a0_.image AS image_1, a0_.image_height AS image_height_2, a0_.image_width AS image_width_3, a0_.image_alt_desc AS image_alt_desc_4, a0_.user_id AS user_id_5 FROM Avatar a0_) dctrn_result ORDER BY image_alt_desc_4 DESC', + $query->getSQL() + ); + } + public function testLimitSubqueryWithOrderByInnerJoined() { $query = $this->entityManager->createQuery( @@ -238,7 +314,7 @@ public function testLimitSubqueryWithOrderByInnerJoined() $query->setHint(Query::HINT_CUSTOM_OUTPUT_WALKER, 'Doctrine\ORM\Tools\Pagination\LimitSubqueryOutputWalker'); $this->assertEquals( - 'SELECT DISTINCT id_0 FROM (SELECT b0_.id AS id_0, b0_.author_id AS author_id_1, b0_.category_id AS category_id_2 FROM BlogPost b0_ INNER JOIN Author a1_ ON b0_.author_id = a1_.id ORDER BY a1_.name ASC) dctrn_result', + 'SELECT DISTINCT id_0, name_1 FROM (SELECT b0_.id AS id_0, a1_.name AS name_1, b0_.author_id AS author_id_2, b0_.category_id AS category_id_3 FROM BlogPost b0_ INNER JOIN Author a1_ ON b0_.author_id = a1_.id) dctrn_result ORDER BY name_1 ASC', $query->getSQL() ); } diff --git a/tests/Doctrine/Tests/ORM/Tools/Pagination/PaginationTestCase.php b/tests/Doctrine/Tests/ORM/Tools/Pagination/PaginationTestCase.php index d3e77f6f06c..f1553c13c7b 100644 --- a/tests/Doctrine/Tests/ORM/Tools/Pagination/PaginationTestCase.php +++ b/tests/Doctrine/Tests/ORM/Tools/Pagination/PaginationTestCase.php @@ -24,7 +24,7 @@ public function setUp() class MyBlogPost { - /** @Id @column(type="integer") @generatedValue */ + /** @Id @Column(type="integer") @GeneratedValue */ public $id; /** * @ManyToOne(targetEntity="Author") @@ -34,7 +34,7 @@ class MyBlogPost * @ManyToOne(targetEntity="Category") */ public $category; - /** @column(type="string") */ + /** @Column(type="string") */ public $title; } @@ -44,7 +44,7 @@ class MyBlogPost class MyAuthor { - /** @Id @column(type="integer") @generatedValue */ + /** @Id @Column(type="integer") @GeneratedValue */ public $id; } @@ -55,7 +55,7 @@ class MyAuthor class MyCategory { - /** @id @column(type="integer") @generatedValue */ + /** @Id @Column(type="integer") @GeneratedValue */ public $id; } @@ -67,7 +67,7 @@ class MyCategory class BlogPost { - /** @Id @column(type="integer") @generatedValue */ + /** @Id @Column(type="integer") @GeneratedValue */ public $id; /** * @ManyToOne(targetEntity="Author") @@ -85,7 +85,7 @@ class BlogPost class Author { - /** @Id @column(type="integer") @generatedValue */ + /** @Id @Column(type="integer") @GeneratedValue */ public $id; /** @Column(type="string") */ public $name; @@ -98,7 +98,7 @@ class Author class Person { - /** @Id @column(type="integer") @generatedValue */ + /** @Id @Column(type="integer") @GeneratedValue */ public $id; /** @Column(type="string") */ public $name; @@ -113,7 +113,7 @@ class Person class Category { - /** @id @column(type="integer") @generatedValue */ + /** @Id @Column(type="integer") @GeneratedValue */ public $id; } @@ -123,7 +123,7 @@ class Category class Group { - /** @Id @column(type="integer") @generatedValue */ + /** @Id @Column(type="integer") @GeneratedValue */ public $id; /** @ManyToMany(targetEntity="User", mappedBy="groups") */ public $users; @@ -133,7 +133,7 @@ class Group class User { - /** @Id @column(type="integer") @generatedValue */ + /** @Id @Column(type="integer") @GeneratedValue */ public $id; /** * @ManyToMany(targetEntity="Group", inversedBy="users") @@ -144,4 +144,28 @@ class User * ) */ public $groups; + /** + * @OneToOne(targetEntity="Avatar", mappedBy="user") + */ + public $avatar; } + +/** @Entity */ +class Avatar +{ + /** @Id @Column(type="integer") @GeneratedValue */ + public $id; + /** + * @OneToOne(targetEntity="User", inversedBy="avatar") + * @JoinColumn(name="user_id", referencedColumnName="id") + */ + public $user; + /** @Column(type="string", length=255) */ + public $image; + /** @Column(type="integer") */ + public $image_height; + /** @Column(type="integer") */ + public $image_width; + /** @Column(type="string", length=255) */ + public $image_alt_desc; +} \ No newline at end of file diff --git a/tests/Doctrine/Tests/OrmFunctionalTestCase.php b/tests/Doctrine/Tests/OrmFunctionalTestCase.php index 6feb8ab9733..c03a4d04ab1 100644 --- a/tests/Doctrine/Tests/OrmFunctionalTestCase.php +++ b/tests/Doctrine/Tests/OrmFunctionalTestCase.php @@ -267,6 +267,10 @@ abstract class OrmFunctionalTestCase extends OrmTestCase 'Doctrine\Tests\Models\CustomType\CustomIdObjectTypeParent', 'Doctrine\Tests\Models\CustomType\CustomIdObjectTypeChild', ), + 'pagination' => array( + 'Doctrine\Tests\Models\Pagination\Company', + 'Doctrine\Tests\Models\Pagination\Logo', + ), ); /** @@ -515,6 +519,11 @@ protected function tearDown() $conn->executeUpdate('DELETE FROM custom_id_type_parent'); } + if (isset($this->_usedModelSets['pagination'])) { + $conn->executeUpdate('DELETE FROM pagination_logo'); + $conn->executeUpdate('DELETE FROM pagination_company'); + } + $this->_em->clear(); }