/**
  * {@inheritdoc}
  */
 public function getSql(SqlWalker $sqlWalker)
 {
     $platform = $sqlWalker->getEntityManager()->getConnection()->getDatabasePlatform();
     $quoteStrategy = $sqlWalker->getEntityManager()->getConfiguration()->getQuoteStrategy();
     $dqlAlias = $this->pathExpression->identificationVariable;
     $assocField = $this->pathExpression->field;
     $qComp = $sqlWalker->getQueryComponent($dqlAlias);
     $class = $qComp['metadata'];
     $assoc = $class->associationMappings[$assocField];
     $targetEntity = $sqlWalker->getEntityManager()->getClassMetadata($assoc['targetEntity']);
     $joinColumn = reset($assoc['joinColumns']);
     if ($this->fieldMapping !== null) {
         if (!isset($targetEntity->fieldMappings[$this->fieldMapping])) {
             throw new QueryException(sprintf('Undefined reference field mapping "%s"', $this->fieldMapping));
         }
         $field = $targetEntity->fieldMappings[$this->fieldMapping];
         $joinColumn = null;
         foreach ($assoc['joinColumns'] as $mapping) {
             if ($mapping['referencedColumnName'] === $field['columnName']) {
                 $joinColumn = $mapping;
                 break;
             }
         }
         if ($joinColumn === null) {
             throw new QueryException(sprintf('Unable to resolve the reference field mapping "%s"', $this->fieldMapping));
         }
     }
     // The table with the relation may be a subclass, so get the table name from the association definition
     $tableName = $sqlWalker->getEntityManager()->getClassMetadata($assoc['sourceEntity'])->getTableName();
     $tableAlias = $sqlWalker->getSQLTableAlias($tableName, $dqlAlias);
     $columnName = $quoteStrategy->getJoinColumnName($joinColumn, $targetEntity, $platform);
     return $tableAlias . '.' . $columnName;
 }
 /**
  * @override
  * @todo If the collection being counted is already joined, the SQL can be simpler (more efficient).
  */
 public function getSql(\Doctrine\ORM\Query\SqlWalker $sqlWalker)
 {
     $platform = $sqlWalker->getEntityManager()->getConnection()->getDatabasePlatform();
     $quoteStrategy = $sqlWalker->getEntityManager()->getConfiguration()->getQuoteStrategy();
     $dqlAlias = $this->collectionPathExpression->identificationVariable;
     $assocField = $this->collectionPathExpression->field;
     $qComp = $sqlWalker->getQueryComponent($dqlAlias);
     $class = $qComp['metadata'];
     $assoc = $class->associationMappings[$assocField];
     $sql = 'SELECT COUNT(*) FROM ';
     if ($assoc['type'] == \Doctrine\ORM\Mapping\ClassMetadata::ONE_TO_MANY) {
         $targetClass = $sqlWalker->getEntityManager()->getClassMetadata($assoc['targetEntity']);
         $targetTableAlias = $sqlWalker->getSQLTableAlias($targetClass->getTableName());
         $sourceTableAlias = $sqlWalker->getSQLTableAlias($class->getTableName(), $dqlAlias);
         $sql .= $quoteStrategy->getTableName($targetClass, $platform) . ' ' . $targetTableAlias . ' WHERE ';
         $owningAssoc = $targetClass->associationMappings[$assoc['mappedBy']];
         $first = true;
         foreach ($owningAssoc['targetToSourceKeyColumns'] as $targetColumn => $sourceColumn) {
             if ($first) {
                 $first = false;
             } else {
                 $sql .= ' AND ';
             }
             $sql .= $targetTableAlias . '.' . $sourceColumn . ' = ' . $sourceTableAlias . '.' . $quoteStrategy->getColumnName($class->fieldNames[$targetColumn], $class, $platform);
         }
     } else {
         // many-to-many
         $targetClass = $sqlWalker->getEntityManager()->getClassMetadata($assoc['targetEntity']);
         $owningAssoc = $assoc['isOwningSide'] ? $assoc : $targetClass->associationMappings[$assoc['mappedBy']];
         $joinTable = $owningAssoc['joinTable'];
         // SQL table aliases
         $joinTableAlias = $sqlWalker->getSQLTableAlias($joinTable['name']);
         $sourceTableAlias = $sqlWalker->getSQLTableAlias($class->getTableName(), $dqlAlias);
         // join to target table
         $sql .= $quoteStrategy->getJoinTableName($owningAssoc, $targetClass, $platform) . ' ' . $joinTableAlias . ' WHERE ';
         $joinColumns = $assoc['isOwningSide'] ? $joinTable['joinColumns'] : $joinTable['inverseJoinColumns'];
         $first = true;
         foreach ($joinColumns as $joinColumn) {
             if ($first) {
                 $first = false;
             } else {
                 $sql .= ' AND ';
             }
             $sourceColumnName = $quoteStrategy->getColumnName($class->fieldNames[$joinColumn['referencedColumnName']], $class, $platform);
             $sql .= $joinTableAlias . '.' . $joinColumn['name'] . ' = ' . $sourceTableAlias . '.' . $sourceColumnName;
         }
     }
     return '(' . $sql . ')';
 }
