typeorm
Version:
Data-Mapper ORM for TypeScript and ES2023+. Supports MySQL/MariaDB, PostgreSQL, MS SQL Server, Oracle, SAP HANA, SQLite, MongoDB databases.
2,293 lines • 110 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.MysqlQueryRunner = void 0;
const error_1 = require("../../error");
const QueryFailedError_1 = require("../../error/QueryFailedError");
const QueryRunnerAlreadyReleasedError_1 = require("../../error/QueryRunnerAlreadyReleasedError");
const TransactionNotStartedError_1 = require("../../error/TransactionNotStartedError");
const BaseQueryRunner_1 = require("../../query-runner/BaseQueryRunner");
const QueryResult_1 = require("../../query-runner/QueryResult");
const Table_1 = require("../../schema-builder/table/Table");
const TableColumn_1 = require("../../schema-builder/table/TableColumn");
const TableForeignKey_1 = require("../../schema-builder/table/TableForeignKey");
const TableIndex_1 = require("../../schema-builder/table/TableIndex");
const TableUnique_1 = require("../../schema-builder/table/TableUnique");
const View_1 = require("../../schema-builder/view/View");
const Broadcaster_1 = require("../../subscriber/Broadcaster");
const BroadcasterResult_1 = require("../../subscriber/BroadcasterResult");
const InstanceChecker_1 = require("../../util/InstanceChecker");
const OrmUtils_1 = require("../../util/OrmUtils");
const VersionUtils_1 = require("../../util/VersionUtils");
const Query_1 = require("../Query");
const validate_isolation_level_1 = require("../validate-isolation-level");
const MetadataTableType_1 = require("../types/MetadataTableType");
/**
* Runs queries on a single mysql database connection.
*/
class MysqlQueryRunner extends BaseQueryRunner_1.BaseQueryRunner {
// -------------------------------------------------------------------------
// Constructor
// -------------------------------------------------------------------------
constructor(driver, mode) {
super();
this.driver = driver;
this.dataSource = driver.dataSource;
this.broadcaster = new Broadcaster_1.Broadcaster(this);
this.mode = mode;
}
// -------------------------------------------------------------------------
// Public Methods
// -------------------------------------------------------------------------
/**
* Creates/uses database connection from the connection pool to perform further operations.
* Returns obtained database connection.
*/
connect() {
if (this.databaseConnection)
return Promise.resolve(this.databaseConnection);
if (this.databaseConnectionPromise)
return this.databaseConnectionPromise;
if (this.mode === "slave" && this.driver.isReplicated) {
this.databaseConnectionPromise = this.driver
.obtainSlaveConnection()
.then((connection) => {
this.databaseConnection = connection;
return this.databaseConnection;
});
}
else {
// master
this.databaseConnectionPromise = this.driver
.obtainMasterConnection()
.then((connection) => {
this.databaseConnection = connection;
return this.databaseConnection;
});
}
return this.databaseConnectionPromise;
}
/**
* Releases used database connection.
* You cannot use query runner methods once its released.
*/
release() {
this.isReleased = true;
if (this.databaseConnection)
this.databaseConnection.release();
return Promise.resolve();
}
/**
* Starts transaction on the current connection.
*
* @param isolationLevel
*/
async startTransaction(isolationLevel) {
isolationLevel ??= this.dataSource.options.isolationLevel;
(0, validate_isolation_level_1.validateIsolationLevel)(this.driver.supportedIsolationLevels, isolationLevel);
this.isTransactionActive = true;
try {
await this.broadcaster.broadcast("BeforeTransactionStart");
}
catch (err) {
this.isTransactionActive = false;
throw err;
}
if (this.transactionDepth === 0) {
if (isolationLevel) {
await this.query("SET TRANSACTION ISOLATION LEVEL " + isolationLevel);
}
await this.query("START TRANSACTION");
}
else {
await this.query(`SAVEPOINT typeorm_${this.transactionDepth}`);
}
this.transactionDepth += 1;
await this.broadcaster.broadcast("AfterTransactionStart");
}
/**
* Commits transaction.
* Error will be thrown if transaction was not started.
*/
async commitTransaction() {
if (!this.isTransactionActive)
throw new TransactionNotStartedError_1.TransactionNotStartedError();
await this.broadcaster.broadcast("BeforeTransactionCommit");
if (this.transactionDepth > 1) {
await this.query(`RELEASE SAVEPOINT typeorm_${this.transactionDepth - 1}`);
}
else {
await this.query("COMMIT");
this.isTransactionActive = false;
}
this.transactionDepth -= 1;
await this.broadcaster.broadcast("AfterTransactionCommit");
}
/**
* Rollbacks transaction.
* Error will be thrown if transaction was not started.
*/
async rollbackTransaction() {
if (!this.isTransactionActive)
throw new TransactionNotStartedError_1.TransactionNotStartedError();
await this.broadcaster.broadcast("BeforeTransactionRollback");
if (this.transactionDepth > 1) {
await this.query(`ROLLBACK TO SAVEPOINT typeorm_${this.transactionDepth - 1}`);
}
else {
await this.query("ROLLBACK");
this.isTransactionActive = false;
}
this.transactionDepth -= 1;
await this.broadcaster.broadcast("AfterTransactionRollback");
}
/**
* Executes a raw SQL query.
*
* @param query
* @param parameters
* @param useStructuredResult
*/
async query(query, parameters, useStructuredResult = false) {
if (this.isReleased)
throw new QueryRunnerAlreadyReleasedError_1.QueryRunnerAlreadyReleasedError();
const databaseConnection = await this.connect();
this.driver.dataSource.logger.logQuery(query, parameters, this);
await this.broadcaster.broadcast("BeforeQuery", query, parameters);
const broadcasterResult = new BroadcasterResult_1.BroadcasterResult();
const maxQueryExecutionTime = this.driver.options.maxQueryExecutionTime;
const enableQueryTimeout = this.driver.options.enableQueryTimeout;
const queryPayload = enableQueryTimeout && maxQueryExecutionTime
? { sql: query, timeout: maxQueryExecutionTime }
: query;
const queryStartTime = Date.now();
try {
const [raw] = await databaseConnection
.promise()
.query(queryPayload, parameters);
const queryExecutionTime = Date.now() - queryStartTime;
if (maxQueryExecutionTime &&
queryExecutionTime > maxQueryExecutionTime)
this.driver.dataSource.logger.logQuerySlow(queryExecutionTime, query, parameters, this);
this.broadcaster.broadcastAfterQueryEvent(broadcasterResult, query, parameters, true, queryExecutionTime, raw, undefined);
const result = new QueryResult_1.QueryResult();
result.raw = raw;
try {
result.records = Array.from(raw);
}
catch {
// Do nothing.
}
if (raw?.hasOwnProperty("affectedRows")) {
result.affected = raw.affectedRows;
}
return useStructuredResult ? result : result.raw;
}
catch (err) {
const queryExecutionTime = Date.now() - queryStartTime;
if (maxQueryExecutionTime &&
queryExecutionTime > maxQueryExecutionTime)
this.driver.dataSource.logger.logQuerySlow(queryExecutionTime, query, parameters, this);
this.driver.dataSource.logger.logQueryError(err, query, parameters, this);
this.broadcaster.broadcastAfterQueryEvent(broadcasterResult, query, parameters, false, undefined, undefined, err);
throw new QueryFailedError_1.QueryFailedError(query, parameters, err);
}
finally {
await broadcasterResult.wait();
}
}
/**
* Returns raw data stream.
*
* @param query
* @param parameters
* @param onEnd
* @param onError
*/
stream(query, parameters, onEnd, onError) {
if (this.isReleased)
throw new QueryRunnerAlreadyReleasedError_1.QueryRunnerAlreadyReleasedError();
return new Promise(async (ok, fail) => {
try {
const databaseConnection = await this.connect();
this.driver.dataSource.logger.logQuery(query, parameters, this);
const databaseQuery = databaseConnection.query(query, parameters);
if (onEnd)
databaseQuery.on("end", onEnd);
if (onError)
databaseQuery.on("error", onError);
ok(databaseQuery.stream());
}
catch (err) {
fail(err);
}
});
}
/**
* Returns all available database names including system databases.
*/
async getDatabases() {
return Promise.resolve([]);
}
/**
* Returns all available schema names including system schemas.
* If database parameter specified, returns schemas of that database.
*
* @param database
*/
async getSchemas(database) {
throw new error_1.TypeORMError(`MySql driver does not support table schemas`);
}
/**
* Checks if database with the given name exist.
*
* @param database
*/
async hasDatabase(database) {
const result = await this.query(`SELECT * FROM \`INFORMATION_SCHEMA\`.\`SCHEMATA\` WHERE \`SCHEMA_NAME\` = ?`, [database]);
return result.length ? true : false;
}
/**
* Loads currently using database
*/
async getCurrentDatabase() {
const query = await this.query(`SELECT DATABASE() AS \`db_name\``);
return query[0]["db_name"];
}
/**
* Checks if schema with the given name exist.
*
* @param schema
*/
async hasSchema(schema) {
throw new error_1.TypeORMError(`MySql driver does not support table schemas`);
}
/**
* Loads currently using database schema
*/
async getCurrentSchema() {
const query = await this.query(`SELECT SCHEMA() AS \`schema_name\``);
return query[0]["schema_name"];
}
/**
* Checks if table with the given name exist in the database.
*
* @param tableOrName
*/
async hasTable(tableOrName) {
const parsedTableName = this.driver.parseTableName(tableOrName);
const sql = `SELECT * FROM \`INFORMATION_SCHEMA\`.\`COLUMNS\` WHERE \`TABLE_SCHEMA\` = ? AND \`TABLE_NAME\` = ?`;
const result = await this.query(sql, [
parsedTableName.database,
parsedTableName.tableName,
]);
return result.length ? true : false;
}
/**
* Checks if column with the given name exist in the given table.
*
* @param tableOrName
* @param column
*/
async hasColumn(tableOrName, column) {
const parsedTableName = this.driver.parseTableName(tableOrName);
const columnName = InstanceChecker_1.InstanceChecker.isTableColumn(column)
? column.name
: column;
const sql = `SELECT * FROM \`INFORMATION_SCHEMA\`.\`COLUMNS\` WHERE \`TABLE_SCHEMA\` = ? AND \`TABLE_NAME\` = ? AND \`COLUMN_NAME\` = ?`;
const result = await this.query(sql, [
parsedTableName.database,
parsedTableName.tableName,
columnName,
]);
return result.length ? true : false;
}
/**
* Creates a new database.
*
* @param database
* @param ifNotExists
*/
async createDatabase(database, ifNotExists) {
const up = ifNotExists
? `CREATE DATABASE IF NOT EXISTS ${this.driver.escape(database)}`
: `CREATE DATABASE ${this.driver.escape(database)}`;
const down = `DROP DATABASE ${this.driver.escape(database)}`;
await this.executeQueries(new Query_1.Query(up), new Query_1.Query(down));
}
/**
* Drops database.
*
* @param database
* @param ifExists
*/
async dropDatabase(database, ifExists) {
const up = ifExists
? `DROP DATABASE IF EXISTS ${this.driver.escape(database)}`
: `DROP DATABASE ${this.driver.escape(database)}`;
const down = `CREATE DATABASE ${this.driver.escape(database)}`;
await this.executeQueries(new Query_1.Query(up), new Query_1.Query(down));
}
/**
* Creates a new table schema.
*
* @param schemaPath
* @param ifNotExists
*/
async createSchema(schemaPath, ifNotExists) {
throw new error_1.TypeORMError(`Schema create queries are not supported by MySql driver.`);
}
/**
* Drops table schema.
*
* @param schemaPath
* @param ifExists
*/
async dropSchema(schemaPath, ifExists) {
throw new error_1.TypeORMError(`Schema drop queries are not supported by MySql driver.`);
}
/**
* Creates a new table.
*
* @param table
* @param ifNotExists
* @param createForeignKeys
*/
async createTable(table, ifNotExists = false, createForeignKeys = true) {
if (ifNotExists) {
const isTableExist = await this.hasTable(table);
if (isTableExist)
return Promise.resolve();
}
const upQueries = [];
const downQueries = [];
upQueries.push(this.createTableSql(table, createForeignKeys));
downQueries.push(this.dropTableSql(table));
// we must first drop indices, than drop foreign keys, because drop queries runs in reversed order
// and foreign keys will be dropped first as indices. This order is very important, because we can't drop index
// if it related to the foreign key.
// createTable does not need separate method to create indices, because it create indices in the same query with table creation.
table.indices.forEach((index) => downQueries.push(this.dropIndexSql(table, index)));
// if createForeignKeys is true, we must drop created foreign keys in down query.
// createTable does not need separate method to create foreign keys, because it create fk's in the same query with table creation.
if (createForeignKeys)
table.foreignKeys.forEach((foreignKey) => downQueries.push(this.dropForeignKeySql(table, foreignKey)));
// if table has column with generated type, we must add the expression to the metadata table
const generatedColumns = table.columns.filter((column) => column.generatedType && column.asExpression);
if (generatedColumns.length > 0) {
const currentDatabase = await this.getCurrentDatabase();
for (const column of generatedColumns) {
const insertQuery = this.insertTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: column.name,
value: column.asExpression,
});
const deleteQuery = this.deleteTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: column.name,
});
upQueries.push(insertQuery);
downQueries.push(deleteQuery);
}
}
return this.executeQueries(upQueries, downQueries);
}
/**
* Drop the table.
*
* @param target
* @param ifExists
* @param dropForeignKeys
*/
async dropTable(target, ifExists, dropForeignKeys = true) {
// It needs because if table does not exist and dropForeignKeys or dropIndices is true, we don't need
// to perform drop queries for foreign keys and indices.
if (ifExists) {
const isTableExist = await this.hasTable(target);
if (!isTableExist)
return Promise.resolve();
}
// if dropTable called with dropForeignKeys = true, we must create foreign keys in down query.
const createForeignKeys = dropForeignKeys;
const tablePath = this.getTablePath(target);
const table = await this.getCachedTable(tablePath);
const upQueries = [];
const downQueries = [];
if (dropForeignKeys)
table.foreignKeys.forEach((foreignKey) => upQueries.push(this.dropForeignKeySql(table, foreignKey)));
table.indices.forEach((index) => upQueries.push(this.dropIndexSql(table, index)));
upQueries.push(this.dropTableSql(table));
downQueries.push(this.createTableSql(table, createForeignKeys));
// if table had columns with generated type, we must remove the expression from the metadata table
const generatedColumns = table.columns.filter((column) => column.generatedType && column.asExpression);
if (generatedColumns.length > 0) {
const currentDatabase = await this.getCurrentDatabase();
for (const column of generatedColumns) {
const deleteQuery = this.deleteTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: column.name,
});
const insertQuery = this.insertTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: column.name,
value: column.asExpression,
});
upQueries.push(deleteQuery);
downQueries.push(insertQuery);
}
}
await this.executeQueries(upQueries, downQueries);
}
/**
* Creates a new view.
*
* @param view
* @param syncWithMetadata
*/
async createView(view, syncWithMetadata = false) {
const upQueries = [];
const downQueries = [];
upQueries.push(this.createViewSql(view));
if (syncWithMetadata)
upQueries.push(await this.insertViewDefinitionSql(view));
downQueries.push(this.dropViewSql(view));
if (syncWithMetadata)
downQueries.push(await this.deleteViewDefinitionSql(view));
await this.executeQueries(upQueries, downQueries);
}
/**
* Drops the view.
*
* @param target
* @param ifExists
*/
async dropView(target, ifExists) {
const viewName = InstanceChecker_1.InstanceChecker.isView(target) ? target.name : target;
const view = await this.getCachedView(viewName);
const upQueries = [];
const downQueries = [];
upQueries.push(await this.deleteViewDefinitionSql(view));
if (ifExists) {
upQueries.push(new Query_1.Query(`DROP VIEW IF EXISTS ${this.escapePath(view)}`));
}
else {
upQueries.push(this.dropViewSql(view));
}
downQueries.push(await this.insertViewDefinitionSql(view));
downQueries.push(this.createViewSql(view));
await this.executeQueries(upQueries, downQueries);
}
/**
* Renames a table.
*
* @param oldTableOrName
* @param newTableName
*/
async renameTable(oldTableOrName, newTableName) {
const upQueries = [];
const downQueries = [];
const oldTable = InstanceChecker_1.InstanceChecker.isTable(oldTableOrName)
? oldTableOrName
: await this.getCachedTable(oldTableOrName);
const newTable = oldTable.clone();
const { database } = this.driver.parseTableName(oldTable);
newTable.name = database ? `${database}.${newTableName}` : newTableName;
// rename table
upQueries.push(new Query_1.Query(`RENAME TABLE ${this.escapePath(oldTable)} TO ${this.escapePath(newTable)}`));
downQueries.push(new Query_1.Query(`RENAME TABLE ${this.escapePath(newTable)} TO ${this.escapePath(oldTable)}`));
// rename index constraints
newTable.indices.forEach((index) => {
const oldIndexName = this.dataSource.namingStrategy.indexName(oldTable, index.columnNames);
// Skip renaming if Index has user defined constraint name
if (index.name !== oldIndexName)
return;
// build new constraint name
const columnNames = index.columnNames
.map((column) => `\`${column}\``)
.join(", ");
const newIndexName = this.dataSource.namingStrategy.indexName(newTable, index.columnNames, index.where);
// build queries
let indexType = "";
if (index.isUnique)
indexType += "UNIQUE ";
if (index.isSpatial)
indexType += "SPATIAL ";
if (index.isFulltext)
indexType += "FULLTEXT ";
const indexParser = index.isFulltext && index.parser
? ` WITH PARSER ${index.parser}`
: "";
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(newTable)} DROP INDEX \`${index.name}\`, ADD ${indexType}INDEX \`${newIndexName}\` (${columnNames})${indexParser}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(newTable)} DROP INDEX \`${newIndexName}\`, ADD ${indexType}INDEX \`${index.name}\` (${columnNames})${indexParser}`));
// replace constraint name
index.name = newIndexName;
});
// rename foreign key constraint
newTable.foreignKeys.forEach((foreignKey) => {
const oldForeignKeyName = this.dataSource.namingStrategy.foreignKeyName(oldTable, foreignKey.columnNames, this.getTablePath(foreignKey), foreignKey.referencedColumnNames);
// Skip renaming if foreign key has user defined constraint name
if (foreignKey.name !== oldForeignKeyName)
return;
// build new constraint name
const columnNames = foreignKey.columnNames
.map((column) => `\`${column}\``)
.join(", ");
const referencedColumnNames = foreignKey.referencedColumnNames
.map((column) => `\`${column}\``)
.join(",");
const newForeignKeyName = this.dataSource.namingStrategy.foreignKeyName(newTable, foreignKey.columnNames, this.getTablePath(foreignKey), foreignKey.referencedColumnNames);
// build queries
let up = `ALTER TABLE ${this.escapePath(newTable)} DROP FOREIGN KEY \`${foreignKey.name}\`, ADD CONSTRAINT \`${newForeignKeyName}\` FOREIGN KEY (${columnNames}) ` +
`REFERENCES ${this.escapePath(this.getTablePath(foreignKey))}(${referencedColumnNames})`;
if (foreignKey.onDelete)
up += ` ON DELETE ${foreignKey.onDelete}`;
if (foreignKey.onUpdate)
up += ` ON UPDATE ${foreignKey.onUpdate}`;
let down = `ALTER TABLE ${this.escapePath(newTable)} DROP FOREIGN KEY \`${newForeignKeyName}\`, ADD CONSTRAINT \`${foreignKey.name}\` FOREIGN KEY (${columnNames}) ` +
`REFERENCES ${this.escapePath(this.getTablePath(foreignKey))}(${referencedColumnNames})`;
if (foreignKey.onDelete)
down += ` ON DELETE ${foreignKey.onDelete}`;
if (foreignKey.onUpdate)
down += ` ON UPDATE ${foreignKey.onUpdate}`;
upQueries.push(new Query_1.Query(up));
downQueries.push(new Query_1.Query(down));
// replace constraint name
foreignKey.name = newForeignKeyName;
});
await this.executeQueries(upQueries, downQueries);
// rename old table and replace it in cached tabled;
oldTable.name = newTable.name;
this.replaceCachedTable(oldTable, newTable);
}
/**
* Change table comment.
*
* @param tableOrName
* @param newComment
*/
async changeTableComment(tableOrName, newComment) {
const upQueries = [];
const downQueries = [];
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
const escapedNewComment = this.escapeComment(newComment);
const escapedComment = this.escapeComment(table.comment);
if (escapedNewComment === escapedComment) {
return;
}
const newTable = table.clone();
newTable.comment = newComment;
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(newTable)} COMMENT ${escapedNewComment}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} COMMENT ${escapedComment}`));
await this.executeQueries(upQueries, downQueries);
// change table comment and replace it in cached tabled;
table.comment = newTable.comment;
this.replaceCachedTable(table, newTable);
}
/**
* Creates a new column from the column in the table.
*
* @param tableOrName
* @param column
*/
async addColumn(tableOrName, column) {
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
const clonedTable = table.clone();
const upQueries = [];
const downQueries = [];
const skipColumnLevelPrimary = clonedTable.primaryColumns.length > 0;
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD ${this.buildCreateColumnSql(column, skipColumnLevelPrimary, false)}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP COLUMN \`${column.name}\``));
// create or update primary key constraint
if (column.isPrimary && skipColumnLevelPrimary) {
// if we already have generated column, we must temporary drop AUTO_INCREMENT property.
const generatedColumn = clonedTable.columns.find((column) => column.isGenerated &&
column.generationStrategy === "increment");
if (generatedColumn) {
const nonGeneratedColumn = generatedColumn.clone();
nonGeneratedColumn.isGenerated = false;
nonGeneratedColumn.generationStrategy = undefined;
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${column.name}\` ${this.buildCreateColumnSql(nonGeneratedColumn, true)}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${nonGeneratedColumn.name}\` ${this.buildCreateColumnSql(column, true)}`));
}
const primaryColumns = clonedTable.primaryColumns;
let columnNames = primaryColumns
.map((column) => `\`${column.name}\``)
.join(", ");
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP PRIMARY KEY`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD PRIMARY KEY (${columnNames})`));
primaryColumns.push(column);
columnNames = primaryColumns
.map((column) => `\`${column.name}\``)
.join(", ");
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD PRIMARY KEY (${columnNames})`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP PRIMARY KEY`));
// if we previously dropped AUTO_INCREMENT property, we must bring it back
if (generatedColumn) {
const nonGeneratedColumn = generatedColumn.clone();
nonGeneratedColumn.isGenerated = false;
nonGeneratedColumn.generationStrategy = undefined;
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${nonGeneratedColumn.name}\` ${this.buildCreateColumnSql(column, true)}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${column.name}\` ${this.buildCreateColumnSql(nonGeneratedColumn, true)}`));
}
}
if (column.generatedType && column.asExpression) {
const currentDatabase = await this.getCurrentDatabase();
const insertQuery = this.insertTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: column.name,
value: column.asExpression,
});
const deleteQuery = this.deleteTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: column.name,
});
upQueries.push(insertQuery);
downQueries.push(deleteQuery);
}
// create column index
const columnIndex = clonedTable.indices.find((index) => index.columnNames.length === 1 &&
index.columnNames[0] === column.name);
if (columnIndex) {
upQueries.push(this.createIndexSql(table, columnIndex));
downQueries.push(this.dropIndexSql(table, columnIndex));
}
else if (column.isUnique) {
const uniqueIndex = new TableIndex_1.TableIndex({
name: this.dataSource.namingStrategy.indexName(table, [
column.name,
]),
columnNames: [column.name],
isUnique: true,
});
clonedTable.indices.push(uniqueIndex);
clonedTable.uniques.push(new TableUnique_1.TableUnique({
name: uniqueIndex.name,
columnNames: uniqueIndex.columnNames,
}));
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD UNIQUE INDEX \`${uniqueIndex.name}\` (\`${column.name}\`)`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP INDEX \`${uniqueIndex.name}\``));
}
await this.executeQueries(upQueries, downQueries);
clonedTable.addColumn(column);
this.replaceCachedTable(table, clonedTable);
}
/**
* Creates a new columns from the column in the table.
*
* @param tableOrName
* @param columns
*/
async addColumns(tableOrName, columns) {
for (const column of columns) {
await this.addColumn(tableOrName, column);
}
}
/**
* Renames column in the given table.
*
* @param tableOrName
* @param oldTableColumnOrName
* @param newTableColumnOrName
*/
async renameColumn(tableOrName, oldTableColumnOrName, newTableColumnOrName) {
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
const oldColumn = InstanceChecker_1.InstanceChecker.isTableColumn(oldTableColumnOrName)
? oldTableColumnOrName
: table.columns.find((c) => c.name === oldTableColumnOrName);
if (!oldColumn)
throw new error_1.TypeORMError(`Column "${oldTableColumnOrName}" was not found in the "${table.name}" table.`);
let newColumn;
if (InstanceChecker_1.InstanceChecker.isTableColumn(newTableColumnOrName)) {
newColumn = newTableColumnOrName;
}
else {
newColumn = oldColumn.clone();
newColumn.name = newTableColumnOrName;
}
await this.changeColumn(table, oldColumn, newColumn);
}
/**
* Changes a column in the table.
*
* @param tableOrName
* @param oldColumnOrName
* @param newColumn
*/
async changeColumn(tableOrName, oldColumnOrName, newColumn) {
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
let clonedTable = table.clone();
const upQueries = [];
const downQueries = [];
const oldColumn = InstanceChecker_1.InstanceChecker.isTableColumn(oldColumnOrName)
? oldColumnOrName
: table.columns.find((column) => column.name === oldColumnOrName);
if (!oldColumn)
throw new error_1.TypeORMError(`Column "${oldColumnOrName}" was not found in the "${table.name}" table.`);
if ((newColumn.isGenerated !== oldColumn.isGenerated &&
newColumn.generationStrategy !== "uuid") ||
oldColumn.type !== newColumn.type ||
oldColumn.length !== newColumn.length ||
(oldColumn.generatedType &&
newColumn.generatedType &&
oldColumn.generatedType !== newColumn.generatedType) ||
(!oldColumn.generatedType &&
newColumn.generatedType === "VIRTUAL") ||
(oldColumn.generatedType === "VIRTUAL" && !newColumn.generatedType)) {
await this.dropColumn(table, oldColumn);
await this.addColumn(table, newColumn);
// update cloned table
clonedTable = table.clone();
}
else {
if (newColumn.name !== oldColumn.name) {
// We don't change any column properties, just rename it.
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${oldColumn.name}\` \`${newColumn.name}\` ${this.buildCreateColumnSql(oldColumn, true, true)}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${newColumn.name}\` \`${oldColumn.name}\` ${this.buildCreateColumnSql(oldColumn, true, true)}`));
// rename index constraints
clonedTable.findColumnIndices(oldColumn).forEach((index) => {
const oldUniqueName = this.dataSource.namingStrategy.indexName(clonedTable, index.columnNames);
// Skip renaming if Index has user defined constraint name
if (index.name !== oldUniqueName)
return;
// build new constraint name
index.columnNames.splice(index.columnNames.indexOf(oldColumn.name), 1);
index.columnNames.push(newColumn.name);
const columnNames = index.columnNames
.map((column) => `\`${column}\``)
.join(", ");
const newIndexName = this.dataSource.namingStrategy.indexName(clonedTable, index.columnNames, index.where);
// build queries
let indexType = "";
if (index.isUnique)
indexType += "UNIQUE ";
if (index.isSpatial)
indexType += "SPATIAL ";
if (index.isFulltext)
indexType += "FULLTEXT ";
const indexParser = index.isFulltext && index.parser
? ` WITH PARSER ${index.parser}`
: "";
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP INDEX \`${index.name}\`, ADD ${indexType}INDEX \`${newIndexName}\` (${columnNames})${indexParser}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP INDEX \`${newIndexName}\`, ADD ${indexType}INDEX \`${index.name}\` (${columnNames})${indexParser}`));
// replace constraint name
index.name = newIndexName;
});
// rename foreign key constraints
clonedTable
.findColumnForeignKeys(oldColumn)
.forEach((foreignKey) => {
const foreignKeyName = this.dataSource.namingStrategy.foreignKeyName(clonedTable, foreignKey.columnNames, this.getTablePath(foreignKey), foreignKey.referencedColumnNames);
// Skip renaming if foreign key has user defined constraint name
if (foreignKey.name !== foreignKeyName)
return;
// build new constraint name
foreignKey.columnNames.splice(foreignKey.columnNames.indexOf(oldColumn.name), 1);
foreignKey.columnNames.push(newColumn.name);
const columnNames = foreignKey.columnNames
.map((column) => `\`${column}\``)
.join(", ");
const referencedColumnNames = foreignKey.referencedColumnNames
.map((column) => `\`${column}\``)
.join(",");
const newForeignKeyName = this.dataSource.namingStrategy.foreignKeyName(clonedTable, foreignKey.columnNames, this.getTablePath(foreignKey), foreignKey.referencedColumnNames);
// build queries
let up = `ALTER TABLE ${this.escapePath(table)} DROP FOREIGN KEY \`${foreignKey.name}\`, ADD CONSTRAINT \`${newForeignKeyName}\` FOREIGN KEY (${columnNames}) ` +
`REFERENCES ${this.escapePath(this.getTablePath(foreignKey))}(${referencedColumnNames})`;
if (foreignKey.onDelete)
up += ` ON DELETE ${foreignKey.onDelete}`;
if (foreignKey.onUpdate)
up += ` ON UPDATE ${foreignKey.onUpdate}`;
let down = `ALTER TABLE ${this.escapePath(table)} DROP FOREIGN KEY \`${newForeignKeyName}\`, ADD CONSTRAINT \`${foreignKey.name}\` FOREIGN KEY (${columnNames}) ` +
`REFERENCES ${this.escapePath(this.getTablePath(foreignKey))}(${referencedColumnNames})`;
if (foreignKey.onDelete)
down += ` ON DELETE ${foreignKey.onDelete}`;
if (foreignKey.onUpdate)
down += ` ON UPDATE ${foreignKey.onUpdate}`;
upQueries.push(new Query_1.Query(up));
downQueries.push(new Query_1.Query(down));
// replace constraint name
foreignKey.name = newForeignKeyName;
});
// rename old column in the Table object
const oldTableColumn = clonedTable.columns.find((column) => column.name === oldColumn.name);
clonedTable.columns[clonedTable.columns.indexOf(oldTableColumn)].name = newColumn.name;
oldColumn.name = newColumn.name;
}
if (this.isColumnChanged(oldColumn, newColumn, true, true)) {
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${oldColumn.name}\` ${this.buildCreateColumnSql(newColumn, true)}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${newColumn.name}\` ${this.buildCreateColumnSql(oldColumn, true)}`));
if (oldColumn.generatedType && !newColumn.generatedType) {
// if column changed from generated to non-generated, delete record from typeorm metadata
const currentDatabase = await this.getCurrentDatabase();
const deleteQuery = this.deleteTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: oldColumn.name,
});
const insertQuery = this.insertTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: oldColumn.name,
value: oldColumn.asExpression,
});
upQueries.push(deleteQuery);
downQueries.push(insertQuery);
}
else if (!oldColumn.generatedType &&
newColumn.generatedType) {
// if column changed from non-generated to generated, insert record into typeorm metadata
const currentDatabase = await this.getCurrentDatabase();
const insertQuery = this.insertTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: newColumn.name,
value: newColumn.asExpression,
});
const deleteQuery = this.deleteTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: newColumn.name,
});
upQueries.push(insertQuery);
downQueries.push(deleteQuery);
}
else if (oldColumn.asExpression !== newColumn.asExpression) {
// if only expression changed, just update it in typeorm_metadata table
const currentDatabase = await this.getCurrentDatabase();
const updateQuery = this.dataSource
.createQueryBuilder()
.update(this.getTypeormMetadataTableName())
.set({ value: newColumn.asExpression })
.where("`type` = :type", {
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
})
.andWhere("`name` = :name", { name: oldColumn.name })
.andWhere("`schema` = :schema", {
schema: currentDatabase,
})
.andWhere("`table` = :table", { table: table.name })
.getQueryAndParameters();
const revertUpdateQuery = this.dataSource
.createQueryBuilder()
.update(this.getTypeormMetadataTableName())
.set({ value: oldColumn.asExpression })
.where("`type` = :type", {
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
})
.andWhere("`name` = :name", { name: newColumn.name })
.andWhere("`schema` = :schema", {
schema: currentDatabase,
})
.andWhere("`table` = :table", { table: table.name })
.getQueryAndParameters();
upQueries.push(new Query_1.Query(updateQuery[0], updateQuery[1]));
downQueries.push(new Query_1.Query(revertUpdateQuery[0], revertUpdateQuery[1]));
}
}
if (newColumn.isPrimary !== oldColumn.isPrimary) {
// if table have generated column, we must drop AUTO_INCREMENT before changing primary constraints.
const generatedColumn = clonedTable.columns.find((column) => column.isGenerated &&
column.generationStrategy === "increment");
if (generatedColumn) {
const nonGeneratedColumn = generatedColumn.clone();
nonGeneratedColumn.isGenerated = false;
nonGeneratedColumn.generationStrategy = undefined;
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${generatedColumn.name}\` ${this.buildCreateColumnSql(nonGeneratedColumn, true)}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${nonGeneratedColumn.name}\` ${this.buildCreateColumnSql(generatedColumn, true)}`));
}
const primaryColumns = clonedTable.primaryColumns;
// if primary column state changed, we must always drop existed constraint.
if (primaryColumns.length > 0) {
const columnNames = primaryColumns
.map((column) => `\`${column.name}\``)
.join(", ");
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP PRIMARY KEY`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD PRIMARY KEY (${columnNames})`));
}
if (newColumn.isPrimary === true) {
primaryColumns.push(newColumn);
// update column in table
const column = clonedTable.columns.find((column) => column.name === newColumn.name);
column.isPrimary = true;
const columnNames = primaryColumns
.map((column) => `\`${column.name}\``)
.join(", ");
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD PRIMARY KEY (${columnNames})`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP PRIMARY KEY`));
}
else {
const primaryColumn = primaryColumns.find((c) => c.name === newColumn.name);
primaryColumns.splice(primaryColumns.indexOf(primaryColumn), 1);
// update column in table
const column = clonedTable.columns.find((column) => column.name === newColumn.name);
column.isPrimary = false;
// if we have another primary keys, we must recreate constraint.
if (primaryColumns.length > 0) {
const columnNames = primaryColumns
.map((column) => `\`${column.name}\``)
.join(", ");
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD PRIMARY KEY (${columnNames})`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP PRIMARY KEY`));
}
}
// if we have generated column, and we dropped AUTO_INCREMENT property before, we must bring it back
if (generatedColumn) {
const nonGeneratedColumn = generatedColumn.clone();
nonGeneratedColumn.isGenerated = false;
nonGeneratedColumn.generationStrategy = undefined;
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${nonGeneratedColumn.name}\` ${this.buildCreateColumnSql(generatedColumn, true)}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${generatedColumn.name}\` ${this.buildCreateColumnSql(nonGeneratedColumn, true)}`));
}
}
if (newColumn.isUnique !== oldColumn.isUnique) {
if (newColumn.isUnique === true) {
const uniqueIndex = new TableIndex_1.TableIndex({
name: this.dataSource.namingStrategy.indexName(table, [
newColumn.name,
]),
columnNames: [newColumn.name],
isUnique: true,
});
clonedTable.indices.push(uniqueIndex);
clonedTable.uniques.push(new TableUnique_1.TableUnique({
name: uniqueIndex.name,
columnNames: uniqueIndex.columnNames,
}));
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD UNIQUE INDEX \`${uniqueIndex.name}\` (\`${newColumn.name}\`)`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP INDEX \`${uniqueIndex.name}\``));
}
else {
const uniqueIndex = clonedTable.indices.find((index) => {
return (index.columnNames.length === 1 &&
index.isUnique === true &&
!!index.columnNames.find((columnName) => columnName === newColumn.name));
});
clonedTable.indices.splice(clonedTable.indices.indexOf(uniqueIndex), 1);
const tableUnique = clonedTable.uniques.find((unique) => unique.name === uniqueIndex.name);
clonedTable.uniques.splice(clonedTable.uniques.indexOf(tableUnique), 1);
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP INDEX \`${uniqueIndex.name}\``));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD UNIQUE INDEX \`${uniqueIndex.name}\` (\`${newColumn.name}\`)`));
}
}
}
await this.executeQueries(upQueries, downQueries);
this.replaceCachedTable(table, clonedTable);
}
/**
* Changes a column in the table.
*
* @param tableOrName
* @param changedColumns
*/
async changeColumns(tableOrName, changedColumns) {
for (const { oldColumn, newColumn } of changedColumns) {
await this.changeColumn(tableOrName, oldColumn, newColumn);
}
}
/**
* Drops column in the table.
*
* @param tableOrName
* @param columnOrName
* @param ifExists
*/
async dropColumn(tableOrName, columnOrName, ifExists) {
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
const column = InstanceChecker_1.InstanceChecker.isTableColumn(columnOrName)
? columnOrName
: table.findColumnByName(columnOrName);
if (!column) {
if (ifExists)
return;
throw new error_1.TypeORMError(`Column "${columnOrName}" was not found in table "${table.name}"`);
}
const clonedTable = table.clone();
const upQueries = [];
const downQueries = [];
// drop primary key constraint
if (column.isPrimary) {
// if table have generated column, we must drop AUTO_INCREMENT before changing primary constraints.
const generatedColumn = clonedTable.columns.find((column) => column.isGenerated &&
column.generationStrategy === "increment");
if (generatedColumn) {
const nonGeneratedColumn = generatedColumn.clone();
nonGeneratedColumn.isGenerated = false;
nonGeneratedColumn.generationStrategy = undefined;
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${generatedColumn.name}\` ${this.buildCreateColumnSql(nonGeneratedColumn, true)}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${nonGeneratedColumn.name}\` ${this.buildCreateColumnSql(generatedColumn, true)}`));
}
// dropping primary key constraint
const columnNames = clonedTable.primaryColumns
.map((primaryColumn) => `\`${primaryColumn.name}\``)
.join(", ");
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(clonedTable)} DROP PRIMARY KEY`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(clonedTable)} ADD PRIMARY KEY (${columnNames})`));
// update column in table
const tableColumn = clonedTable.findColumnByName(column.name);
tableColumn.isPrimary = false;
// if primary key have multiple columns, we must recreate it without dropped column
if (clonedTable.primaryColumns.length > 0) {
const columnNames = clonedTable.primaryColumns
.map((primaryColumn) => `\`${primaryColumn.name}\``)
.join(", ");
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(clonedTable)} ADD PRIMARY KEY (${columnNames})`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(clonedTable)} DROP PRIMARY KEY`));
}
// if we have generated column, and we dropped AUTO_INCREMENT property before, and this column is not current dropping column, we must bring it back
if (generatedColumn && generatedColumn.name !== column.name) {
const nonGeneratedColumn = generatedColumn.clone();
nonGeneratedColumn.isGenerated = false;
nonGeneratedColumn.generationStrategy = undefined;
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${nonGeneratedColumn.name}\` ${this.buildCreateColumnSql(generatedColumn, true)}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${generatedColumn.name}\` ${this.buildCreateColumnSql(nonGeneratedColumn, true)}`));
}
}
// drop column index
const columnIndex = clonedTable.indices.find((index) => index.columnNames.length === 1 &&
index.columnNames[0] === column.name);
if (columnIndex) {
clonedTable.indices.splice(clonedTable.indices.indexOf(columnIndex), 1);
upQueries.push(this.dropIndexSql(table, columnIndex));
downQueries.push(this.createIndexSql(table, columnIndex));
}
else if (column.isUnique) {
// we splice constraints both from table uniques and indices.
const uniqueName = this.dataSource.namingStrategy.uniqueConstraintName(table, [
column.name,
]);
const foundUnique = clonedTable.uniques.find((unique) => unique.name === uniqueName);
if (foundUnique)
clonedTable.uniques.splice(clonedTable.uniques.indexOf(foundUnique), 1);
const indexName = this.dataSource.namingStrategy.indexName(table, [
column.name,
]);
const foundIndex = clonedTable.indices.find((index) => index.name === indexName);
if (foundIndex)
clonedTable.indices.splice(clonedTable.indices.indexOf(foundIndex), 1);
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP INDEX \`${indexName}\``));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD UNIQUE INDEX \`${indexName}\` (\`${column.name}\`)`));
}
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP COLUMN \`${column.name}\``));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD ${this.buildCreateColumnSql(column, true)}`));
if (column.generatedType && column.asExpression) {
const currentDatabase = await this.getCurrentDatabase();
const deleteQuery = this.deleteTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: column.name,
});
const insertQuery = this.insertTypeormMetadataSql({
schema: currentDatabase,
table: table.name,
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: column.name,
value: column.asExpression,
});
upQueries.push(deleteQuery);
downQueries.push(insertQuery);
}
await this.executeQueries(upQueries, downQueries);
clonedTable.removeColumn(column);
this.replaceCachedTable(table, clonedTable);
}
/**
* Drops the columns in the table.
*
* @param tableOrName
* @param columns
* @param ifExists
*/
async dropColumns(tableOrName, columns, ifExists) {
for (const column of [...columns]) {
await this.dropColumn(tableOrName, column, ifExists);
}
}
/**
* Creates a new primary key.
*
* @param tableOrName
* @param columnNames
*/
async createPrimaryKey(tableOrName, columnNames) {
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
const clonedTable = table.clone();
const up = this.createPrimaryKeySql(table, columnNames);
const down = this.dropPrimaryKeySql(table);
await this.executeQueries(up, down);
clonedTable.columns.forEach((column) => {
if (columnNames.find((columnName) => columnName === column.name))
column.isPrimary = true;
});
this.replaceCachedTable(table, clonedTable);
}
/**
* Updates composite primary keys.
*
* @param tableOrName
* @param columns
*/
async updatePrimaryKeys(tableOrName, columns) {
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
const clonedTable = table.clone();
const columnNames = columns.map((column) => column.name);
const upQueries = [];
const downQueries = [];
// if table have generated column, we must drop AUTO_INCREMENT before changing primary constraints.
const generatedColumn = clonedTable.columns.find((column) => column.isGenerated && column.generationStrategy === "increment");
if (generatedColumn) {
const nonGeneratedColumn = generatedColumn.clone();
nonGeneratedColumn.isGenerated = false;
nonGeneratedColumn.generationStrategy = undefined;
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${generatedColumn.name}\` ${this.buildCreateColumnSql(nonGeneratedColumn, true)}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${nonGeneratedColumn.name}\` ${this.buildCreateColumnSql(generatedColumn, true)}`));
}
// if table already have primary columns, we must drop them.
const primaryColumns = clonedTable.primaryColumns;
if (primaryColumns.length > 0) {
const columnNames = primaryColumns
.map((column) => `\`${column.name}\``)
.join(", ");
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP PRIMARY KEY`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD PRIMARY KEY (${columnNames})`));
}
// update columns in table.
clonedTable.columns
.filter((column) => columnNames.indexOf(column.name) !== -1)
.forEach((column) => {
column.isPrimary = true;
});
const columnNamesString = columnNames
.map((columnName) => `\`${columnName}\``)
.join(", ");
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD PRIMARY KEY (${columnNamesString})`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP PRIMARY KEY`));
// if we already have generated column or column is changed to generated, and we dropped AUTO_INCREMENT property before, we must bring it back
const newOrExistGeneratedColumn = generatedColumn ??
columns.find((column) => column.isGenerated &&
column.generationStrategy === "increment");
if (newOrExistGeneratedColumn) {
const nonGeneratedColumn = newOrExistGeneratedColumn.clone();
nonGeneratedColumn.isGenerated = false;
nonGeneratedColumn.generationStrategy = undefined;
upQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${nonGeneratedColumn.name}\` ${this.buildCreateColumnSql(newOrExistGeneratedColumn, true)}`));
downQueries.push(new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} CHANGE \`${newOrExistGeneratedColumn.name}\` ${this.buildCreateColumnSql(nonGeneratedColumn, true)}`));
// if column changed to generated, we must update it in table
const changedGeneratedColumn = clonedTable.columns.find((column) => column.name === newOrExistGeneratedColumn.name);
changedGeneratedColumn.isGenerated = true;
changedGeneratedColumn.generationStrategy = "increment";
}
await this.executeQueries(upQueries, downQueries);
this.replaceCachedTable(table, clonedTable);
}
/**
* Drops a primary key.
*
* @param tableOrName
* @param constraintName not used in MySQL
* @param ifExists
*/
async dropPrimaryKey(tableOrName, constraintName, ifExists) {
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
if (table.primaryColumns.length === 0) {
if (ifExists)
return;
}
const up = this.dropPrimaryKeySql(table);
const down = this.createPrimaryKeySql(table, table.primaryColumns.map((column) => column.name));
await this.executeQueries(up, down);
table.primaryColumns.forEach((column) => {
column.isPrimary = false;
});
}
/**
* Creates a new unique constraint.
*
* @param tableOrName
* @param uniqueConstraint
*/
async createUniqueConstraint(tableOrName, uniqueConstraint) {
throw new error_1.TypeORMError(`MySql does not support unique constraints. Use unique index instead.`);
}
/**
* Creates a new unique constraints.
*
* @param tableOrName
* @param uniqueConstraints
*/
async createUniqueConstraints(tableOrName, uniqueConstraints) {
throw new error_1.TypeORMError(`MySql does not support unique constraints. Use unique index instead.`);
}
/**
* Drops a unique constraint.
*
* @param tableOrName
* @param uniqueOrName
* @param ifExists
*/
async dropUniqueConstraint(tableOrName, uniqueOrName, ifExists) {
if (ifExists)
return;
throw new error_1.TypeORMError(`MySql does not support unique constraints. Use unique index instead.`);
}
/**
* Drops a unique constraints.
*
* @param tableOrName
* @param uniqueConstraints
* @param ifExists
*/
async dropUniqueConstraints(tableOrName, uniqueConstraints, ifExists) {
if (ifExists)
return;
throw new error_1.TypeORMError(`MySql does not support unique constraints. Use unique index instead.`);
}
/**
* Creates a new check constraint.
*
* @param tableOrName
* @param checkConstraint
*/
async createCheckConstraint(tableOrName, checkConstraint) {
throw new error_1.TypeORMError(`MySql does not support check constraints.`);
}
/**
* Creates a new check constraints.
*
* @param tableOrName
* @param checkConstraints
*/
async createCheckConstraints(tableOrName, checkConstraints) {
throw new error_1.TypeORMError(`MySql does not support check constraints.`);
}
/**
* Drops check constraint.
*
* @param tableOrName
* @param checkOrName
* @param ifExists
*/
async dropCheckConstraint(tableOrName, checkOrName, ifExists) {
throw new error_1.TypeORMError(`MySql does not support check constraints.`);
}
/**
* Drops check constraints.
*
* @param tableOrName
* @param checkConstraints
* @param ifExists
*/
async dropCheckConstraints(tableOrName, checkConstraints, ifExists) {
throw new error_1.TypeORMError(`MySql does not support check constraints.`);
}
/**
* Creates a new exclusion constraint.
*
* @param tableOrName
* @param exclusionConstraint
*/
async createExclusionConstraint(tableOrName, exclusionConstraint) {
throw new error_1.TypeORMError(`MySql does not support exclusion constraints.`);
}
/**
* Creates a new exclusion constraints.
*
* @param tableOrName
* @param exclusionConstraints
*/
async createExclusionConstraints(tableOrName, exclusionConstraints) {
throw new error_1.TypeORMError(`MySql does not support exclusion constraints.`);
}
/**
* Drops exclusion constraint.
*
* @param tableOrName
* @param exclusionOrName
* @param ifExists
*/
async dropExclusionConstraint(tableOrName, exclusionOrName, ifExists) {
throw new error_1.TypeORMError(`MySql does not support exclusion constraints.`);
}
/**
* Drops exclusion constraints.
*
* @param tableOrName
* @param exclusionConstraints
* @param ifExists
*/
async dropExclusionConstraints(tableOrName, exclusionConstraints, ifExists) {
throw new error_1.TypeORMError(`MySql does not support exclusion constraints.`);
}
/**
* Creates a new foreign key.
*
* @param tableOrName
* @param foreignKey
*/
async createForeignKey(tableOrName, foreignKey) {
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
// new FK may be passed without name. In this case we generate FK name manually.
foreignKey.name ??= this.dataSource.namingStrategy.foreignKeyName(table, foreignKey.columnNames, this.getTablePath(foreignKey), foreignKey.referencedColumnNames);
const up = this.createForeignKeySql(table, foreignKey);
const down = this.dropForeignKeySql(table, foreignKey);
await this.executeQueries(up, down);
table.addForeignKey(foreignKey);
}
/**
* Creates a new foreign keys.
*
* @param tableOrName
* @param foreignKeys
*/
async createForeignKeys(tableOrName, foreignKeys) {
const promises = foreignKeys.map((foreignKey) => this.createForeignKey(tableOrName, foreignKey));
await Promise.all(promises);
}
/**
* Drops a foreign key.
*
* @param tableOrName
* @param foreignKeyOrName
* @param ifExists
*/
async dropForeignKey(tableOrName, foreignKeyOrName, ifExists) {
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
const foreignKey = InstanceChecker_1.InstanceChecker.isTableForeignKey(foreignKeyOrName)
? foreignKeyOrName
: table.foreignKeys.find((fk) => fk.name === foreignKeyOrName);
if (!foreignKey) {
if (ifExists)
return;
throw new error_1.TypeORMError(`Supplied foreign key was not found in table ${table.name}`);
}
foreignKey.name ??= this.dataSource.namingStrategy.foreignKeyName(table, foreignKey.columnNames, this.getTablePath(foreignKey), foreignKey.referencedColumnNames);
const up = this.dropForeignKeySql(table, foreignKey);
const down = this.createForeignKeySql(table, foreignKey);
await this.executeQueries(up, down);
table.removeForeignKey(foreignKey);
}
/**
* Drops a foreign keys from the table.
*
* @param tableOrName
* @param foreignKeys
* @param ifExists
*/
async dropForeignKeys(tableOrName, foreignKeys, ifExists) {
const promises = foreignKeys.map((foreignKey) => this.dropForeignKey(tableOrName, foreignKey, ifExists));
await Promise.all(promises);
}
/**
* Creates a new index.
*
* @param tableOrName
* @param index
*/
async createIndex(tableOrName, index) {
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
// new index may be passed without name. In this case we generate index name manually.
index.name ??= this.generateIndexName(table, index);
const up = this.createIndexSql(table, index);
const down = this.dropIndexSql(table, index);
await this.executeQueries(up, down);
table.addIndex(index, true);
}
/**
* Creates a new indices
*
* @param tableOrName
* @param indices
*/
async createIndices(tableOrName, indices) {
const promises = indices.map((index) => this.createIndex(tableOrName, index));
await Promise.all(promises);
}
/**
* Drops an index.
*
* @param tableOrName
* @param indexOrName
* @param ifExists
*/
async dropIndex(tableOrName, indexOrName, ifExists) {
const table = InstanceChecker_1.InstanceChecker.isTable(tableOrName)
? tableOrName
: await this.getCachedTable(tableOrName);
const index = InstanceChecker_1.InstanceChecker.isTableIndex(indexOrName)
? indexOrName
: table.indices.find((i) => i.name === indexOrName);
if (!index) {
if (ifExists)
return;
throw new error_1.TypeORMError(`Supplied index ${indexOrName} was not found in table ${table.name}`);
}
// old index may be passed without name. In this case we generate index name manually.
index.name ??= this.generateIndexName(table, index);
const up = this.dropIndexSql(table, index);
const down = this.createIndexSql(table, index);
await this.executeQueries(up, down);
table.removeIndex(index, true);
}
/**
* Drops an indices from the table.
*
* @param tableOrName
* @param indices
* @param ifExists
*/
async dropIndices(tableOrName, indices, ifExists) {
const promises = indices.map((index) => this.dropIndex(tableOrName, index, ifExists));
await Promise.all(promises);
}
/**
* Clears all table contents.
* Note: this operation uses SQL's TRUNCATE query which cannot be reverted in transactions.
*
* @param tableOrName
* @param options
* @param options.cascade
*/
async clearTable(tableOrName, options) {
if (options?.cascade) {
throw new error_1.TypeORMError(`MySql does not support clearing table with cascade option`);
}
await this.query(`TRUNCATE TABLE ${this.escapePath(tableOrName)}`);
}
/**
* Removes all tables from the currently connected database.
* Be careful using this method and avoid using it in production or migrations
* (because it can clear all your database).
*
* @param database
*/
async clearDatabase(database) {
const dbName = database ?? this.driver.database;
if (dbName) {
const isDatabaseExist = await this.hasDatabase(dbName);
if (!isDatabaseExist)
return Promise.resolve();
}
else {
throw new error_1.TypeORMError(`Can not clear database. No database is specified`);
}
const isAnotherTransactionActive = this.isTransactionActive;
if (!isAnotherTransactionActive)
await this.startTransaction();
try {
const selectViewDropsQuery = `SELECT concat('DROP VIEW IF EXISTS \`', table_schema, '\`.\`', table_name, '\`') AS \`query\` FROM \`INFORMATION_SCHEMA\`.\`VIEWS\` WHERE \`TABLE_SCHEMA\` = ?`;
const dropViewQueries = await this.query(selectViewDropsQuery, [dbName]);
await Promise.all(dropViewQueries.map((q) => this.query(q["query"])));
const disableForeignKeysCheckQuery = `SET FOREIGN_KEY_CHECKS = 0;`;
const dropTablesQuery = `SELECT concat('DROP TABLE IF EXISTS \`', table_schema, '\`.\`', table_name, '\`') AS \`query\` FROM \`INFORMATION_SCHEMA\`.\`TABLES\` WHERE \`TABLE_SCHEMA\` = ?`;
const enableForeignKeysCheckQuery = `SET FOREIGN_KEY_CHECKS = 1;`;
await this.query(disableForeignKeysCheckQuery);
const dropQueries = await this.query(dropTablesQuery, [dbName]);
await Promise.all(dropQueries.map((query) => this.query(query["query"])));
await this.query(enableForeignKeysCheckQuery);
if (!isAnotherTransactionActive)
await this.commitTransaction();
}
catch (error) {
try {
// we throw original error even if rollback thrown an error
if (!isAnotherTransactionActive)
await this.rollbackTransaction();
}
catch (rollbackError) { }
throw error;
}
}
// -------------------------------------------------------------------------
// Protected Methods
// -------------------------------------------------------------------------
async loadViews(viewNames) {
const hasTable = await this.hasTable(this.getTypeormMetadataTableName());
if (!hasTable) {
return [];
}
viewNames ??= [];
const currentDatabase = await this.getCurrentDatabase();
const viewsCondition = viewNames
.map((tableName) => {
let { database, tableName: name } = this.driver.parseTableName(tableName);
database ??= currentDatabase;
return `(\`t\`.\`schema\` = '${database}' AND \`t\`.\`name\` = '${name}')`;
})
.join(" OR ");
const query = `SELECT \`t\`.*, \`v\`.\`check_option\` FROM ${this.escapePath(this.getTypeormMetadataTableName())} \`t\` ` +
`INNER JOIN \`information_schema\`.\`views\` \`v\` ON \`v\`.\`table_schema\` = \`t\`.\`schema\` AND \`v\`.\`table_name\` = \`t\`.\`name\` WHERE \`t\`.\`type\` = '${MetadataTableType_1.MetadataTableType.VIEW}' ${viewsCondition ? `AND (${viewsCondition})` : ""}`;
const dbViews = await this.query(query);
return dbViews.map((dbView) => {
const view = new View_1.View();
const db = dbView["schema"] === currentDatabase
? undefined
: dbView["schema"];
view.database = dbView["schema"];
view.name = this.driver.buildTableName(dbView["name"], undefined, db);
view.expression = dbView["value"];
return view;
});
}
/**
* Loads all tables (with given names) from the database and creates a Table from them.
*
* @param tableNames
*/
async loadTables(tableNames) {
if (tableNames?.length === 0) {
return [];
}
const currentDatabase = await this.getCurrentDatabase();
// The following SQL brought to you by:
// A terrible understanding of https://dev.mysql.com/doc/refman/8.0/en/information-schema-optimization.html
//
// Short Version:
// INFORMATION_SCHEMA is a weird metadata virtual table and follows VERY FEW of the normal
// query optimization rules. Depending on the columns you query against & the columns you're SELECTing
// there can be a drastically different query performance - this is because the tables map to
// data on the disk and some pieces of data require a scan of the data directory, the database files, etc
// With most of these, you'll want to do an `EXPLAIN` when making changes to make sure
// the changes you're making aren't changing the query performance profile negatively
// When you do the explain you'll want to look at the `Extra` field -
// It will look something like: "Using where; {FILE_OPENING}; Scanned {DB_NUM} databases"
// FILE_OPENING will commonly be OPEN_FRM_ONLY or OPEN_FULL_TABLE - you want to aim to NOT do
// an OPEN_FULL_TABLE unless necessary. DB_NUM may be a number or "all" - you really want to
// keep this to 0 or 1. Ideally 0. "All" means you've scanned all databases - not good.
//
// For more info, see the above link to the MySQL docs.
//
// Something not noted in the docs is that complex `WHERE` clauses - such as `OR` expressions -
// will cause the query to not hit the optimizations & do full scans. This is why
// a number of queries below do `UNION`s of single `WHERE` clauses.
const dbTables = [];
if (!tableNames) {
// Since we don't have any of this data we have to do a scan
const tablesSql = `SELECT \`TABLE_SCHEMA\`, \`TABLE_NAME\`, \`TABLE_COMMENT\` FROM \`INFORMATION_SCHEMA\`.\`TABLES\``;
dbTables.push(...(await this.query(tablesSql)));
}
else {
// Avoid data directory scan: TABLE_SCHEMA
// Avoid database directory scan: TABLE_NAME
// We only use `TABLE_SCHEMA` and `TABLE_NAME` which is `SKIP_OPEN_TABLE`
const tablesSql = tableNames
.filter((tableName) => tableName)
.map((tableName) => {
let { database, tableName: name } = this.driver.parseTableName(tableName);
database ??= currentDatabase;
return `SELECT \`TABLE_SCHEMA\`, \`TABLE_NAME\`, \`TABLE_COMMENT\` FROM \`INFORMATION_SCHEMA\`.\`TABLES\` WHERE \`TABLE_SCHEMA\` = '${database}' AND \`TABLE_NAME\` = '${name}'`;
})
.join(" UNION ");
dbTables.push(...(await this.query(tablesSql)));
}
// if tables were not found in the db, no need to proceed
if (!dbTables.length)
return [];
// Avoid data directory scan: TABLE_SCHEMA
// Avoid database directory scan: TABLE_NAME
// Full columns: CARDINALITY & INDEX_TYPE - everything else is FRM only
const statsSubquerySql = dbTables
.map(({ TABLE_SCHEMA, TABLE_NAME }) => {
return (`SELECT * FROM \`INFORMATION_SCHEMA\`.\`STATISTICS\` ` +
`WHERE ` +
`\`TABLE_SCHEMA\` = '${TABLE_SCHEMA}' ` +
`AND ` +
`\`TABLE_NAME\` = '${TABLE_NAME}'`);
})
.join(" UNION ");
// Avoid data directory scan: TABLE_SCHEMA
// Avoid database directory scan: TABLE_NAME
// All columns will hit the full table.
const kcuSubquerySql = dbTables
.map(({ TABLE_SCHEMA, TABLE_NAME }) => {
return (`SELECT * FROM \`INFORMATION_SCHEMA\`.\`KEY_COLUMN_USAGE\` \`kcu\` ` +
`WHERE ` +
`\`kcu\`.\`TABLE_SCHEMA\` = '${TABLE_SCHEMA}' ` +
`AND ` +
`\`kcu\`.\`TABLE_NAME\` = '${TABLE_NAME}'`);
})
.join(" UNION ");
// Avoid data directory scan: CONSTRAINT_SCHEMA
// Avoid database directory scan: TABLE_NAME
// All columns will hit the full table.
const rcSubquerySql = dbTables
.map(({ TABLE_SCHEMA, TABLE_NAME }) => {
return (`SELECT * FROM \`INFORMATION_SCHEMA\`.\`REFERENTIAL_CONSTRAINTS\` ` +
`WHERE ` +
`\`CONSTRAINT_SCHEMA\` = '${TABLE_SCHEMA}' ` +
`AND ` +
`\`TABLE_NAME\` = '${TABLE_NAME}'`);
})
.join(" UNION ");
// Avoid data directory scan: TABLE_SCHEMA
// Avoid database directory scan: TABLE_NAME
// OPEN_FRM_ONLY applies to all columns
const columnsSql = dbTables
.map(({ TABLE_SCHEMA, TABLE_NAME }) => {
return (`SELECT * FROM \`INFORMATION_SCHEMA\`.\`COLUMNS\` ` +
`WHERE ` +
`\`TABLE_SCHEMA\` = '${TABLE_SCHEMA}' ` +
`AND ` +
`\`TABLE_NAME\` = '${TABLE_NAME}'`);
})
.join(" UNION ");
// No Optimizations are available for COLLATIONS
const collationsSql = `SELECT ` +
`\`SCHEMA_NAME\`, ` +
`\`DEFAULT_CHARACTER_SET_NAME\` as \`CHARSET\`, ` +
`\`DEFAULT_COLLATION_NAME\` AS \`COLLATION\` ` +
`FROM \`INFORMATION_SCHEMA\`.\`SCHEMATA\``;
// Key Column Usage but only for PKs
const primaryKeySql = `SELECT * FROM (${kcuSubquerySql}) \`kcu\` WHERE \`CONSTRAINT_NAME\` = 'PRIMARY'`;
// Combine stats & referential constraints
const indicesSql = `SELECT ` +
`\`s\`.* ` +
`FROM (${statsSubquerySql}) \`s\` ` +
`LEFT JOIN (${rcSubquerySql}) \`rc\` ` +
`ON ` +
`\`s\`.\`INDEX_NAME\` = \`rc\`.\`CONSTRAINT_NAME\` ` +
`AND ` +
`\`s\`.\`TABLE_SCHEMA\` = \`rc\`.\`CONSTRAINT_SCHEMA\` ` +
`WHERE ` +
`\`s\`.\`INDEX_NAME\` != 'PRIMARY' ` +
`AND ` +
`\`rc\`.\`CONSTRAINT_NAME\` IS NULL`;
// Combine Key Column Usage & Referential Constraints
const foreignKeysSql = `SELECT ` +
`\`kcu\`.\`TABLE_SCHEMA\`, ` +
`\`kcu\`.\`TABLE_NAME\`, ` +
`\`kcu\`.\`CONSTRAINT_NAME\`, ` +
`\`kcu\`.\`COLUMN_NAME\`, ` +
`\`kcu\`.\`REFERENCED_TABLE_SCHEMA\`, ` +
`\`kcu\`.\`REFERENCED_TABLE_NAME\`, ` +
`\`kcu\`.\`REFERENCED_COLUMN_NAME\`, ` +
`\`rc\`.\`DELETE_RULE\` \`ON_DELETE\`, ` +
`\`rc\`.\`UPDATE_RULE\` \`ON_UPDATE\` ` +
`FROM (${kcuSubquerySql}) \`kcu\` ` +
`INNER JOIN (${rcSubquerySql}) \`rc\` ` +
`ON ` +
`\`rc\`.\`CONSTRAINT_SCHEMA\` = \`kcu\`.\`CONSTRAINT_SCHEMA\` ` +
`AND ` +
`\`rc\`.\`TABLE_NAME\` = \`kcu\`.\`TABLE_NAME\` ` +
`AND ` +
`\`rc\`.\`CONSTRAINT_NAME\` = \`kcu\`.\`CONSTRAINT_NAME\``;
const [dbColumns, dbPrimaryKeys, dbCollations, dbIndices, dbForeignKeys,] = await Promise.all([
this.query(columnsSql),
this.query(primaryKeySql),
this.query(collationsSql),
this.query(indicesSql),
this.query(foreignKeysSql),
]);
const isMariaDb = this.driver.options.type === "mariadb";
const dbVersion = this.driver.version;
// create tables for loaded tables
return Promise.all(dbTables.map(async (dbTable) => {
const table = new Table_1.Table();
const dbCollation = dbCollations.find((coll) => coll["SCHEMA_NAME"] === dbTable["TABLE_SCHEMA"]);
const defaultCollation = dbCollation["COLLATION"];
const defaultCharset = dbCollation["CHARSET"];
// We do not need to join database name, when database is by default.
const db = dbTable["TABLE_SCHEMA"] === currentDatabase
? undefined
: dbTable["TABLE_SCHEMA"];
table.database = dbTable["TABLE_SCHEMA"];
table.name = this.driver.buildTableName(dbTable["TABLE_NAME"], undefined, db);
// create columns from the loaded columns
table.columns = await Promise.all(dbColumns
.filter((dbColumn) => dbColumn["TABLE_NAME"] ===
dbTable["TABLE_NAME"] &&
dbColumn["TABLE_SCHEMA"] ===
dbTable["TABLE_SCHEMA"])
.map(async (dbColumn) => {
const columnUniqueIndices = dbIndices.filter((dbIndex) => {
return (dbIndex["TABLE_NAME"] ===
dbTable["TABLE_NAME"] &&
dbIndex["TABLE_SCHEMA"] ===
dbTable["TABLE_SCHEMA"] &&
dbIndex["COLUMN_NAME"] ===
dbColumn["COLUMN_NAME"] &&
parseInt(dbIndex["NON_UNIQUE"], 10) ===
0);
});
const tableMetadata = this.dataSource.entityMetadatas.find((metadata) => this.getTablePath(table) ===
this.getTablePath(metadata));
const hasIgnoredIndex = columnUniqueIndices.length > 0 &&
tableMetadata?.indices.some((index) => {
return columnUniqueIndices.some((uniqueIndex) => {
return (index.name ===
uniqueIndex["INDEX_NAME"] &&
index.synchronize === false);
});
});
const isConstraintComposite = columnUniqueIndices.every((uniqueIndex) => {
return dbIndices.some((dbIndex) => dbIndex["INDEX_NAME"] ===
uniqueIndex["INDEX_NAME"] &&
dbIndex["COLUMN_NAME"] !==
dbColumn["COLUMN_NAME"]);
});
const tableColumn = new TableColumn_1.TableColumn();
tableColumn.name = dbColumn["COLUMN_NAME"];
tableColumn.type =
dbColumn["DATA_TYPE"].toLowerCase();
// since mysql 8.0, "geometrycollection" returned as "geomcollection"
// typeorm still use "geometrycollection"
if (tableColumn.type === "geomcollection") {
tableColumn.type = "geometrycollection";
}
tableColumn.unsigned =
dbColumn["COLUMN_TYPE"].includes("unsigned");
if (dbColumn["COLUMN_DEFAULT"] === null ||
dbColumn["COLUMN_DEFAULT"] === undefined ||
(isMariaDb &&
dbColumn["COLUMN_DEFAULT"] === "NULL")) {
tableColumn.default = undefined;
}
else if (/^CURRENT_TIMESTAMP(\([0-9]*\))?$/i.test(dbColumn["COLUMN_DEFAULT"])) {
// New versions of MariaDB return expressions in lowercase. We need to set it in
// uppercase so the comparison in MysqlDriver#compareDefaultValues does not fail.
tableColumn.default =
dbColumn["COLUMN_DEFAULT"].toUpperCase();
}
else if (isMariaDb &&
VersionUtils_1.VersionUtils.isGreaterOrEqual(dbVersion, "10.2.7")) {
// MariaDB started adding quotes to literals in COLUMN_DEFAULT since version 10.2.7
// See https://mariadb.com/kb/en/library/information-schema-columns-table/
tableColumn.default = dbColumn["COLUMN_DEFAULT"];
}
else {
tableColumn.default = `'${dbColumn["COLUMN_DEFAULT"]}'`;
}
if (dbColumn["EXTRA"].indexOf("on update") !== -1) {
// New versions of MariaDB return expressions in lowercase. We need to set it in
// uppercase so the comparison in MysqlDriver#compareExtraValues does not fail.
tableColumn.onUpdate = dbColumn["EXTRA"]
.slice(dbColumn["EXTRA"].indexOf("on update") +
10)
.toUpperCase();
}
if (dbColumn["GENERATION_EXPRESSION"]) {
tableColumn.generatedType =
dbColumn["EXTRA"].indexOf("VIRTUAL") !== -1
? "VIRTUAL"
: "STORED";
// We cannot relay on information_schema.columns.generation_expression, because it is formatted different.
const asExpressionQuery = this.selectTypeormMetadataSql({
schema: dbTable["TABLE_SCHEMA"],
table: dbTable["TABLE_NAME"],
type: MetadataTableType_1.MetadataTableType.GENERATED_COLUMN,
name: tableColumn.name,
});
const results = await this.query(asExpressionQuery.query, asExpressionQuery.parameters);
if (results[0]?.value) {
tableColumn.asExpression = results[0].value;
}
else {
tableColumn.asExpression = "";
}
}
tableColumn.isUnique =
columnUniqueIndices.length > 0 &&
!hasIgnoredIndex &&
!isConstraintComposite;
if (isMariaDb && tableColumn.generatedType) {
// do nothing - MariaDB does not support NULL/NOT NULL expressions for generated columns
}
else {
tableColumn.isNullable =
dbColumn["IS_NULLABLE"] === "YES";
}
tableColumn.isPrimary = dbPrimaryKeys.some((dbPrimaryKey) => {
return (dbPrimaryKey["TABLE_NAME"] ===
dbColumn["TABLE_NAME"] &&
dbPrimaryKey["TABLE_SCHEMA"] ===
dbColumn["TABLE_SCHEMA"] &&
dbPrimaryKey["COLUMN_NAME"] ===
dbColumn["COLUMN_NAME"]);
});
tableColumn.isGenerated =
dbColumn["EXTRA"].indexOf("auto_increment") !==
-1;
if (tableColumn.isGenerated)
tableColumn.generationStrategy = "increment";
tableColumn.comment =
typeof dbColumn["COLUMN_COMMENT"] ===
"string" &&
dbColumn["COLUMN_COMMENT"].length === 0
? undefined
: dbColumn["COLUMN_COMMENT"];
if (dbColumn["CHARACTER_SET_NAME"])
tableColumn.charset =
dbColumn["CHARACTER_SET_NAME"] ===
defaultCharset
? undefined
: dbColumn["CHARACTER_SET_NAME"];
if (dbColumn["COLLATION_NAME"])
tableColumn.collation =
dbColumn["COLLATION_NAME"] ===
defaultCollation
? undefined
: dbColumn["COLLATION_NAME"];
// check only columns that have length property
if (this.driver.withLengthColumnTypes.indexOf(tableColumn.type) !== -1 &&
dbColumn["CHARACTER_MAXIMUM_LENGTH"]) {
let length = dbColumn["CHARACTER_MAXIMUM_LENGTH"];
if (tableColumn.type === "vector") {
// MySQL and MariaDb store the vector length in bytes, not in number of dimensions.
length = length / 4;
}
tableColumn.length =
!this.isDefaultColumnLength(table, tableColumn, length.toString())
? length.toString()
: "";
}
if (tableColumn.type === "decimal" ||
tableColumn.type === "double" ||
tableColumn.type === "float") {
if (dbColumn["NUMERIC_PRECISION"] !== null &&
!this.isDefaultColumnPrecision(table, tableColumn, dbColumn["NUMERIC_PRECISION"]))
tableColumn.precision = parseInt(dbColumn["NUMERIC_PRECISION"]);
if (dbColumn["NUMERIC_SCALE"] !== null &&
!this.isDefaultColumnScale(table, tableColumn, dbColumn["NUMERIC_SCALE"]))
tableColumn.scale = parseInt(dbColumn["NUMERIC_SCALE"]);
}
if (tableColumn.type === "enum" ||
tableColumn.type === "simple-enum" ||
tableColumn.type === "set") {
const colType = dbColumn["COLUMN_TYPE"];
const items = colType
.slice(colType.indexOf("(") + 1, colType.lastIndexOf(")"))
.split(",");
tableColumn.enum = items.map((item) => {
return item.slice(1, -1);
});
tableColumn.length = "";
}
if ((tableColumn.type === "datetime" ||
tableColumn.type === "time" ||
tableColumn.type === "timestamp") &&
dbColumn["DATETIME_PRECISION"] !== null &&
dbColumn["DATETIME_PRECISION"] !== undefined &&
!this.isDefaultColumnPrecision(table, tableColumn, parseInt(dbColumn["DATETIME_PRECISION"]))) {
tableColumn.precision = parseInt(dbColumn["DATETIME_PRECISION"]);
}
return tableColumn;
}));
// find foreign key constraints of table, group them by constraint name and build TableForeignKey.
const tableForeignKeyConstraints = OrmUtils_1.OrmUtils.uniq(dbForeignKeys.filter((dbForeignKey) => {
return (dbForeignKey["TABLE_NAME"] ===
dbTable["TABLE_NAME"] &&
dbForeignKey["TABLE_SCHEMA"] ===
dbTable["TABLE_SCHEMA"]);
}), (dbForeignKey) => dbForeignKey["CONSTRAINT_NAME"]);
table.foreignKeys = tableForeignKeyConstraints.map((dbForeignKey) => {
const foreignKeys = dbForeignKeys.filter((dbFk) => dbFk["CONSTRAINT_NAME"] ===
dbForeignKey["CONSTRAINT_NAME"]);
// if referenced table located in currently used db, we don't need to concat db name to table name.
const database = dbForeignKey["REFERENCED_TABLE_SCHEMA"] ===
currentDatabase
? undefined
: dbForeignKey["REFERENCED_TABLE_SCHEMA"];
const referencedTableName = this.driver.buildTableName(dbForeignKey["REFERENCED_TABLE_NAME"], undefined, database);
return new TableForeignKey_1.TableForeignKey({
name: dbForeignKey["CONSTRAINT_NAME"],
columnNames: foreignKeys.map((dbFk) => dbFk["COLUMN_NAME"]),
referencedDatabase: dbForeignKey["REFERENCED_TABLE_SCHEMA"],
referencedTableName: referencedTableName,
referencedColumnNames: foreignKeys.map((dbFk) => dbFk["REFERENCED_COLUMN_NAME"]),
onDelete: dbForeignKey["ON_DELETE"],
onUpdate: dbForeignKey["ON_UPDATE"],
});
});
// find index constraints of table, group them by constraint name and build TableIndex.
const tableIndexConstraints = OrmUtils_1.OrmUtils.uniq(dbIndices.filter((dbIndex) => dbIndex["TABLE_NAME"] === dbTable["TABLE_NAME"] &&
dbIndex["TABLE_SCHEMA"] === dbTable["TABLE_SCHEMA"]), (dbIndex) => dbIndex["INDEX_NAME"]);
table.indices = tableIndexConstraints.map((constraint) => {
const indices = dbIndices.filter((index) => {
return (index["TABLE_SCHEMA"] ===
constraint["TABLE_SCHEMA"] &&
index["TABLE_NAME"] === constraint["TABLE_NAME"] &&
index["INDEX_NAME"] === constraint["INDEX_NAME"]);
});
const nonUnique = parseInt(constraint["NON_UNIQUE"], 10);
return new TableIndex_1.TableIndex({
table: table,
name: constraint["INDEX_NAME"],
columnNames: indices.map((i) => i["COLUMN_NAME"]),
isUnique: nonUnique === 0,
isSpatial: constraint["INDEX_TYPE"] === "SPATIAL",
isFulltext: constraint["INDEX_TYPE"] === "FULLTEXT",
});
});
table.comment = dbTable["TABLE_COMMENT"];
return table;
}));
}
/**
* Builds create table sql
*
* @param table
* @param createForeignKeys
*/
createTableSql(table, createForeignKeys) {
const columnDefinitions = table.columns
.map((column) => this.buildCreateColumnSql(column, true))
.join(", ");
let sql = `CREATE TABLE ${this.escapePath(table)} (${columnDefinitions}`;
// we create unique indexes instead of unique constraints, because MySql does not have unique constraints.
// if we mark column as Unique, it means that we create UNIQUE INDEX.
table.columns
.filter((column) => column.isUnique)
.forEach((column) => {
const isUniqueIndexExist = table.indices.some((index) => {
return (index.columnNames.length === 1 &&
!!index.isUnique &&
index.columnNames.indexOf(column.name) !== -1);
});
const isUniqueConstraintExist = table.uniques.some((unique) => {
return (unique.columnNames.length === 1 &&
unique.columnNames.indexOf(column.name) !== -1);
});
if (!isUniqueIndexExist && !isUniqueConstraintExist)
table.indices.push(new TableIndex_1.TableIndex({
name: this.dataSource.namingStrategy.uniqueConstraintName(table, [column.name]),
columnNames: [column.name],
isUnique: true,
}));
});
// as MySql does not have unique constraints, we must create table indices from table uniques and mark them as unique.
if (table.uniques.length > 0) {
table.uniques.forEach((unique) => {
const uniqueExist = table.indices.some((index) => index.name === unique.name);
if (!uniqueExist) {
table.indices.push(new TableIndex_1.TableIndex({
name: unique.name,
columnNames: unique.columnNames,
isUnique: true,
}));
}
});
}
if (table.indices.length > 0) {
const indicesSql = table.indices
.map((index) => {
const columnNames = index.columnNames
.map((columnName) => `\`${columnName}\``)
.join(", ");
index.name ??= this.dataSource.namingStrategy.indexName(table, index.columnNames, index.where);
let indexType = "";
if (index.isUnique)
indexType += "UNIQUE ";
if (index.isSpatial)
indexType += "SPATIAL ";
if (index.isFulltext)
indexType += "FULLTEXT ";
const indexParser = index.isFulltext && index.parser
? ` WITH PARSER ${index.parser}`
: "";
return `${indexType}INDEX \`${index.name}\` (${columnNames})${indexParser}`;
})
.join(", ");
sql += `, ${indicesSql}`;
}
if (table.foreignKeys.length > 0 && createForeignKeys) {
const foreignKeysSql = table.foreignKeys
.map((fk) => {
const columnNames = fk.columnNames
.map((columnName) => `\`${columnName}\``)
.join(", ");
fk.name ??= this.dataSource.namingStrategy.foreignKeyName(table, fk.columnNames, this.getTablePath(fk), fk.referencedColumnNames);
const referencedColumnNames = fk.referencedColumnNames
.map((columnName) => `\`${columnName}\``)
.join(", ");
let constraint = `CONSTRAINT \`${fk.name}\` FOREIGN KEY (${columnNames}) REFERENCES ${this.escapePath(this.getTablePath(fk))} (${referencedColumnNames})`;
if (fk.onDelete)
constraint += ` ON DELETE ${fk.onDelete}`;
if (fk.onUpdate)
constraint += ` ON UPDATE ${fk.onUpdate}`;
return constraint;
})
.join(", ");
sql += `, ${foreignKeysSql}`;
}
if (table.primaryColumns.length > 0) {
const columnNames = table.primaryColumns
.map((column) => `\`${column.name}\``)
.join(", ");
sql += `, PRIMARY KEY (${columnNames})`;
}
sql += `) ENGINE=${table.engine ?? "InnoDB"}`;
if (table.comment) {
sql += ` COMMENT="${table.comment}"`;
}
return new Query_1.Query(sql);
}
/**
* Builds drop table sql
*
* @param tableOrName
*/
dropTableSql(tableOrName) {
return new Query_1.Query(`DROP TABLE ${this.escapePath(tableOrName)}`);
}
createViewSql(view) {
if (typeof view.expression === "string") {
return new Query_1.Query(`CREATE VIEW ${this.escapePath(view)} AS ${view.expression}`);
}
else {
return new Query_1.Query(`CREATE VIEW ${this.escapePath(view)} AS ${view
.expression(this.dataSource)
.getQuery()}`);
}
}
async insertViewDefinitionSql(view) {
const currentDatabase = await this.getCurrentDatabase();
const expression = typeof view.expression === "string"
? view.expression.trim()
: view.expression(this.dataSource).getQuery();
return this.insertTypeormMetadataSql({
type: MetadataTableType_1.MetadataTableType.VIEW,
schema: currentDatabase,
name: view.name,
value: expression,
});
}
/**
* Builds drop view sql.
*
* @param viewOrPath
*/
dropViewSql(viewOrPath) {
return new Query_1.Query(`DROP VIEW ${this.escapePath(viewOrPath)}`);
}
/**
* Builds remove view sql.
*
* @param viewOrPath
*/
async deleteViewDefinitionSql(viewOrPath) {
const currentDatabase = await this.getCurrentDatabase();
const viewName = InstanceChecker_1.InstanceChecker.isView(viewOrPath)
? viewOrPath.name
: viewOrPath;
return this.deleteTypeormMetadataSql({
type: MetadataTableType_1.MetadataTableType.VIEW,
schema: currentDatabase,
name: viewName,
});
}
/**
* Builds create index sql.
*
* @param table
* @param index
*/
createIndexSql(table, index) {
const columns = index.columnNames
.map((columnName) => `\`${columnName}\``)
.join(", ");
let indexType = "";
if (index.isUnique)
indexType += "UNIQUE ";
if (index.isSpatial)
indexType += "SPATIAL ";
if (index.isFulltext)
indexType += "FULLTEXT ";
const indexParser = index.isFulltext && index.parser
? ` WITH PARSER ${index.parser}`
: "";
return new Query_1.Query(`CREATE ${indexType}INDEX \`${index.name}\` ON ${this.escapePath(table)} (${columns})${indexParser}`);
}
/**
* Builds drop index sql.
*
* @param table
* @param indexOrName
*/
dropIndexSql(table, indexOrName) {
const indexName = InstanceChecker_1.InstanceChecker.isTableIndex(indexOrName)
? indexOrName.name
: indexOrName;
return new Query_1.Query(`DROP INDEX \`${indexName}\` ON ${this.escapePath(table)}`);
}
/**
* Builds create primary key sql.
*
* @param table
* @param columnNames
*/
createPrimaryKeySql(table, columnNames) {
const columnNamesString = columnNames
.map((columnName) => `\`${columnName}\``)
.join(", ");
return new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} ADD PRIMARY KEY (${columnNamesString})`);
}
/**
* Builds drop primary key sql.
*
* @param table
*/
dropPrimaryKeySql(table) {
return new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP PRIMARY KEY`);
}
/**
* Builds create foreign key sql.
*
* @param table
* @param foreignKey
*/
createForeignKeySql(table, foreignKey) {
const columnNames = foreignKey.columnNames
.map((column) => `\`${column}\``)
.join(", ");
const referencedColumnNames = foreignKey.referencedColumnNames
.map((column) => `\`${column}\``)
.join(",");
let sql = `ALTER TABLE ${this.escapePath(table)} ADD CONSTRAINT \`${foreignKey.name}\` FOREIGN KEY (${columnNames}) ` +
`REFERENCES ${this.escapePath(this.getTablePath(foreignKey))}(${referencedColumnNames})`;
if (foreignKey.onDelete)
sql += ` ON DELETE ${foreignKey.onDelete}`;
if (foreignKey.onUpdate)
sql += ` ON UPDATE ${foreignKey.onUpdate}`;
return new Query_1.Query(sql);
}
/**
* Builds drop foreign key sql.
*
* @param table
* @param foreignKeyOrName
*/
dropForeignKeySql(table, foreignKeyOrName) {
const foreignKeyName = InstanceChecker_1.InstanceChecker.isTableForeignKey(foreignKeyOrName)
? foreignKeyOrName.name
: foreignKeyOrName;
return new Query_1.Query(`ALTER TABLE ${this.escapePath(table)} DROP FOREIGN KEY \`${foreignKeyName}\``);
}
/**
* Escapes a given comment so it's safe to include in a query.
*
* @param comment
*/
escapeComment(comment) {
if (!comment || comment.length === 0) {
return `''`;
}
comment = comment
.replaceAll("\\", "\\\\") // MySQL allows escaping characters via backslashes
.replaceAll("'", "''")
.replaceAll("\u0000", ""); // Null bytes aren't allowed in comments
return `'${comment}'`;
}
/**
* Escapes given table or view path.
*
* @param target
*/
escapePath(target) {
const { database, tableName } = this.driver.parseTableName(target);
if (database && database !== this.driver.database) {
return `\`${database}\`.\`${tableName}\``;
}
return `\`${tableName}\``;
}
/**
* Builds a part of query to create/change a column.
*
* @param column
* @param skipPrimary
* @param skipName
*/
buildCreateColumnSql(column, skipPrimary, skipName = false) {
let c;
if (skipName) {
c = this.dataSource.driver.createFullType(column);
}
else {
c = `\`${column.name}\` ${this.dataSource.driver.createFullType(column)}`;
}
if (column.charset)
c += ` CHARACTER SET "${column.charset}"`;
if (column.collation)
c += ` COLLATE "${column.collation}"`;
if (column.asExpression)
c += ` AS (${column.asExpression}) ${column.generatedType ?? "VIRTUAL"}`;
if (column.unsigned) {
c += " UNSIGNED";
}
if (column.enum)
c += ` (${column.enum
.map((value) => "'" + value.replaceAll("'", "''") + "'")
.join(", ")})`;
const isMariaDb = this.driver.options.type === "mariadb";
if (isMariaDb &&
column.asExpression &&
["VIRTUAL", "STORED"].includes(column.generatedType ?? "VIRTUAL")) {
// do nothing - MariaDB does not support NULL/NOT NULL expressions for VIRTUAL columns and STORED columns
}
else {
if (!column.isNullable)
c += " NOT NULL";
if (column.isNullable)
c += " NULL";
}
if (column.isPrimary && !skipPrimary)
c += " PRIMARY KEY";
if (column.isGenerated && column.generationStrategy === "increment")
// don't use skipPrimary here since updates can update already exist primary without auto inc.
c += " AUTO_INCREMENT";
if (column.comment && column.comment.length > 0)
c += ` COMMENT ${this.escapeComment(column.comment)}`;
if (column.default !== undefined && column.default !== null)
c += ` DEFAULT ${column.default}`;
if (column.onUpdate)
c += ` ON UPDATE ${column.onUpdate}`;
return c;
}
async getVersion() {
const result = await this.query("SELECT version() as version");
// MariaDB: https://mariadb.com/kb/en/version/
// - "10.2.27-MariaDB-10.2.27+maria~jessie-log"
// MySQL: https://dev.mysql.com/doc/refman/8.4/en/information-functions.html#function_version
// - "8.4.3"
// - "8.4.4-standard"
const versionString = result[0]["version"];
return versionString.replace(/^([\d.]+).*$/, "$1");
}
}
exports.MysqlQueryRunner = MysqlQueryRunner;
//# sourceMappingURL=MysqlQueryRunner.js.map