vendor/doctrine/orm/lib/Doctrine/ORM/Query/AST/Functions/SizeFunction.php

   1 <?php
   2 /*
   3  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
   4  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
   5  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
   6  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
   7  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
   8  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
   9  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  10  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  11  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  12  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  13  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  14  *
  15  * This software consists of voluntary contributions made by many individuals
  16  * and is licensed under the MIT license. For more information, see
  17  * <http://www.doctrine-project.org>.
  18  */
  19
  20 namespace Doctrine\ORM\Query\AST\Functions;
  21
  22 use Doctrine\ORM\Query\Lexer;
  23
  24 /**
  25  * "SIZE" "(" CollectionValuedPathExpression ")"
  26  *
  27  *
  28  * @link    www.doctrine-project.org
  29  * @since   2.0
  30  * @author  Guilherme Blanco <guilhermeblanco@hotmail.com>
  31  * @author  Jonathan Wage <jonwage@gmail.com>
  32  * @author  Roman Borschel <roman@code-factory.org>
  33  * @author  Benjamin Eberlei <kontakt@beberlei.de>
  34  */
  35 class SizeFunction extends FunctionNode
  36 {
  37     public $collectionPathExpression;
  38
  39     /**
  40      * @override
  41      * @todo If the collection being counted is already joined, the SQL can be simpler (more efficient).
  42      */
  43     public function getSql(\Doctrine\ORM\Query\SqlWalker $sqlWalker)
  44     {
  45         $platform       = $sqlWalker->getEntityManager()->getConnection()->getDatabasePlatform();
  46         $quoteStrategy  = $sqlWalker->getEntityManager()->getConfiguration()->getQuoteStrategy();
  47         $dqlAlias       = $this->collectionPathExpression->identificationVariable;
  48         $assocField     = $this->collectionPathExpression->field;
  49
  50         $qComp  = $sqlWalker->getQueryComponent($dqlAlias);
  51         $class  = $qComp['metadata'];
  52         $assoc  = $class->associationMappings[$assocField];
  53         $sql    = 'SELECT COUNT(*) FROM ';
  54
  55         if ($assoc['type'] == \Doctrine\ORM\Mapping\ClassMetadata::ONE_TO_MANY) {
  56             $targetClass        = $sqlWalker->getEntityManager()->getClassMetadata($assoc['targetEntity']);
  57             $targetTableAlias   = $sqlWalker->getSQLTableAlias($targetClass->getTableName());
  58             $sourceTableAlias   = $sqlWalker->getSQLTableAlias($class->getTableName(), $dqlAlias);
  59
  60             $sql .= $quoteStrategy->getTableName($targetClass, $platform) . ' ' . $targetTableAlias . ' WHERE ';
  61
  62             $owningAssoc = $targetClass->associationMappings[$assoc['mappedBy']];
  63
  64             $first = true;
  65
  66             foreach ($owningAssoc['targetToSourceKeyColumns'] as $targetColumn => $sourceColumn) {
  67                 if ($first) $first = false; else $sql .= ' AND ';
  68
  69                 $sql .= $targetTableAlias . '.' . $sourceColumn
  70                       . ' = '
  71                       . $sourceTableAlias . '.' . $quoteStrategy->getColumnName($class->fieldNames[$targetColumn], $class, $platform);
  72             }
  73         } else { // many-to-many
  74             $targetClass = $sqlWalker->getEntityManager()->getClassMetadata($assoc['targetEntity']);
  75
  76             $owningAssoc = $assoc['isOwningSide'] ? $assoc : $targetClass->associationMappings[$assoc['mappedBy']];
  77             $joinTable = $owningAssoc['joinTable'];
  78
  79             // SQL table aliases
  80             $joinTableAlias = $sqlWalker->getSQLTableAlias($joinTable['name']);
  81             $sourceTableAlias = $sqlWalker->getSQLTableAlias($class->getTableName(), $dqlAlias);
  82
  83             // join to target table
  84             $sql .= $quoteStrategy->getJoinTableName($owningAssoc, $targetClass, $platform) . ' ' . $joinTableAlias . ' WHERE ';
  85
  86             $joinColumns = $assoc['isOwningSide']
  87                 ? $joinTable['joinColumns']
  88                 : $joinTable['inverseJoinColumns'];
  89
  90             $first = true;
  91
  92             foreach ($joinColumns as $joinColumn) {
  93                 if ($first) $first = false; else $sql .= ' AND ';
  94
  95                 $sourceColumnName = $quoteStrategy->getColumnName(
  96                     $class->fieldNames[$joinColumn['referencedColumnName']], $class, $platform
  97                 );
  98
  99                 $sql .= $joinTableAlias . '.' . $joinColumn['name']
 100                       . ' = '
 101                       . $sourceTableAlias . '.' . $sourceColumnName;
 102             }
 103         }
 104
 105         return '(' . $sql . ')';
 106     }
 107
 108     /**
 109      * @override
 110      */
 111     public function parse(\Doctrine\ORM\Query\Parser $parser)
 112     {
 113         $parser->match(Lexer::T_IDENTIFIER);
 114         $parser->match(Lexer::T_OPEN_PARENTHESIS);
 115
 116         $this->collectionPathExpression = $parser->CollectionValuedPathExpression();
 117
 118         $parser->match(Lexer::T_CLOSE_PARENTHESIS);
 119     }
 120 }
 121