Ejemplo n.º 1
0
 /**
  * Initializes a new <tt>MultiTableDeleteExecutor</tt>.
  *
  * @param Node $AST The root AST node of the DQL query.
  * @param 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();
     $primaryClass = $em->getClassMetadata($AST->deleteClause->abstractSchemaName);
     $primaryDqlAlias = $AST->deleteClause->aliasIdentificationVariable;
     $rootClass = $em->getClassMetadata($primaryClass->rootEntityName);
     $tempTable = $rootClass->getTemporaryIdTableName();
     $idColumnNames = $rootClass->getIdentifierColumnNames();
     $idColumnList = implode(', ', $idColumnNames);
     // 1. Create an INSERT INTO temptable ... SELECT identifiers WHERE $AST->getWhereClause()
     $this->_insertSql = 'INSERT INTO ' . $tempTable . ' (' . $idColumnList . ')' . ' SELECT t0.' . implode(', t0.', $idColumnNames);
     $sqlWalker->setSqlTableAlias($primaryClass->primaryTable['name'] . $primaryDqlAlias, 't0');
     $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 = $em->getClassMetadata($className)->getQuotedTableName($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) . ', PRIMARY KEY(' . $idColumnList . '))';
     $this->_dropTempTableSql = 'DROP TABLE ' . $tempTable;
 }
Ejemplo n.º 2
0
 /**
  * Initializes a new <tt>MultiTableUpdateExecutor</tt>.
  *
  * @param Node $AST The root AST node of the DQL query.
  * @param 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();
     $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()
     $this->_insertSql = 'INSERT INTO ' . $tempTable . ' (' . $idColumnList . ')' . ' SELECT t0.' . implode(', t0.', $idColumnNames);
     $sqlWalker->setSqlTableAlias($primaryClass->table['name'] . $updateClause->aliasIdentificationVariable, 't0');
     $rangeDecl = new AST\RangeVariableDeclaration($primaryClass->name, $updateClause->aliasIdentificationVariable);
     $fromClause = new AST\FromClause(array(new AST\IdentificationVariableDeclaration($rangeDecl, null, array())));
     $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, array($primaryClass->name), $primaryClass->subClasses);
     $i = -1;
     foreach (array_reverse($classNames) as $className) {
         $affected = false;
         $class = $em->getClassMetadata($className);
         $updateSql = 'UPDATE ' . $class->getQuotedTableName($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);
                 //FIXME: parameters can be more deeply nested. traverse the tree.
                 //FIXME (URGENT): With query cache the parameter is out of date. Move to execute() stage.
                 if ($newValue instanceof AST\InputParameter) {
                     $paramKey = $newValue->name;
                     $this->_sqlParameters[$i][] = $sqlWalker->getQuery()->getParameter($paramKey);
                     ++$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 = array();
     foreach ($idColumnNames as $idColumnName) {
         $columnDefinitions[$idColumnName] = array('notnull' => true, 'type' => Type::getType($rootClass->getTypeOfColumn($idColumnName)));
     }
     $this->_createTempTableSql = $platform->getCreateTemporaryTableSnippetSQL() . ' ' . $tempTable . ' (' . $platform->getColumnDeclarationListSQL($columnDefinitions) . ')';
     $this->_dropTempTableSql = 'DROP TABLE ' . $tempTable;
 }