123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941 |
- <?php
- /*
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *
- * This software consists of voluntary contributions made by many individuals
- * and is licensed under the MIT license. For more information, see
- * <http://www.doctrine-project.org>.
- */
- namespace Doctrine\ORM\Tools;
- use Doctrine\DBAL\Platforms\AbstractPlatform;
- use Doctrine\DBAL\Schema\AbstractAsset;
- use Doctrine\DBAL\Schema\Comparator;
- use Doctrine\DBAL\Schema\Index;
- use Doctrine\DBAL\Schema\Schema;
- use Doctrine\DBAL\Schema\Table;
- use Doctrine\DBAL\Schema\Visitor\DropSchemaSqlCollector;
- use Doctrine\DBAL\Schema\Visitor\RemoveNamespacedAssets;
- use Doctrine\ORM\EntityManagerInterface;
- use Doctrine\ORM\Mapping\ClassMetadata;
- use Doctrine\ORM\Mapping\QuoteStrategy;
- use Doctrine\ORM\ORMException;
- use Doctrine\ORM\Tools\Event\GenerateSchemaEventArgs;
- use Doctrine\ORM\Tools\Event\GenerateSchemaTableEventArgs;
- use Throwable;
- use function array_diff;
- use function array_diff_key;
- use function array_filter;
- use function array_flip;
- use function array_intersect_key;
- use function assert;
- use function count;
- use function current;
- use function implode;
- use function in_array;
- use function is_array;
- use function is_numeric;
- use function strtolower;
- /**
- * The SchemaTool is a tool to create/drop/update database schemas based on
- * <tt>ClassMetadata</tt> class descriptors.
- *
- * @link www.doctrine-project.org
- */
- class SchemaTool
- {
- private const KNOWN_COLUMN_OPTIONS = ['comment', 'unsigned', 'fixed', 'default'];
- /** @var EntityManagerInterface */
- private $em;
- /** @var AbstractPlatform */
- private $platform;
- /**
- * The quote strategy.
- *
- * @var QuoteStrategy
- */
- private $quoteStrategy;
- /**
- * Initializes a new SchemaTool instance that uses the connection of the
- * provided EntityManager.
- */
- public function __construct(EntityManagerInterface $em)
- {
- $this->em = $em;
- $this->platform = $em->getConnection()->getDatabasePlatform();
- $this->quoteStrategy = $em->getConfiguration()->getQuoteStrategy();
- }
- /**
- * Creates the database schema for the given array of ClassMetadata instances.
- *
- * @return void
- *
- * @throws ToolsException
- *
- * @psalm-param list<ClassMetadata> $classes
- */
- public function createSchema(array $classes)
- {
- $createSchemaSql = $this->getCreateSchemaSql($classes);
- $conn = $this->em->getConnection();
- foreach ($createSchemaSql as $sql) {
- try {
- $conn->executeQuery($sql);
- } catch (Throwable $e) {
- throw ToolsException::schemaToolFailure($sql, $e);
- }
- }
- }
- /**
- * Gets the list of DDL statements that are required to create the database schema for
- * the given list of ClassMetadata instances.
- *
- * @return string[] The SQL statements needed to create the schema for the classes.
- *
- * @psalm-param list<ClassMetadata> $classes
- */
- public function getCreateSchemaSql(array $classes)
- {
- $schema = $this->getSchemaFromMetadata($classes);
- return $schema->toSql($this->platform);
- }
- /**
- * Detects instances of ClassMetadata that don't need to be processed in the SchemaTool context.
- *
- * @psalm-param array<string, bool> $processedClasses
- */
- private function processingNotRequired(
- ClassMetadata $class,
- array $processedClasses
- ): bool {
- return isset($processedClasses[$class->name]) ||
- $class->isMappedSuperclass ||
- $class->isEmbeddedClass ||
- ($class->isInheritanceTypeSingleTable() && $class->name !== $class->rootEntityName);
- }
- /**
- * Creates a Schema instance from a given set of metadata classes.
- *
- * @return Schema
- *
- * @throws ORMException
- *
- * @psalm-param list<ClassMetadata> $classes
- */
- public function getSchemaFromMetadata(array $classes)
- {
- // Reminder for processed classes, used for hierarchies
- $processedClasses = [];
- $eventManager = $this->em->getEventManager();
- $schemaManager = $this->em->getConnection()->getSchemaManager();
- $metadataSchemaConfig = $schemaManager->createSchemaConfig();
- $metadataSchemaConfig->setExplicitForeignKeyIndexes(false);
- $schema = new Schema([], [], $metadataSchemaConfig);
- $addedFks = [];
- $blacklistedFks = [];
- foreach ($classes as $class) {
- if ($this->processingNotRequired($class, $processedClasses)) {
- continue;
- }
- $table = $schema->createTable($this->quoteStrategy->getTableName($class, $this->platform));
- if ($class->isInheritanceTypeSingleTable()) {
- $this->gatherColumns($class, $table);
- $this->gatherRelationsSql($class, $table, $schema, $addedFks, $blacklistedFks);
- // Add the discriminator column
- $this->addDiscriminatorColumnDefinition($class, $table);
- // Aggregate all the information from all classes in the hierarchy
- foreach ($class->parentClasses as $parentClassName) {
- // Parent class information is already contained in this class
- $processedClasses[$parentClassName] = true;
- }
- foreach ($class->subClasses as $subClassName) {
- $subClass = $this->em->getClassMetadata($subClassName);
- $this->gatherColumns($subClass, $table);
- $this->gatherRelationsSql($subClass, $table, $schema, $addedFks, $blacklistedFks);
- $processedClasses[$subClassName] = true;
- }
- } elseif ($class->isInheritanceTypeJoined()) {
- // Add all non-inherited fields as columns
- foreach ($class->fieldMappings as $fieldName => $mapping) {
- if (! isset($mapping['inherited'])) {
- $this->gatherColumn($class, $mapping, $table);
- }
- }
- $this->gatherRelationsSql($class, $table, $schema, $addedFks, $blacklistedFks);
- // Add the discriminator column only to the root table
- if ($class->name === $class->rootEntityName) {
- $this->addDiscriminatorColumnDefinition($class, $table);
- } else {
- // Add an ID FK column to child tables
- $pkColumns = [];
- $inheritedKeyColumns = [];
- foreach ($class->identifier as $identifierField) {
- if (isset($class->fieldMappings[$identifierField]['inherited'])) {
- $idMapping = $class->fieldMappings[$identifierField];
- $this->gatherColumn($class, $idMapping, $table);
- $columnName = $this->quoteStrategy->getColumnName(
- $identifierField,
- $class,
- $this->platform
- );
- // TODO: This seems rather hackish, can we optimize it?
- $table->getColumn($columnName)->setAutoincrement(false);
- $pkColumns[] = $columnName;
- $inheritedKeyColumns[] = $columnName;
- continue;
- }
- if (isset($class->associationMappings[$identifierField]['inherited'])) {
- $idMapping = $class->associationMappings[$identifierField];
- $targetEntity = current(
- array_filter(
- $classes,
- static function (ClassMetadata $class) use ($idMapping): bool {
- return $class->name === $idMapping['targetEntity'];
- }
- )
- );
- foreach ($idMapping['joinColumns'] as $joinColumn) {
- if (isset($targetEntity->fieldMappings[$joinColumn['referencedColumnName']])) {
- $columnName = $this->quoteStrategy->getJoinColumnName(
- $joinColumn,
- $class,
- $this->platform
- );
- $pkColumns[] = $columnName;
- $inheritedKeyColumns[] = $columnName;
- }
- }
- }
- }
- if (! empty($inheritedKeyColumns)) {
- // Add a FK constraint on the ID column
- $table->addForeignKeyConstraint(
- $this->quoteStrategy->getTableName(
- $this->em->getClassMetadata($class->rootEntityName),
- $this->platform
- ),
- $inheritedKeyColumns,
- $inheritedKeyColumns,
- ['onDelete' => 'CASCADE']
- );
- }
- if (! empty($pkColumns)) {
- $table->setPrimaryKey($pkColumns);
- }
- }
- } elseif ($class->isInheritanceTypeTablePerClass()) {
- throw ORMException::notSupported();
- } else {
- $this->gatherColumns($class, $table);
- $this->gatherRelationsSql($class, $table, $schema, $addedFks, $blacklistedFks);
- }
- $pkColumns = [];
- foreach ($class->identifier as $identifierField) {
- if (isset($class->fieldMappings[$identifierField])) {
- $pkColumns[] = $this->quoteStrategy->getColumnName($identifierField, $class, $this->platform);
- } elseif (isset($class->associationMappings[$identifierField])) {
- $assoc = $class->associationMappings[$identifierField];
- assert(is_array($assoc));
- foreach ($assoc['joinColumns'] as $joinColumn) {
- $pkColumns[] = $this->quoteStrategy->getJoinColumnName($joinColumn, $class, $this->platform);
- }
- }
- }
- if (! $table->hasIndex('primary')) {
- $table->setPrimaryKey($pkColumns);
- }
- // there can be unique indexes automatically created for join column
- // if join column is also primary key we should keep only primary key on this column
- // so, remove indexes overruled by primary key
- $primaryKey = $table->getIndex('primary');
- foreach ($table->getIndexes() as $idxKey => $existingIndex) {
- if ($primaryKey->overrules($existingIndex)) {
- $table->dropIndex($idxKey);
- }
- }
- if (isset($class->table['indexes'])) {
- foreach ($class->table['indexes'] as $indexName => $indexData) {
- if (! isset($indexData['flags'])) {
- $indexData['flags'] = [];
- }
- $table->addIndex($indexData['columns'], is_numeric($indexName) ? null : $indexName, (array) $indexData['flags'], $indexData['options'] ?? []);
- }
- }
- if (isset($class->table['uniqueConstraints'])) {
- foreach ($class->table['uniqueConstraints'] as $indexName => $indexData) {
- $uniqIndex = new Index($indexName, $indexData['columns'], true, false, [], $indexData['options'] ?? []);
- foreach ($table->getIndexes() as $tableIndexName => $tableIndex) {
- if ($tableIndex->isFullfilledBy($uniqIndex)) {
- $table->dropIndex($tableIndexName);
- break;
- }
- }
- $table->addUniqueIndex($indexData['columns'], is_numeric($indexName) ? null : $indexName, $indexData['options'] ?? []);
- }
- }
- if (isset($class->table['options'])) {
- foreach ($class->table['options'] as $key => $val) {
- $table->addOption($key, $val);
- }
- }
- $processedClasses[$class->name] = true;
- if ($class->isIdGeneratorSequence() && $class->name === $class->rootEntityName) {
- $seqDef = $class->sequenceGeneratorDefinition;
- $quotedName = $this->quoteStrategy->getSequenceName($seqDef, $class, $this->platform);
- if (! $schema->hasSequence($quotedName)) {
- $schema->createSequence(
- $quotedName,
- $seqDef['allocationSize'],
- $seqDef['initialValue']
- );
- }
- }
- if ($eventManager->hasListeners(ToolEvents::postGenerateSchemaTable)) {
- $eventManager->dispatchEvent(
- ToolEvents::postGenerateSchemaTable,
- new GenerateSchemaTableEventArgs($class, $schema, $table)
- );
- }
- }
- if (! $this->platform->supportsSchemas() && ! $this->platform->canEmulateSchemas()) {
- $schema->visit(new RemoveNamespacedAssets());
- }
- if ($eventManager->hasListeners(ToolEvents::postGenerateSchema)) {
- $eventManager->dispatchEvent(
- ToolEvents::postGenerateSchema,
- new GenerateSchemaEventArgs($this->em, $schema)
- );
- }
- return $schema;
- }
- /**
- * Gets a portable column definition as required by the DBAL for the discriminator
- * column of a class.
- *
- * @param ClassMetadata $class
- *
- * @return void
- */
- private function addDiscriminatorColumnDefinition($class, Table $table)
- {
- $discrColumn = $class->discriminatorColumn;
- if (
- ! isset($discrColumn['type']) ||
- (strtolower($discrColumn['type']) === 'string' && ! isset($discrColumn['length']))
- ) {
- $discrColumn['type'] = 'string';
- $discrColumn['length'] = 255;
- }
- $options = [
- 'length' => $discrColumn['length'] ?? null,
- 'notnull' => true,
- ];
- if (isset($discrColumn['columnDefinition'])) {
- $options['columnDefinition'] = $discrColumn['columnDefinition'];
- }
- $table->addColumn($discrColumn['name'], $discrColumn['type'], $options);
- }
- /**
- * Gathers the column definitions as required by the DBAL of all field mappings
- * found in the given class.
- *
- * @param ClassMetadata $class
- *
- * @return void
- */
- private function gatherColumns($class, Table $table)
- {
- $pkColumns = [];
- foreach ($class->fieldMappings as $mapping) {
- if ($class->isInheritanceTypeSingleTable() && isset($mapping['inherited'])) {
- continue;
- }
- $this->gatherColumn($class, $mapping, $table);
- if ($class->isIdentifier($mapping['fieldName'])) {
- $pkColumns[] = $this->quoteStrategy->getColumnName($mapping['fieldName'], $class, $this->platform);
- }
- }
- }
- /**
- * Creates a column definition as required by the DBAL from an ORM field mapping definition.
- *
- * @param ClassMetadata $class The class that owns the field mapping.
- *
- * @psalm-param array<string, mixed> $mapping The field mapping.
- */
- private function gatherColumn(
- ClassMetadata $class,
- array $mapping,
- Table $table
- ): void {
- $columnName = $this->quoteStrategy->getColumnName($mapping['fieldName'], $class, $this->platform);
- $columnType = $mapping['type'];
- $options = [];
- $options['length'] = $mapping['length'] ?? null;
- $options['notnull'] = isset($mapping['nullable']) ? ! $mapping['nullable'] : true;
- if ($class->isInheritanceTypeSingleTable() && $class->parentClasses) {
- $options['notnull'] = false;
- }
- $options['platformOptions'] = [];
- $options['platformOptions']['version'] = $class->isVersioned && $class->versionField === $mapping['fieldName'];
- if (strtolower($columnType) === 'string' && $options['length'] === null) {
- $options['length'] = 255;
- }
- if (isset($mapping['precision'])) {
- $options['precision'] = $mapping['precision'];
- }
- if (isset($mapping['scale'])) {
- $options['scale'] = $mapping['scale'];
- }
- if (isset($mapping['default'])) {
- $options['default'] = $mapping['default'];
- }
- if (isset($mapping['columnDefinition'])) {
- $options['columnDefinition'] = $mapping['columnDefinition'];
- }
- // the 'default' option can be overwritten here
- $options = $this->gatherColumnOptions($mapping) + $options;
- if ($class->isIdGeneratorIdentity() && $class->getIdentifierFieldNames() === [$mapping['fieldName']]) {
- $options['autoincrement'] = true;
- }
- if ($class->isInheritanceTypeJoined() && $class->name !== $class->rootEntityName) {
- $options['autoincrement'] = false;
- }
- if ($table->hasColumn($columnName)) {
- // required in some inheritance scenarios
- $table->changeColumn($columnName, $options);
- } else {
- $table->addColumn($columnName, $columnType, $options);
- }
- $isUnique = $mapping['unique'] ?? false;
- if ($isUnique) {
- $table->addUniqueIndex([$columnName]);
- }
- }
- /**
- * Gathers the SQL for properly setting up the relations of the given class.
- * This includes the SQL for foreign key constraints and join tables.
- *
- * @throws ORMException
- *
- * @psalm-param array<string, array{
- * foreignTableName: string,
- * foreignColumns: list<string>
- * }> $addedFks
- * @psalm-param array<string, bool> $blacklistedFks
- */
- private function gatherRelationsSql(
- ClassMetadata $class,
- Table $table,
- Schema $schema,
- array &$addedFks,
- array &$blacklistedFks
- ): void {
- foreach ($class->associationMappings as $id => $mapping) {
- if (isset($mapping['inherited']) && ! in_array($id, $class->identifier, true)) {
- continue;
- }
- $foreignClass = $this->em->getClassMetadata($mapping['targetEntity']);
- if ($mapping['type'] & ClassMetadata::TO_ONE && $mapping['isOwningSide']) {
- $primaryKeyColumns = []; // PK is unnecessary for this relation-type
- $this->gatherRelationJoinColumns(
- $mapping['joinColumns'],
- $table,
- $foreignClass,
- $mapping,
- $primaryKeyColumns,
- $addedFks,
- $blacklistedFks
- );
- } elseif ($mapping['type'] === ClassMetadata::ONE_TO_MANY && $mapping['isOwningSide']) {
- //... create join table, one-many through join table supported later
- throw ORMException::notSupported();
- } elseif ($mapping['type'] === ClassMetadata::MANY_TO_MANY && $mapping['isOwningSide']) {
- // create join table
- $joinTable = $mapping['joinTable'];
- $theJoinTable = $schema->createTable(
- $this->quoteStrategy->getJoinTableName($mapping, $foreignClass, $this->platform)
- );
- $primaryKeyColumns = [];
- // Build first FK constraint (relation table => source table)
- $this->gatherRelationJoinColumns(
- $joinTable['joinColumns'],
- $theJoinTable,
- $class,
- $mapping,
- $primaryKeyColumns,
- $addedFks,
- $blacklistedFks
- );
- // Build second FK constraint (relation table => target table)
- $this->gatherRelationJoinColumns(
- $joinTable['inverseJoinColumns'],
- $theJoinTable,
- $foreignClass,
- $mapping,
- $primaryKeyColumns,
- $addedFks,
- $blacklistedFks
- );
- $theJoinTable->setPrimaryKey($primaryKeyColumns);
- }
- }
- }
- /**
- * Gets the class metadata that is responsible for the definition of the referenced column name.
- *
- * Previously this was a simple task, but with DDC-117 this problem is actually recursive. If its
- * not a simple field, go through all identifier field names that are associations recursively and
- * find that referenced column name.
- *
- * TODO: Is there any way to make this code more pleasing?
- *
- * @psalm-return array{ClassMetadata, string}|null
- */
- private function getDefiningClass(ClassMetadata $class, string $referencedColumnName): ?array
- {
- $referencedFieldName = $class->getFieldName($referencedColumnName);
- if ($class->hasField($referencedFieldName)) {
- return [$class, $referencedFieldName];
- }
- if (in_array($referencedColumnName, $class->getIdentifierColumnNames())) {
- // it seems to be an entity as foreign key
- foreach ($class->getIdentifierFieldNames() as $fieldName) {
- if (
- $class->hasAssociation($fieldName)
- && $class->getSingleAssociationJoinColumnName($fieldName) === $referencedColumnName
- ) {
- return $this->getDefiningClass(
- $this->em->getClassMetadata($class->associationMappings[$fieldName]['targetEntity']),
- $class->getSingleAssociationReferencedJoinColumnName($fieldName)
- );
- }
- }
- }
- return null;
- }
- /**
- * Gathers columns and fk constraints that are required for one part of relationship.
- *
- * @throws ORMException
- *
- * @psalm-param array<string, mixed> $joinColumns
- * @psalm-param array<string, mixed> $mapping
- * @psalm-param list<string> $primaryKeyColumns
- * @psalm-param array<string, array{
- * foreignTableName: string,
- * foreignColumns: list<string>
- * }> $addedFks
- * @psalm-param array<string,bool> $blacklistedFks
- */
- private function gatherRelationJoinColumns(
- array $joinColumns,
- Table $theJoinTable,
- ClassMetadata $class,
- array $mapping,
- array &$primaryKeyColumns,
- array &$addedFks,
- array &$blacklistedFks
- ): void {
- $localColumns = [];
- $foreignColumns = [];
- $fkOptions = [];
- $foreignTableName = $this->quoteStrategy->getTableName($class, $this->platform);
- $uniqueConstraints = [];
- foreach ($joinColumns as $joinColumn) {
- [$definingClass, $referencedFieldName] = $this->getDefiningClass(
- $class,
- $joinColumn['referencedColumnName']
- );
- if (! $definingClass) {
- throw new ORMException(
- 'Column name `' . $joinColumn['referencedColumnName'] . '` referenced for relation from '
- . $mapping['sourceEntity'] . ' towards ' . $mapping['targetEntity'] . ' does not exist.'
- );
- }
- $quotedColumnName = $this->quoteStrategy->getJoinColumnName($joinColumn, $class, $this->platform);
- $quotedRefColumnName = $this->quoteStrategy->getReferencedJoinColumnName(
- $joinColumn,
- $class,
- $this->platform
- );
- $primaryKeyColumns[] = $quotedColumnName;
- $localColumns[] = $quotedColumnName;
- $foreignColumns[] = $quotedRefColumnName;
- if (! $theJoinTable->hasColumn($quotedColumnName)) {
- // Only add the column to the table if it does not exist already.
- // It might exist already if the foreign key is mapped into a regular
- // property as well.
- $fieldMapping = $definingClass->getFieldMapping($referencedFieldName);
- $columnDef = null;
- if (isset($joinColumn['columnDefinition'])) {
- $columnDef = $joinColumn['columnDefinition'];
- } elseif (isset($fieldMapping['columnDefinition'])) {
- $columnDef = $fieldMapping['columnDefinition'];
- }
- $columnOptions = ['notnull' => false, 'columnDefinition' => $columnDef];
- if (isset($joinColumn['nullable'])) {
- $columnOptions['notnull'] = ! $joinColumn['nullable'];
- }
- $columnOptions += $this->gatherColumnOptions($fieldMapping);
- if ($fieldMapping['type'] === 'string' && isset($fieldMapping['length'])) {
- $columnOptions['length'] = $fieldMapping['length'];
- } elseif ($fieldMapping['type'] === 'decimal') {
- $columnOptions['scale'] = $fieldMapping['scale'];
- $columnOptions['precision'] = $fieldMapping['precision'];
- }
- $theJoinTable->addColumn($quotedColumnName, $fieldMapping['type'], $columnOptions);
- }
- if (isset($joinColumn['unique']) && $joinColumn['unique'] === true) {
- $uniqueConstraints[] = ['columns' => [$quotedColumnName]];
- }
- if (isset($joinColumn['onDelete'])) {
- $fkOptions['onDelete'] = $joinColumn['onDelete'];
- }
- }
- // Prefer unique constraints over implicit simple indexes created for foreign keys.
- // Also avoids index duplication.
- foreach ($uniqueConstraints as $indexName => $unique) {
- $theJoinTable->addUniqueIndex($unique['columns'], is_numeric($indexName) ? null : $indexName);
- }
- $compositeName = $theJoinTable->getName() . '.' . implode('', $localColumns);
- if (
- isset($addedFks[$compositeName])
- && ($foreignTableName !== $addedFks[$compositeName]['foreignTableName']
- || 0 < count(array_diff($foreignColumns, $addedFks[$compositeName]['foreignColumns'])))
- ) {
- foreach ($theJoinTable->getForeignKeys() as $fkName => $key) {
- if (
- count(array_diff($key->getLocalColumns(), $localColumns)) === 0
- && (($key->getForeignTableName() !== $foreignTableName)
- || 0 < count(array_diff($key->getForeignColumns(), $foreignColumns)))
- ) {
- $theJoinTable->removeForeignKey($fkName);
- break;
- }
- }
- $blacklistedFks[$compositeName] = true;
- } elseif (! isset($blacklistedFks[$compositeName])) {
- $addedFks[$compositeName] = ['foreignTableName' => $foreignTableName, 'foreignColumns' => $foreignColumns];
- $theJoinTable->addUnnamedForeignKeyConstraint(
- $foreignTableName,
- $localColumns,
- $foreignColumns,
- $fkOptions
- );
- }
- }
- /**
- * @param mixed[] $mapping
- *
- * @return mixed[]
- */
- private function gatherColumnOptions(array $mapping): array
- {
- if (! isset($mapping['options'])) {
- return [];
- }
- $options = array_intersect_key($mapping['options'], array_flip(self::KNOWN_COLUMN_OPTIONS));
- $options['customSchemaOptions'] = array_diff_key($mapping['options'], $options);
- return $options;
- }
- /**
- * Drops the database schema for the given classes.
- *
- * In any way when an exception is thrown it is suppressed since drop was
- * issued for all classes of the schema and some probably just don't exist.
- *
- * @return void
- *
- * @psalm-param list<ClassMetadata> $classes
- */
- public function dropSchema(array $classes)
- {
- $dropSchemaSql = $this->getDropSchemaSQL($classes);
- $conn = $this->em->getConnection();
- foreach ($dropSchemaSql as $sql) {
- try {
- $conn->executeQuery($sql);
- } catch (Throwable $e) {
- // ignored
- }
- }
- }
- /**
- * Drops all elements in the database of the current connection.
- *
- * @return void
- */
- public function dropDatabase()
- {
- $dropSchemaSql = $this->getDropDatabaseSQL();
- $conn = $this->em->getConnection();
- foreach ($dropSchemaSql as $sql) {
- $conn->executeQuery($sql);
- }
- }
- /**
- * Gets the SQL needed to drop the database schema for the connections database.
- *
- * @return string[]
- */
- public function getDropDatabaseSQL()
- {
- $sm = $this->em->getConnection()->getSchemaManager();
- $schema = $sm->createSchema();
- $visitor = new DropSchemaSqlCollector($this->platform);
- $schema->visit($visitor);
- return $visitor->getQueries();
- }
- /**
- * Gets SQL to drop the tables defined by the passed classes.
- *
- * @return string[]
- *
- * @psalm-param list<ClassMetadata> $classes
- */
- public function getDropSchemaSQL(array $classes)
- {
- $visitor = new DropSchemaSqlCollector($this->platform);
- $schema = $this->getSchemaFromMetadata($classes);
- $sm = $this->em->getConnection()->getSchemaManager();
- $fullSchema = $sm->createSchema();
- foreach ($fullSchema->getTables() as $table) {
- if (! $schema->hasTable($table->getName())) {
- foreach ($table->getForeignKeys() as $foreignKey) {
- if ($schema->hasTable($foreignKey->getForeignTableName())) {
- $visitor->acceptForeignKey($table, $foreignKey);
- }
- }
- } else {
- $visitor->acceptTable($table);
- foreach ($table->getForeignKeys() as $foreignKey) {
- $visitor->acceptForeignKey($table, $foreignKey);
- }
- }
- }
- if ($this->platform->supportsSequences()) {
- foreach ($schema->getSequences() as $sequence) {
- $visitor->acceptSequence($sequence);
- }
- foreach ($schema->getTables() as $table) {
- if ($table->hasPrimaryKey()) {
- $columns = $table->getPrimaryKey()->getColumns();
- if (count($columns) === 1) {
- $checkSequence = $table->getName() . '_' . $columns[0] . '_seq';
- if ($fullSchema->hasSequence($checkSequence)) {
- $visitor->acceptSequence($fullSchema->getSequence($checkSequence));
- }
- }
- }
- }
- }
- return $visitor->getQueries();
- }
- /**
- * Updates the database schema of the given classes by comparing the ClassMetadata
- * instances to the current database schema that is inspected.
- *
- * @param mixed[] $classes
- * @param bool $saveMode If TRUE, only performs a partial update
- * without dropping assets which are scheduled for deletion.
- *
- * @return void
- */
- public function updateSchema(array $classes, $saveMode = false)
- {
- $updateSchemaSql = $this->getUpdateSchemaSql($classes, $saveMode);
- $conn = $this->em->getConnection();
- foreach ($updateSchemaSql as $sql) {
- $conn->executeQuery($sql);
- }
- }
- /**
- * Gets the sequence of SQL statements that need to be performed in order
- * to bring the given class mappings in-synch with the relational schema.
- *
- * @param mixed[] $classes The classes to consider.
- * @param bool $saveMode If TRUE, only generates SQL for a partial update
- * that does not include SQL for dropping assets which are scheduled for deletion.
- *
- * @return string[] The sequence of SQL statements.
- */
- public function getUpdateSchemaSql(array $classes, $saveMode = false)
- {
- $toSchema = $this->getSchemaFromMetadata($classes);
- $fromSchema = $this->createSchemaForComparison($toSchema);
- $comparator = new Comparator();
- $schemaDiff = $comparator->compare($fromSchema, $toSchema);
- if ($saveMode) {
- return $schemaDiff->toSaveSql($this->platform);
- }
- return $schemaDiff->toSql($this->platform);
- }
- /**
- * Creates the schema from the database, ensuring tables from the target schema are whitelisted for comparison.
- */
- private function createSchemaForComparison(Schema $toSchema): Schema
- {
- $connection = $this->em->getConnection();
- $schemaManager = $connection->getSchemaManager();
- // backup schema assets filter
- $config = $connection->getConfiguration();
- $previousFilter = $config->getSchemaAssetsFilter();
- if ($previousFilter === null) {
- return $schemaManager->createSchema();
- }
- // whitelist assets we already know about in $toSchema, use the existing filter otherwise
- $config->setSchemaAssetsFilter(static function ($asset) use ($previousFilter, $toSchema): bool {
- $assetName = $asset instanceof AbstractAsset ? $asset->getName() : $asset;
- return $toSchema->hasTable($assetName) || $toSchema->hasSequence($assetName) || $previousFilter($asset);
- });
- try {
- return $schemaManager->createSchema();
- } finally {
- // restore schema assets filter
- $config->setSchemaAssetsFilter($previousFilter);
- }
- }
- }
|