Пример #3
0
 /**
  * @override
  */
 public function getSql(\Doctrine\ORM\Query\SqlWalker $sqlWalker)
 {
     $dqlAlias = $this->collectionPathExpression->identificationVariable;
     $parts = $this->collectionPathExpression->parts;
     $assocField = array_pop($parts);
     $qComp = $sqlWalker->getQueryComponent(implode('.', array_merge((array) $dqlAlias, $parts)));
     $assoc = $qComp['metadata']->associationMappings[$assocField];
     $sql = '';
     if ($assoc->isOneToMany()) {
         $targetClass = $sqlWalker->getEntityManager()->getClassMetadata($assoc->targetEntityName);
         $targetAssoc = $targetClass->associationMappings[$assoc->mappedByFieldName];
         $targetTableAlias = $sqlWalker->getSqlTableAlias($targetClass->primaryTable['name']);
         $sourceTableAlias = $sqlWalker->getSqlTableAlias($qComp['metadata']->primaryTable['name'], $dqlAlias);
         $whereSql = '';
         foreach ($targetAssoc->targetToSourceKeyColumns as $targetKeyColumn => $sourceKeyColumn) {
             $whereSql .= ($whereSql == '' ? ' WHERE ' : ' AND ') . $targetTableAlias . '.' . $sourceKeyColumn . ' = ' . $sourceTableAlias . '.' . $targetKeyColumn;
         }
         $sql = '(SELECT COUNT(' . "{$targetTableAlias}." . implode(", {$targetTableAlias}.", $targetAssoc->targetToSourceKeyColumns) . ') FROM ' . $targetClass->primaryTable['name'] . ' ' . $targetTableAlias . $whereSql . ')';
     } else {
         if ($assoc->isManyToMany()) {
             // TODO
             $targetTableAlias = $sqlWalker->getSqlTableAlias($assoc->joinTable['name']);
             $sourceTableAlias = $sqlWalker->getSqlTableAlias($qComp['metadata']->primaryTable['name'], $dqlAlias);
             $whereSql = '';
             foreach ($assoc->relationToSourceKeyColumns as $targetKeyColumn => $sourceKeyColumn) {
                 $whereSql .= ($whereSql == '' ? ' WHERE ' : ' AND ') . $targetTableAlias . '.' . $targetKeyColumn . ' = ' . $sourceTableAlias . '.' . $sourceKeyColumn;
             }
             $sql = '(SELECT COUNT(' . "{$targetTableAlias}." . implode(", {$targetTableAlias}.", $assoc->joinTableColumns) . ') FROM ' . $assoc->joinTable['name'] . ' ' . $targetTableAlias . $whereSql . ')';
         }
     }
     return $sql;
 }
Пример #4
0
 /**
  * Initializes a new <tt>MultiTableDeleteExecutor</tt>.
  *
  * @param \Doctrine\ORM\Query\AST\Node  $AST       The root AST node of the DQL query.
  * @param \Doctrine\ORM\Query\SqlWalker $sqlWalker The walker used for SQL generation from the AST.
  *
  * @internal Any SQL construction and preparation takes place in the constructor for
  *           best performance. With a query cache the executor will be cached.
  */
 public function __construct(AST\Node $AST, $sqlWalker)
 {
     $em = $sqlWalker->getEntityManager();
     $conn = $em->getConnection();
     $platform = $conn->getDatabasePlatform();
     $quoteStrategy = $em->getConfiguration()->getQuoteStrategy();
     $primaryClass = $em->getClassMetadata($AST->deleteClause->abstractSchemaName);
     $primaryDqlAlias = $AST->deleteClause->aliasIdentificationVariable;
     $rootClass = $em->getClassMetadata($primaryClass->rootEntityName);
     $tempTable = $platform->getTemporaryTableName($rootClass->getTemporaryIdTableName());
     $idColumnNames = $rootClass->getIdentifierColumnNames();
     $idColumnList = implode(', ', $idColumnNames);
     // 1. Create an INSERT INTO temptable ... SELECT identifiers WHERE $AST->getWhereClause()
     $sqlWalker->setSQLTableAlias($primaryClass->getTableName(), 't0', $primaryDqlAlias);
     $this->_insertSql = 'INSERT INTO ' . $tempTable . ' (' . $idColumnList . ')' . ' SELECT t0.' . implode(', t0.', $idColumnNames);
     $rangeDecl = new AST\RangeVariableDeclaration($primaryClass->name, $primaryDqlAlias);
     $fromClause = new AST\FromClause(array(new AST\IdentificationVariableDeclaration($rangeDecl, null, array())));
     $this->_insertSql .= $sqlWalker->walkFromClause($fromClause);
     // Append WHERE clause, if there is one.
     if ($AST->whereClause) {
         $this->_insertSql .= $sqlWalker->walkWhereClause($AST->whereClause);
     }
     // 2. Create ID subselect statement used in DELETE ... WHERE ... IN (subselect)
     $idSubselect = 'SELECT ' . $idColumnList . ' FROM ' . $tempTable;
     // 3. Create and store DELETE statements
     $classNames = array_merge($primaryClass->parentClasses, array($primaryClass->name), $primaryClass->subClasses);
     foreach (array_reverse($classNames) as $className) {
         $tableName = $quoteStrategy->getTableName($em->getClassMetadata($className), $platform);
         $this->_sqlStatements[] = 'DELETE FROM ' . $tableName . ' WHERE (' . $idColumnList . ') IN (' . $idSubselect . ')';
     }
     // 4. Store DDL for temporary identifier table.
     $columnDefinitions = array();
     foreach ($idColumnNames as $idColumnName) {
         $columnDefinitions[$idColumnName] = array('notnull' => true, 'type' => \Doctrine\DBAL\Types\Type::getType($rootClass->getTypeOfColumn($idColumnName)));
     }
     $this->_createTempTableSql = $platform->getCreateTemporaryTableSnippetSQL() . ' ' . $tempTable . ' (' . $platform->getColumnDeclarationListSQL($columnDefinitions) . ')';
     $this->_dropTempTableSql = $platform->getDropTemporaryTableSQL($tempTable);
 }
 /**
  * Initializes a new <tt>MultiTableUpdateExecutor</tt>.
  *
  * Internal note: Any SQL construction and preparation takes place in the constructor for
  *                best performance. With a query cache the executor will be cached.
  *
  * @param \Doctrine\ORM\Query\AST\Node  $AST The root AST node of the DQL query.
  * @param \Doctrine\ORM\Query\SqlWalker $sqlWalker The walker used for SQL generation from the AST.
  */
 public function __construct(AST\Node $AST, $sqlWalker)
 {
     $em = $sqlWalker->getEntityManager();
     $conn = $em->getConnection();
     $platform = $conn->getDatabasePlatform();
     $quoteStrategy = $em->getConfiguration()->getQuoteStrategy();
     $updateClause = $AST->updateClause;
     $primaryClass = $sqlWalker->getEntityManager()->getClassMetadata($updateClause->abstractSchemaName);
     $rootClass = $em->getClassMetadata($primaryClass->rootEntityName);
     $updateItems = $updateClause->updateItems;
     $tempTable = $platform->getTemporaryTableName($rootClass->getTemporaryIdTableName());
     $idColumnNames = $rootClass->getIdentifierColumnNames();
     $idColumnList = implode(', ', $idColumnNames);
     // 1. Create an INSERT INTO temptable ... SELECT identifiers WHERE $AST->getWhereClause()
     $sqlWalker->setSQLTableAlias($primaryClass->getTableName(), 't0', $updateClause->aliasIdentificationVariable);
     $this->_insertSql = 'INSERT INTO ' . $tempTable . ' (' . $idColumnList . ')' . ' SELECT t0.' . implode(', t0.', $idColumnNames);
     $rangeDecl = new AST\RangeVariableDeclaration($primaryClass->name, $updateClause->aliasIdentificationVariable);
     $fromClause = new AST\FromClause([new AST\IdentificationVariableDeclaration($rangeDecl, null, [])]);
     $this->_insertSql .= $sqlWalker->walkFromClause($fromClause);
     // 2. Create ID subselect statement used in UPDATE ... WHERE ... IN (subselect)
     $idSubselect = 'SELECT ' . $idColumnList . ' FROM ' . $tempTable;
     // 3. Create and store UPDATE statements
     $classNames = array_merge($primaryClass->parentClasses, [$primaryClass->name], $primaryClass->subClasses);
     $i = -1;
     foreach (array_reverse($classNames) as $className) {
         $affected = false;
         $class = $em->getClassMetadata($className);
         $updateSql = 'UPDATE ' . $quoteStrategy->getTableName($class, $platform) . ' SET ';
         foreach ($updateItems as $updateItem) {
             $field = $updateItem->pathExpression->field;
             if (isset($class->fieldMappings[$field]) && !isset($class->fieldMappings[$field]['inherited']) || isset($class->associationMappings[$field]) && !isset($class->associationMappings[$field]['inherited'])) {
                 $newValue = $updateItem->newValue;
                 if (!$affected) {
                     $affected = true;
                     ++$i;
                 } else {
                     $updateSql .= ', ';
                 }
                 $updateSql .= $sqlWalker->walkUpdateItem($updateItem);
                 if ($newValue instanceof AST\InputParameter) {
                     $this->_sqlParameters[$i][] = $newValue->name;
                     ++$this->_numParametersInUpdateClause;
                 }
             }
         }
         if ($affected) {
             $this->_sqlStatements[$i] = $updateSql . ' WHERE (' . $idColumnList . ') IN (' . $idSubselect . ')';
         }
     }
     // Append WHERE clause to insertSql, if there is one.
     if ($AST->whereClause) {
         $this->_insertSql .= $sqlWalker->walkWhereClause($AST->whereClause);
     }
     // 4. Store DDL for temporary identifier table.
     $columnDefinitions = [];
     foreach ($idColumnNames as $idColumnName) {
         $columnDefinitions[$idColumnName] = ['notnull' => true, 'type' => Type::getType(PersisterHelper::getTypeOfColumn($idColumnName, $rootClass, $em))];
     }
     $this->_createTempTableSql = $platform->getCreateTemporaryTableSnippetSQL() . ' ' . $tempTable . ' (' . $platform->getColumnDeclarationListSQL($columnDefinitions) . ')';
     $this->_dropTempTableSql = $platform->getDropTemporaryTableSQL($tempTable);
 }