"use strict";
|
Object.defineProperty(exports, "__esModule", { value: true });
|
var tslib_1 = require("tslib");
|
var TransactionAlreadyStartedError_1 = require("../../error/TransactionAlreadyStartedError");
|
var TransactionNotStartedError_1 = require("../../error/TransactionNotStartedError");
|
var TableColumn_1 = require("../../schema-builder/table/TableColumn");
|
var Table_1 = require("../../schema-builder/table/Table");
|
var TableForeignKey_1 = require("../../schema-builder/table/TableForeignKey");
|
var TableIndex_1 = require("../../schema-builder/table/TableIndex");
|
var QueryRunnerAlreadyReleasedError_1 = require("../../error/QueryRunnerAlreadyReleasedError");
|
var View_1 = require("../../schema-builder/view/View");
|
var Query_1 = require("../Query");
|
var OrmUtils_1 = require("../../util/OrmUtils");
|
var QueryFailedError_1 = require("../../error/QueryFailedError");
|
var TableUnique_1 = require("../../schema-builder/table/TableUnique");
|
var BaseQueryRunner_1 = require("../../query-runner/BaseQueryRunner");
|
var Broadcaster_1 = require("../../subscriber/Broadcaster");
|
var index_1 = require("../../index");
|
/**
|
* Runs queries on a single mysql database connection.
|
*/
|
var MysqlQueryRunner = /** @class */ (function (_super) {
|
tslib_1.__extends(MysqlQueryRunner, _super);
|
// -------------------------------------------------------------------------
|
// Constructor
|
// -------------------------------------------------------------------------
|
function MysqlQueryRunner(driver, mode) {
|
if (mode === void 0) { mode = "master"; }
|
var _this = _super.call(this) || this;
|
_this.driver = driver;
|
_this.connection = driver.connection;
|
_this.broadcaster = new Broadcaster_1.Broadcaster(_this);
|
_this.mode = mode;
|
return _this;
|
}
|
// -------------------------------------------------------------------------
|
// Public Methods
|
// -------------------------------------------------------------------------
|
/**
|
* Creates/uses database connection from the connection pool to perform further operations.
|
* Returns obtained database connection.
|
*/
|
MysqlQueryRunner.prototype.connect = function () {
|
var _this = this;
|
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(function (connection) {
|
_this.databaseConnection = connection;
|
return _this.databaseConnection;
|
});
|
}
|
else { // master
|
this.databaseConnectionPromise = this.driver.obtainMasterConnection().then(function (connection) {
|
_this.databaseConnection = connection;
|
return _this.databaseConnection;
|
});
|
}
|
return this.databaseConnectionPromise;
|
};
|
/**
|
* Releases used database connection.
|
* You cannot use query runner methods once its released.
|
*/
|
MysqlQueryRunner.prototype.release = function () {
|
this.isReleased = true;
|
if (this.databaseConnection)
|
this.databaseConnection.release();
|
return Promise.resolve();
|
};
|
/**
|
* Starts transaction on the current connection.
|
*/
|
MysqlQueryRunner.prototype.startTransaction = function (isolationLevel) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
if (this.isTransactionActive)
|
throw new TransactionAlreadyStartedError_1.TransactionAlreadyStartedError();
|
this.isTransactionActive = true;
|
if (!isolationLevel) return [3 /*break*/, 3];
|
return [4 /*yield*/, this.query("SET TRANSACTION ISOLATION LEVEL " + isolationLevel)];
|
case 1:
|
_a.sent();
|
return [4 /*yield*/, this.query("START TRANSACTION")];
|
case 2:
|
_a.sent();
|
return [3 /*break*/, 5];
|
case 3: return [4 /*yield*/, this.query("START TRANSACTION")];
|
case 4:
|
_a.sent();
|
_a.label = 5;
|
case 5: return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Commits transaction.
|
* Error will be thrown if transaction was not started.
|
*/
|
MysqlQueryRunner.prototype.commitTransaction = function () {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
if (!this.isTransactionActive)
|
throw new TransactionNotStartedError_1.TransactionNotStartedError();
|
return [4 /*yield*/, this.query("COMMIT")];
|
case 1:
|
_a.sent();
|
this.isTransactionActive = false;
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Rollbacks transaction.
|
* Error will be thrown if transaction was not started.
|
*/
|
MysqlQueryRunner.prototype.rollbackTransaction = function () {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
if (!this.isTransactionActive)
|
throw new TransactionNotStartedError_1.TransactionNotStartedError();
|
return [4 /*yield*/, this.query("ROLLBACK")];
|
case 1:
|
_a.sent();
|
this.isTransactionActive = false;
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Executes a raw SQL query.
|
*/
|
MysqlQueryRunner.prototype.query = function (query, parameters) {
|
var _this = this;
|
if (this.isReleased)
|
throw new QueryRunnerAlreadyReleasedError_1.QueryRunnerAlreadyReleasedError();
|
return new Promise(function (ok, fail) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
|
var databaseConnection, queryStartTime_1, err_1;
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
_a.trys.push([0, 2, , 3]);
|
return [4 /*yield*/, this.connect()];
|
case 1:
|
databaseConnection = _a.sent();
|
this.driver.connection.logger.logQuery(query, parameters, this);
|
queryStartTime_1 = +new Date();
|
databaseConnection.query(query, parameters, function (err, result) {
|
// log slow queries if maxQueryExecution time is set
|
var maxQueryExecutionTime = _this.driver.connection.options.maxQueryExecutionTime;
|
var queryEndTime = +new Date();
|
var queryExecutionTime = queryEndTime - queryStartTime_1;
|
if (maxQueryExecutionTime && queryExecutionTime > maxQueryExecutionTime)
|
_this.driver.connection.logger.logQuerySlow(queryExecutionTime, query, parameters, _this);
|
if (err) {
|
_this.driver.connection.logger.logQueryError(err, query, parameters, _this);
|
return fail(new QueryFailedError_1.QueryFailedError(query, parameters, err));
|
}
|
ok(result);
|
});
|
return [3 /*break*/, 3];
|
case 2:
|
err_1 = _a.sent();
|
fail(err_1);
|
return [3 /*break*/, 3];
|
case 3: return [2 /*return*/];
|
}
|
});
|
}); });
|
};
|
/**
|
* Returns raw data stream.
|
*/
|
MysqlQueryRunner.prototype.stream = function (query, parameters, onEnd, onError) {
|
var _this = this;
|
if (this.isReleased)
|
throw new QueryRunnerAlreadyReleasedError_1.QueryRunnerAlreadyReleasedError();
|
return new Promise(function (ok, fail) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
|
var databaseConnection, stream, err_2;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
_a.trys.push([0, 2, , 3]);
|
return [4 /*yield*/, this.connect()];
|
case 1:
|
databaseConnection = _a.sent();
|
this.driver.connection.logger.logQuery(query, parameters, this);
|
stream = databaseConnection.query(query, parameters);
|
if (onEnd)
|
stream.on("end", onEnd);
|
if (onError)
|
stream.on("error", onError);
|
ok(stream);
|
return [3 /*break*/, 3];
|
case 2:
|
err_2 = _a.sent();
|
fail(err_2);
|
return [3 /*break*/, 3];
|
case 3: return [2 /*return*/];
|
}
|
});
|
}); });
|
};
|
/**
|
* Returns all available database names including system databases.
|
*/
|
MysqlQueryRunner.prototype.getDatabases = function () {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
return [2 /*return*/, Promise.resolve([])];
|
});
|
});
|
};
|
/**
|
* Returns all available schema names including system schemas.
|
* If database parameter specified, returns schemas of that database.
|
*/
|
MysqlQueryRunner.prototype.getSchemas = function (database) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql driver does not support table schemas");
|
});
|
});
|
};
|
/**
|
* Checks if database with the given name exist.
|
*/
|
MysqlQueryRunner.prototype.hasDatabase = function (database) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var result;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0: return [4 /*yield*/, this.query("SELECT * FROM `INFORMATION_SCHEMA`.`SCHEMATA` WHERE `SCHEMA_NAME` = '" + database + "'")];
|
case 1:
|
result = _a.sent();
|
return [2 /*return*/, result.length ? true : false];
|
}
|
});
|
});
|
};
|
/**
|
* Checks if schema with the given name exist.
|
*/
|
MysqlQueryRunner.prototype.hasSchema = function (schema) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql driver does not support table schemas");
|
});
|
});
|
};
|
/**
|
* Checks if table with the given name exist in the database.
|
*/
|
MysqlQueryRunner.prototype.hasTable = function (tableOrName) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var parsedTableName, sql, result;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
parsedTableName = this.parseTableName(tableOrName);
|
sql = "SELECT * FROM `INFORMATION_SCHEMA`.`COLUMNS` WHERE `TABLE_SCHEMA` = '" + parsedTableName.database + "' AND `TABLE_NAME` = '" + parsedTableName.tableName + "'";
|
return [4 /*yield*/, this.query(sql)];
|
case 1:
|
result = _a.sent();
|
return [2 /*return*/, result.length ? true : false];
|
}
|
});
|
});
|
};
|
/**
|
* Checks if column with the given name exist in the given table.
|
*/
|
MysqlQueryRunner.prototype.hasColumn = function (tableOrName, column) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var parsedTableName, columnName, sql, result;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
parsedTableName = this.parseTableName(tableOrName);
|
columnName = column instanceof TableColumn_1.TableColumn ? column.name : column;
|
sql = "SELECT * FROM `INFORMATION_SCHEMA`.`COLUMNS` WHERE `TABLE_SCHEMA` = '" + parsedTableName.database + "' AND `TABLE_NAME` = '" + parsedTableName.tableName + "' AND `COLUMN_NAME` = '" + columnName + "'";
|
return [4 /*yield*/, this.query(sql)];
|
case 1:
|
result = _a.sent();
|
return [2 /*return*/, result.length ? true : false];
|
}
|
});
|
});
|
};
|
/**
|
* Creates a new database.
|
*/
|
MysqlQueryRunner.prototype.createDatabase = function (database, ifNotExist) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var up, down;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
up = ifNotExist ? "CREATE DATABASE IF NOT EXISTS `" + database + "`" : "CREATE DATABASE `" + database + "`";
|
down = "DROP DATABASE `" + database + "`";
|
return [4 /*yield*/, this.executeQueries(new Query_1.Query(up), new Query_1.Query(down))];
|
case 1:
|
_a.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Drops database.
|
*/
|
MysqlQueryRunner.prototype.dropDatabase = function (database, ifExist) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var up, down;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
up = ifExist ? "DROP DATABASE IF EXISTS `" + database + "`" : "DROP DATABASE `" + database + "`";
|
down = "CREATE DATABASE `" + database + "`";
|
return [4 /*yield*/, this.executeQueries(new Query_1.Query(up), new Query_1.Query(down))];
|
case 1:
|
_a.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Creates a new table schema.
|
*/
|
MysqlQueryRunner.prototype.createSchema = function (schema, ifNotExist) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("Schema create queries are not supported by MySql driver.");
|
});
|
});
|
};
|
/**
|
* Drops table schema.
|
*/
|
MysqlQueryRunner.prototype.dropSchema = function (schemaPath, ifExist) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("Schema drop queries are not supported by MySql driver.");
|
});
|
});
|
};
|
/**
|
* Creates a new table.
|
*/
|
MysqlQueryRunner.prototype.createTable = function (table, ifNotExist, createForeignKeys) {
|
if (ifNotExist === void 0) { ifNotExist = false; }
|
if (createForeignKeys === void 0) { createForeignKeys = true; }
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var isTableExist, upQueries, downQueries;
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
if (!ifNotExist) return [3 /*break*/, 2];
|
return [4 /*yield*/, this.hasTable(table)];
|
case 1:
|
isTableExist = _a.sent();
|
if (isTableExist)
|
return [2 /*return*/, Promise.resolve()];
|
_a.label = 2;
|
case 2:
|
upQueries = [];
|
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(function (index) { return 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(function (foreignKey) { return downQueries.push(_this.dropForeignKeySql(table, foreignKey)); });
|
return [2 /*return*/, this.executeQueries(upQueries, downQueries)];
|
}
|
});
|
});
|
};
|
/**
|
* Drop the table.
|
*/
|
MysqlQueryRunner.prototype.dropTable = function (target, ifExist, dropForeignKeys) {
|
if (dropForeignKeys === void 0) { dropForeignKeys = true; }
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var isTableExist, createForeignKeys, tableName, table, upQueries, downQueries;
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
if (!ifExist) return [3 /*break*/, 2];
|
return [4 /*yield*/, this.hasTable(target)];
|
case 1:
|
isTableExist = _a.sent();
|
if (!isTableExist)
|
return [2 /*return*/, Promise.resolve()];
|
_a.label = 2;
|
case 2:
|
createForeignKeys = dropForeignKeys;
|
tableName = target instanceof Table_1.Table ? target.name : target;
|
return [4 /*yield*/, this.getCachedTable(tableName)];
|
case 3:
|
table = _a.sent();
|
upQueries = [];
|
downQueries = [];
|
if (dropForeignKeys)
|
table.foreignKeys.forEach(function (foreignKey) { return upQueries.push(_this.dropForeignKeySql(table, foreignKey)); });
|
table.indices.forEach(function (index) { return upQueries.push(_this.dropIndexSql(table, index)); });
|
upQueries.push(this.dropTableSql(table));
|
downQueries.push(this.createTableSql(table, createForeignKeys));
|
return [4 /*yield*/, this.executeQueries(upQueries, downQueries)];
|
case 4:
|
_a.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Creates a new view.
|
*/
|
MysqlQueryRunner.prototype.createView = function (view) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var upQueries, downQueries, _a, _b, _c, _d;
|
return tslib_1.__generator(this, function (_e) {
|
switch (_e.label) {
|
case 0:
|
upQueries = [];
|
downQueries = [];
|
upQueries.push(this.createViewSql(view));
|
_b = (_a = upQueries).push;
|
return [4 /*yield*/, this.insertViewDefinitionSql(view)];
|
case 1:
|
_b.apply(_a, [_e.sent()]);
|
downQueries.push(this.dropViewSql(view));
|
_d = (_c = downQueries).push;
|
return [4 /*yield*/, this.deleteViewDefinitionSql(view)];
|
case 2:
|
_d.apply(_c, [_e.sent()]);
|
return [4 /*yield*/, this.executeQueries(upQueries, downQueries)];
|
case 3:
|
_e.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Drops the view.
|
*/
|
MysqlQueryRunner.prototype.dropView = function (target) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var viewName, view, upQueries, downQueries, _a, _b, _c, _d;
|
return tslib_1.__generator(this, function (_e) {
|
switch (_e.label) {
|
case 0:
|
viewName = target instanceof View_1.View ? target.name : target;
|
return [4 /*yield*/, this.getCachedView(viewName)];
|
case 1:
|
view = _e.sent();
|
upQueries = [];
|
downQueries = [];
|
_b = (_a = upQueries).push;
|
return [4 /*yield*/, this.deleteViewDefinitionSql(view)];
|
case 2:
|
_b.apply(_a, [_e.sent()]);
|
upQueries.push(this.dropViewSql(view));
|
_d = (_c = downQueries).push;
|
return [4 /*yield*/, this.insertViewDefinitionSql(view)];
|
case 3:
|
_d.apply(_c, [_e.sent()]);
|
downQueries.push(this.createViewSql(view));
|
return [4 /*yield*/, this.executeQueries(upQueries, downQueries)];
|
case 4:
|
_e.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Renames a table.
|
*/
|
MysqlQueryRunner.prototype.renameTable = function (oldTableOrName, newTableName) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var upQueries, downQueries, oldTable, _a, newTable, dbName;
|
var _this = this;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
upQueries = [];
|
downQueries = [];
|
if (!(oldTableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = oldTableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(oldTableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
oldTable = _a;
|
newTable = oldTable.clone();
|
dbName = oldTable.name.indexOf(".") === -1 ? undefined : oldTable.name.split(".")[0];
|
newTable.name = dbName ? dbName + "." + newTableName : newTableName;
|
// rename table
|
upQueries.push(new Query_1.Query("RENAME TABLE " + this.escapePath(oldTable.name) + " TO " + this.escapePath(newTable.name)));
|
downQueries.push(new Query_1.Query("RENAME TABLE " + this.escapePath(newTable.name) + " TO " + this.escapePath(oldTable.name)));
|
// rename index constraints
|
newTable.indices.forEach(function (index) {
|
// build new constraint name
|
var columnNames = index.columnNames.map(function (column) { return "`" + column + "`"; }).join(", ");
|
var newIndexName = _this.connection.namingStrategy.indexName(newTable, index.columnNames, index.where);
|
// build queries
|
var indexType = "";
|
if (index.isUnique)
|
indexType += "UNIQUE ";
|
if (index.isSpatial)
|
indexType += "SPATIAL ";
|
if (index.isFulltext)
|
indexType += "FULLTEXT ";
|
upQueries.push(new Query_1.Query("ALTER TABLE " + _this.escapePath(newTable) + " DROP INDEX `" + index.name + "`, ADD " + indexType + "INDEX `" + newIndexName + "` (" + columnNames + ")"));
|
downQueries.push(new Query_1.Query("ALTER TABLE " + _this.escapePath(newTable) + " DROP INDEX `" + newIndexName + "`, ADD " + indexType + "INDEX `" + index.name + "` (" + columnNames + ")"));
|
// replace constraint name
|
index.name = newIndexName;
|
});
|
// rename foreign key constraint
|
newTable.foreignKeys.forEach(function (foreignKey) {
|
// build new constraint name
|
var columnNames = foreignKey.columnNames.map(function (column) { return "`" + column + "`"; }).join(", ");
|
var referencedColumnNames = foreignKey.referencedColumnNames.map(function (column) { return "`" + column + "`"; }).join(",");
|
var newForeignKeyName = _this.connection.namingStrategy.foreignKeyName(newTable, foreignKey.columnNames);
|
// build queries
|
var up = "ALTER TABLE " + _this.escapePath(newTable) + " DROP FOREIGN KEY `" + foreignKey.name + "`, ADD CONSTRAINT `" + newForeignKeyName + "` FOREIGN KEY (" + columnNames + ") " +
|
("REFERENCES " + _this.escapePath(foreignKey.referencedTableName) + "(" + referencedColumnNames + ")");
|
if (foreignKey.onDelete)
|
up += " ON DELETE " + foreignKey.onDelete;
|
if (foreignKey.onUpdate)
|
up += " ON UPDATE " + foreignKey.onUpdate;
|
var down = "ALTER TABLE " + _this.escapePath(newTable) + " DROP FOREIGN KEY `" + newForeignKeyName + "`, ADD CONSTRAINT `" + foreignKey.name + "` FOREIGN KEY (" + columnNames + ") " +
|
("REFERENCES " + _this.escapePath(foreignKey.referencedTableName) + "(" + 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;
|
});
|
return [4 /*yield*/, this.executeQueries(upQueries, downQueries)];
|
case 4:
|
_b.sent();
|
// rename old table and replace it in cached tabled;
|
oldTable.name = newTable.name;
|
this.replaceCachedTable(oldTable, newTable);
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Creates a new column from the column in the table.
|
*/
|
MysqlQueryRunner.prototype.addColumn = function (tableOrName, column) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var table, _a, clonedTable, upQueries, downQueries, skipColumnLevelPrimary, generatedColumn, nonGeneratedColumn, primaryColumns, columnNames, nonGeneratedColumn, columnIndex, uniqueIndex;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
if (!(tableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = tableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(tableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
table = _a;
|
clonedTable = table.clone();
|
upQueries = [];
|
downQueries = [];
|
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) {
|
generatedColumn = clonedTable.columns.find(function (column) { return column.isGenerated && column.generationStrategy === "increment"; });
|
if (generatedColumn) {
|
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)));
|
}
|
primaryColumns = clonedTable.primaryColumns;
|
columnNames = primaryColumns.map(function (column) { return "`" + 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(function (column) { return "`" + 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) {
|
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)));
|
}
|
}
|
columnIndex = clonedTable.indices.find(function (index) { return 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) {
|
uniqueIndex = new TableIndex_1.TableIndex({
|
name: this.connection.namingStrategy.indexName(table.name, [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 + "`"));
|
}
|
return [4 /*yield*/, this.executeQueries(upQueries, downQueries)];
|
case 4:
|
_b.sent();
|
clonedTable.addColumn(column);
|
this.replaceCachedTable(table, clonedTable);
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Creates a new columns from the column in the table.
|
*/
|
MysqlQueryRunner.prototype.addColumns = function (tableOrName, columns) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0: return [4 /*yield*/, index_1.PromiseUtils.runInSequence(columns, function (column) { return _this.addColumn(tableOrName, column); })];
|
case 1:
|
_a.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Renames column in the given table.
|
*/
|
MysqlQueryRunner.prototype.renameColumn = function (tableOrName, oldTableColumnOrName, newTableColumnOrName) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var table, _a, oldColumn, newColumn;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
if (!(tableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = tableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(tableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
table = _a;
|
oldColumn = oldTableColumnOrName instanceof TableColumn_1.TableColumn ? oldTableColumnOrName : table.columns.find(function (c) { return c.name === oldTableColumnOrName; });
|
if (!oldColumn)
|
throw new Error("Column \"" + oldTableColumnOrName + "\" was not found in the \"" + table.name + "\" table.");
|
newColumn = undefined;
|
if (newTableColumnOrName instanceof TableColumn_1.TableColumn) {
|
newColumn = newTableColumnOrName;
|
}
|
else {
|
newColumn = oldColumn.clone();
|
newColumn.name = newTableColumnOrName;
|
}
|
return [4 /*yield*/, this.changeColumn(table, oldColumn, newColumn)];
|
case 4:
|
_b.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Changes a column in the table.
|
*/
|
MysqlQueryRunner.prototype.changeColumn = function (tableOrName, oldColumnOrName, newColumn) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var table, _a, clonedTable, upQueries, downQueries, oldColumn, oldTableColumn, generatedColumn, nonGeneratedColumn, primaryColumns, columnNames, column, columnNames, primaryColumn, column, columnNames, nonGeneratedColumn, uniqueIndex, uniqueIndex_1, tableUnique;
|
var _this = this;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
if (!(tableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = tableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(tableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
table = _a;
|
clonedTable = table.clone();
|
upQueries = [];
|
downQueries = [];
|
oldColumn = oldColumnOrName instanceof TableColumn_1.TableColumn
|
? oldColumnOrName
|
: table.columns.find(function (column) { return column.name === oldColumnOrName; });
|
if (!oldColumn)
|
throw new Error("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)) return [3 /*break*/, 6];
|
return [4 /*yield*/, this.dropColumn(table, oldColumn)];
|
case 4:
|
_b.sent();
|
return [4 /*yield*/, this.addColumn(table, newColumn)];
|
case 5:
|
_b.sent();
|
// update cloned table
|
clonedTable = table.clone();
|
return [3 /*break*/, 7];
|
case 6:
|
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(function (index) {
|
// build new constraint name
|
index.columnNames.splice(index.columnNames.indexOf(oldColumn.name), 1);
|
index.columnNames.push(newColumn.name);
|
var columnNames = index.columnNames.map(function (column) { return "`" + column + "`"; }).join(", ");
|
var newIndexName = _this.connection.namingStrategy.indexName(clonedTable, index.columnNames, index.where);
|
// build queries
|
var indexType = "";
|
if (index.isUnique)
|
indexType += "UNIQUE ";
|
if (index.isSpatial)
|
indexType += "SPATIAL ";
|
if (index.isFulltext)
|
indexType += "FULLTEXT ";
|
upQueries.push(new Query_1.Query("ALTER TABLE " + _this.escapePath(table) + " DROP INDEX `" + index.name + "`, ADD " + indexType + "INDEX `" + newIndexName + "` (" + columnNames + ")"));
|
downQueries.push(new Query_1.Query("ALTER TABLE " + _this.escapePath(table) + " DROP INDEX `" + newIndexName + "`, ADD " + indexType + "INDEX `" + index.name + "` (" + columnNames + ")"));
|
// replace constraint name
|
index.name = newIndexName;
|
});
|
// rename foreign key constraints
|
clonedTable.findColumnForeignKeys(oldColumn).forEach(function (foreignKey) {
|
// build new constraint name
|
foreignKey.columnNames.splice(foreignKey.columnNames.indexOf(oldColumn.name), 1);
|
foreignKey.columnNames.push(newColumn.name);
|
var columnNames = foreignKey.columnNames.map(function (column) { return "`" + column + "`"; }).join(", ");
|
var referencedColumnNames = foreignKey.referencedColumnNames.map(function (column) { return "`" + column + "`"; }).join(",");
|
var newForeignKeyName = _this.connection.namingStrategy.foreignKeyName(clonedTable, foreignKey.columnNames);
|
// build queries
|
var up = "ALTER TABLE " + _this.escapePath(table) + " DROP FOREIGN KEY `" + foreignKey.name + "`, ADD CONSTRAINT `" + newForeignKeyName + "` FOREIGN KEY (" + columnNames + ") " +
|
("REFERENCES " + _this.escapePath(foreignKey.referencedTableName) + "(" + referencedColumnNames + ")");
|
if (foreignKey.onDelete)
|
up += " ON DELETE " + foreignKey.onDelete;
|
if (foreignKey.onUpdate)
|
up += " ON UPDATE " + foreignKey.onUpdate;
|
var down = "ALTER TABLE " + _this.escapePath(table) + " DROP FOREIGN KEY `" + newForeignKeyName + "`, ADD CONSTRAINT `" + foreignKey.name + "` FOREIGN KEY (" + columnNames + ") " +
|
("REFERENCES " + _this.escapePath(foreignKey.referencedTableName) + "(" + 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;
|
});
|
oldTableColumn = clonedTable.columns.find(function (column) { return column.name === oldColumn.name; });
|
clonedTable.columns[clonedTable.columns.indexOf(oldTableColumn)].name = newColumn.name;
|
oldColumn.name = newColumn.name;
|
}
|
if (this.isColumnChanged(oldColumn, newColumn, 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 (newColumn.isPrimary !== oldColumn.isPrimary) {
|
generatedColumn = clonedTable.columns.find(function (column) { return column.isGenerated && column.generationStrategy === "increment"; });
|
if (generatedColumn) {
|
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)));
|
}
|
primaryColumns = clonedTable.primaryColumns;
|
// if primary column state changed, we must always drop existed constraint.
|
if (primaryColumns.length > 0) {
|
columnNames = primaryColumns.map(function (column) { return "`" + 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);
|
column = clonedTable.columns.find(function (column) { return column.name === newColumn.name; });
|
column.isPrimary = true;
|
columnNames = primaryColumns.map(function (column) { return "`" + 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 {
|
primaryColumn = primaryColumns.find(function (c) { return c.name === newColumn.name; });
|
primaryColumns.splice(primaryColumns.indexOf(primaryColumn), 1);
|
column = clonedTable.columns.find(function (column) { return column.name === newColumn.name; });
|
column.isPrimary = false;
|
// if we have another primary keys, we must recreate constraint.
|
if (primaryColumns.length > 0) {
|
columnNames = primaryColumns.map(function (column) { return "`" + 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) {
|
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) {
|
uniqueIndex = new TableIndex_1.TableIndex({
|
name: this.connection.namingStrategy.indexName(table.name, [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 {
|
uniqueIndex_1 = clonedTable.indices.find(function (index) {
|
return index.columnNames.length === 1 && index.isUnique === true && !!index.columnNames.find(function (columnName) { return columnName === newColumn.name; });
|
});
|
clonedTable.indices.splice(clonedTable.indices.indexOf(uniqueIndex_1), 1);
|
tableUnique = clonedTable.uniques.find(function (unique) { return unique.name === uniqueIndex_1.name; });
|
clonedTable.uniques.splice(clonedTable.uniques.indexOf(tableUnique), 1);
|
upQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " DROP INDEX `" + uniqueIndex_1.name + "`"));
|
downQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " ADD UNIQUE INDEX `" + uniqueIndex_1.name + "` (`" + newColumn.name + "`)"));
|
}
|
}
|
_b.label = 7;
|
case 7: return [4 /*yield*/, this.executeQueries(upQueries, downQueries)];
|
case 8:
|
_b.sent();
|
this.replaceCachedTable(table, clonedTable);
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Changes a column in the table.
|
*/
|
MysqlQueryRunner.prototype.changeColumns = function (tableOrName, changedColumns) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0: return [4 /*yield*/, index_1.PromiseUtils.runInSequence(changedColumns, function (changedColumn) { return _this.changeColumn(tableOrName, changedColumn.oldColumn, changedColumn.newColumn); })];
|
case 1:
|
_a.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Drops column in the table.
|
*/
|
MysqlQueryRunner.prototype.dropColumn = function (tableOrName, columnOrName) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var table, _a, column, clonedTable, upQueries, downQueries, generatedColumn, nonGeneratedColumn, columnNames, tableColumn, columnNames_1, nonGeneratedColumn, columnIndex, uniqueName_1, foundUnique, indexName_1, foundIndex;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
if (!(tableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = tableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(tableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
table = _a;
|
column = columnOrName instanceof TableColumn_1.TableColumn ? columnOrName : table.findColumnByName(columnOrName);
|
if (!column)
|
throw new Error("Column \"" + columnOrName + "\" was not found in table \"" + table.name + "\"");
|
clonedTable = table.clone();
|
upQueries = [];
|
downQueries = [];
|
// drop primary key constraint
|
if (column.isPrimary) {
|
generatedColumn = clonedTable.columns.find(function (column) { return column.isGenerated && column.generationStrategy === "increment"; });
|
if (generatedColumn) {
|
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)));
|
}
|
columnNames = clonedTable.primaryColumns.map(function (primaryColumn) { return "`" + 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 + ")"));
|
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) {
|
columnNames_1 = clonedTable.primaryColumns.map(function (primaryColumn) { return "`" + primaryColumn.name + "`"; }).join(", ");
|
upQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(clonedTable) + " ADD PRIMARY KEY (" + columnNames_1 + ")"));
|
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) {
|
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)));
|
}
|
}
|
columnIndex = clonedTable.indices.find(function (index) { return 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) {
|
uniqueName_1 = this.connection.namingStrategy.uniqueConstraintName(table.name, [column.name]);
|
foundUnique = clonedTable.uniques.find(function (unique) { return unique.name === uniqueName_1; });
|
if (foundUnique)
|
clonedTable.uniques.splice(clonedTable.uniques.indexOf(foundUnique), 1);
|
indexName_1 = this.connection.namingStrategy.indexName(table.name, [column.name]);
|
foundIndex = clonedTable.indices.find(function (index) { return index.name === indexName_1; });
|
if (foundIndex)
|
clonedTable.indices.splice(clonedTable.indices.indexOf(foundIndex), 1);
|
upQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " DROP INDEX `" + indexName_1 + "`"));
|
downQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " ADD UNIQUE INDEX `" + indexName_1 + "` (`" + 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)));
|
return [4 /*yield*/, this.executeQueries(upQueries, downQueries)];
|
case 4:
|
_b.sent();
|
clonedTable.removeColumn(column);
|
this.replaceCachedTable(table, clonedTable);
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Drops the columns in the table.
|
*/
|
MysqlQueryRunner.prototype.dropColumns = function (tableOrName, columns) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0: return [4 /*yield*/, index_1.PromiseUtils.runInSequence(columns, function (column) { return _this.dropColumn(tableOrName, column); })];
|
case 1:
|
_a.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Creates a new primary key.
|
*/
|
MysqlQueryRunner.prototype.createPrimaryKey = function (tableOrName, columnNames) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var table, _a, clonedTable, up, down;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
if (!(tableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = tableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(tableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
table = _a;
|
clonedTable = table.clone();
|
up = this.createPrimaryKeySql(table, columnNames);
|
down = this.dropPrimaryKeySql(table);
|
return [4 /*yield*/, this.executeQueries(up, down)];
|
case 4:
|
_b.sent();
|
clonedTable.columns.forEach(function (column) {
|
if (columnNames.find(function (columnName) { return columnName === column.name; }))
|
column.isPrimary = true;
|
});
|
this.replaceCachedTable(table, clonedTable);
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Updates composite primary keys.
|
*/
|
MysqlQueryRunner.prototype.updatePrimaryKeys = function (tableOrName, columns) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var table, _a, clonedTable, columnNames, upQueries, downQueries, generatedColumn, nonGeneratedColumn, primaryColumns, columnNames_2, columnNamesString, newOrExistGeneratedColumn, nonGeneratedColumn, changedGeneratedColumn;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
if (!(tableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = tableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(tableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
table = _a;
|
clonedTable = table.clone();
|
columnNames = columns.map(function (column) { return column.name; });
|
upQueries = [];
|
downQueries = [];
|
generatedColumn = clonedTable.columns.find(function (column) { return column.isGenerated && column.generationStrategy === "increment"; });
|
if (generatedColumn) {
|
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)));
|
}
|
primaryColumns = clonedTable.primaryColumns;
|
if (primaryColumns.length > 0) {
|
columnNames_2 = primaryColumns.map(function (column) { return "`" + 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_2 + ")"));
|
}
|
// update columns in table.
|
clonedTable.columns
|
.filter(function (column) { return columnNames.indexOf(column.name) !== -1; })
|
.forEach(function (column) { return column.isPrimary = true; });
|
columnNamesString = columnNames.map(function (columnName) { return "`" + 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"));
|
newOrExistGeneratedColumn = generatedColumn ? generatedColumn : columns.find(function (column) { return column.isGenerated && column.generationStrategy === "increment"; });
|
if (newOrExistGeneratedColumn) {
|
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)));
|
changedGeneratedColumn = clonedTable.columns.find(function (column) { return column.name === newOrExistGeneratedColumn.name; });
|
changedGeneratedColumn.isGenerated = true;
|
changedGeneratedColumn.generationStrategy = "increment";
|
}
|
return [4 /*yield*/, this.executeQueries(upQueries, downQueries)];
|
case 4:
|
_b.sent();
|
this.replaceCachedTable(table, clonedTable);
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Drops a primary key.
|
*/
|
MysqlQueryRunner.prototype.dropPrimaryKey = function (tableOrName) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var table, _a, up, down;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
if (!(tableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = tableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(tableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
table = _a;
|
up = this.dropPrimaryKeySql(table);
|
down = this.createPrimaryKeySql(table, table.primaryColumns.map(function (column) { return column.name; }));
|
return [4 /*yield*/, this.executeQueries(up, down)];
|
case 4:
|
_b.sent();
|
table.primaryColumns.forEach(function (column) {
|
column.isPrimary = false;
|
});
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Creates a new unique constraint.
|
*/
|
MysqlQueryRunner.prototype.createUniqueConstraint = function (tableOrName, uniqueConstraint) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support unique constraints. Use unique index instead.");
|
});
|
});
|
};
|
/**
|
* Creates a new unique constraints.
|
*/
|
MysqlQueryRunner.prototype.createUniqueConstraints = function (tableOrName, uniqueConstraints) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support unique constraints. Use unique index instead.");
|
});
|
});
|
};
|
/**
|
* Drops an unique constraint.
|
*/
|
MysqlQueryRunner.prototype.dropUniqueConstraint = function (tableOrName, uniqueOrName) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support unique constraints. Use unique index instead.");
|
});
|
});
|
};
|
/**
|
* Drops an unique constraints.
|
*/
|
MysqlQueryRunner.prototype.dropUniqueConstraints = function (tableOrName, uniqueConstraints) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support unique constraints. Use unique index instead.");
|
});
|
});
|
};
|
/**
|
* Creates a new check constraint.
|
*/
|
MysqlQueryRunner.prototype.createCheckConstraint = function (tableOrName, checkConstraint) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support check constraints.");
|
});
|
});
|
};
|
/**
|
* Creates a new check constraints.
|
*/
|
MysqlQueryRunner.prototype.createCheckConstraints = function (tableOrName, checkConstraints) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support check constraints.");
|
});
|
});
|
};
|
/**
|
* Drops check constraint.
|
*/
|
MysqlQueryRunner.prototype.dropCheckConstraint = function (tableOrName, checkOrName) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support check constraints.");
|
});
|
});
|
};
|
/**
|
* Drops check constraints.
|
*/
|
MysqlQueryRunner.prototype.dropCheckConstraints = function (tableOrName, checkConstraints) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support check constraints.");
|
});
|
});
|
};
|
/**
|
* Creates a new exclusion constraint.
|
*/
|
MysqlQueryRunner.prototype.createExclusionConstraint = function (tableOrName, exclusionConstraint) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support exclusion constraints.");
|
});
|
});
|
};
|
/**
|
* Creates a new exclusion constraints.
|
*/
|
MysqlQueryRunner.prototype.createExclusionConstraints = function (tableOrName, exclusionConstraints) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support exclusion constraints.");
|
});
|
});
|
};
|
/**
|
* Drops exclusion constraint.
|
*/
|
MysqlQueryRunner.prototype.dropExclusionConstraint = function (tableOrName, exclusionOrName) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support exclusion constraints.");
|
});
|
});
|
};
|
/**
|
* Drops exclusion constraints.
|
*/
|
MysqlQueryRunner.prototype.dropExclusionConstraints = function (tableOrName, exclusionConstraints) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
throw new Error("MySql does not support exclusion constraints.");
|
});
|
});
|
};
|
/**
|
* Creates a new foreign key.
|
*/
|
MysqlQueryRunner.prototype.createForeignKey = function (tableOrName, foreignKey) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var table, _a, up, down;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
if (!(tableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = tableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(tableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
table = _a;
|
// new FK may be passed without name. In this case we generate FK name manually.
|
if (!foreignKey.name)
|
foreignKey.name = this.connection.namingStrategy.foreignKeyName(table.name, foreignKey.columnNames);
|
up = this.createForeignKeySql(table, foreignKey);
|
down = this.dropForeignKeySql(table, foreignKey);
|
return [4 /*yield*/, this.executeQueries(up, down)];
|
case 4:
|
_b.sent();
|
table.addForeignKey(foreignKey);
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Creates a new foreign keys.
|
*/
|
MysqlQueryRunner.prototype.createForeignKeys = function (tableOrName, foreignKeys) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var promises;
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
promises = foreignKeys.map(function (foreignKey) { return _this.createForeignKey(tableOrName, foreignKey); });
|
return [4 /*yield*/, Promise.all(promises)];
|
case 1:
|
_a.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Drops a foreign key.
|
*/
|
MysqlQueryRunner.prototype.dropForeignKey = function (tableOrName, foreignKeyOrName) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var table, _a, foreignKey, up, down;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
if (!(tableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = tableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(tableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
table = _a;
|
foreignKey = foreignKeyOrName instanceof TableForeignKey_1.TableForeignKey ? foreignKeyOrName : table.foreignKeys.find(function (fk) { return fk.name === foreignKeyOrName; });
|
if (!foreignKey)
|
throw new Error("Supplied foreign key was not found in table " + table.name);
|
up = this.dropForeignKeySql(table, foreignKey);
|
down = this.createForeignKeySql(table, foreignKey);
|
return [4 /*yield*/, this.executeQueries(up, down)];
|
case 4:
|
_b.sent();
|
table.removeForeignKey(foreignKey);
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Drops a foreign keys from the table.
|
*/
|
MysqlQueryRunner.prototype.dropForeignKeys = function (tableOrName, foreignKeys) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var promises;
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
promises = foreignKeys.map(function (foreignKey) { return _this.dropForeignKey(tableOrName, foreignKey); });
|
return [4 /*yield*/, Promise.all(promises)];
|
case 1:
|
_a.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Creates a new index.
|
*/
|
MysqlQueryRunner.prototype.createIndex = function (tableOrName, index) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var table, _a, up, down;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
if (!(tableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = tableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(tableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
table = _a;
|
// new index may be passed without name. In this case we generate index name manually.
|
if (!index.name)
|
index.name = this.connection.namingStrategy.indexName(table.name, index.columnNames, index.where);
|
up = this.createIndexSql(table, index);
|
down = this.dropIndexSql(table, index);
|
return [4 /*yield*/, this.executeQueries(up, down)];
|
case 4:
|
_b.sent();
|
table.addIndex(index, true);
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Creates a new indices
|
*/
|
MysqlQueryRunner.prototype.createIndices = function (tableOrName, indices) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var promises;
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
promises = indices.map(function (index) { return _this.createIndex(tableOrName, index); });
|
return [4 /*yield*/, Promise.all(promises)];
|
case 1:
|
_a.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Drops an index.
|
*/
|
MysqlQueryRunner.prototype.dropIndex = function (tableOrName, indexOrName) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var table, _a, index, up, down;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
if (!(tableOrName instanceof Table_1.Table)) return [3 /*break*/, 1];
|
_a = tableOrName;
|
return [3 /*break*/, 3];
|
case 1: return [4 /*yield*/, this.getCachedTable(tableOrName)];
|
case 2:
|
_a = _b.sent();
|
_b.label = 3;
|
case 3:
|
table = _a;
|
index = indexOrName instanceof TableIndex_1.TableIndex ? indexOrName : table.indices.find(function (i) { return i.name === indexOrName; });
|
if (!index)
|
throw new Error("Supplied index was not found in table " + table.name);
|
up = this.dropIndexSql(table, index);
|
down = this.createIndexSql(table, index);
|
return [4 /*yield*/, this.executeQueries(up, down)];
|
case 4:
|
_b.sent();
|
table.removeIndex(index, true);
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Drops an indices from the table.
|
*/
|
MysqlQueryRunner.prototype.dropIndices = function (tableOrName, indices) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var promises;
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
promises = indices.map(function (index) { return _this.dropIndex(tableOrName, index); });
|
return [4 /*yield*/, Promise.all(promises)];
|
case 1:
|
_a.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* Clears all table contents.
|
* Note: this operation uses SQL's TRUNCATE query which cannot be reverted in transactions.
|
*/
|
MysqlQueryRunner.prototype.clearTable = function (tableOrName) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0: return [4 /*yield*/, this.query("TRUNCATE TABLE " + this.escapePath(tableOrName))];
|
case 1:
|
_a.sent();
|
return [2 /*return*/];
|
}
|
});
|
});
|
};
|
/**
|
* 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).
|
*/
|
MysqlQueryRunner.prototype.clearDatabase = function (database) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var dbName, isDatabaseExist, selectViewDropsQuery, dropViewQueries, disableForeignKeysCheckQuery, dropTablesQuery, enableForeignKeysCheckQuery, dropQueries, error_1, rollbackError_1;
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0:
|
dbName = database ? database : this.driver.database;
|
if (!dbName) return [3 /*break*/, 2];
|
return [4 /*yield*/, this.hasDatabase(dbName)];
|
case 1:
|
isDatabaseExist = _a.sent();
|
if (!isDatabaseExist)
|
return [2 /*return*/, Promise.resolve()];
|
return [3 /*break*/, 3];
|
case 2: throw new Error("Can not clear database. No database is specified");
|
case 3: return [4 /*yield*/, this.startTransaction()];
|
case 4:
|
_a.sent();
|
_a.label = 5;
|
case 5:
|
_a.trys.push([5, 13, , 18]);
|
selectViewDropsQuery = "SELECT concat('DROP VIEW IF EXISTS `', table_schema, '`.`', table_name, '`') AS `query` FROM `INFORMATION_SCHEMA`.`VIEWS` WHERE `TABLE_SCHEMA` = '" + dbName + "'";
|
return [4 /*yield*/, this.query(selectViewDropsQuery)];
|
case 6:
|
dropViewQueries = _a.sent();
|
return [4 /*yield*/, Promise.all(dropViewQueries.map(function (q) { return _this.query(q["query"]); }))];
|
case 7:
|
_a.sent();
|
disableForeignKeysCheckQuery = "SET FOREIGN_KEY_CHECKS = 0;";
|
dropTablesQuery = "SELECT concat('DROP TABLE IF EXISTS `', table_schema, '`.`', table_name, '`') AS `query` FROM `INFORMATION_SCHEMA`.`TABLES` WHERE `TABLE_SCHEMA` = '" + dbName + "'";
|
enableForeignKeysCheckQuery = "SET FOREIGN_KEY_CHECKS = 1;";
|
return [4 /*yield*/, this.query(disableForeignKeysCheckQuery)];
|
case 8:
|
_a.sent();
|
return [4 /*yield*/, this.query(dropTablesQuery)];
|
case 9:
|
dropQueries = _a.sent();
|
return [4 /*yield*/, Promise.all(dropQueries.map(function (query) { return _this.query(query["query"]); }))];
|
case 10:
|
_a.sent();
|
return [4 /*yield*/, this.query(enableForeignKeysCheckQuery)];
|
case 11:
|
_a.sent();
|
return [4 /*yield*/, this.commitTransaction()];
|
case 12:
|
_a.sent();
|
return [3 /*break*/, 18];
|
case 13:
|
error_1 = _a.sent();
|
_a.label = 14;
|
case 14:
|
_a.trys.push([14, 16, , 17]);
|
return [4 /*yield*/, this.rollbackTransaction()];
|
case 15:
|
_a.sent();
|
return [3 /*break*/, 17];
|
case 16:
|
rollbackError_1 = _a.sent();
|
return [3 /*break*/, 17];
|
case 17: throw error_1;
|
case 18: return [2 /*return*/];
|
}
|
});
|
});
|
};
|
// -------------------------------------------------------------------------
|
// Protected Methods
|
// -------------------------------------------------------------------------
|
/**
|
* Returns current database.
|
*/
|
MysqlQueryRunner.prototype.getCurrentDatabase = function () {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var currentDBQuery;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0: return [4 /*yield*/, this.query("SELECT DATABASE() AS `db_name`")];
|
case 1:
|
currentDBQuery = _a.sent();
|
return [2 /*return*/, currentDBQuery[0]["db_name"]];
|
}
|
});
|
});
|
};
|
MysqlQueryRunner.prototype.loadViews = function (viewNames) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var hasTable, currentDatabase, viewsCondition, query, dbViews;
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
switch (_a.label) {
|
case 0: return [4 /*yield*/, this.hasTable(this.getTypeormMetadataTableName())];
|
case 1:
|
hasTable = _a.sent();
|
if (!hasTable)
|
return [2 /*return*/, Promise.resolve([])];
|
return [4 /*yield*/, this.getCurrentDatabase()];
|
case 2:
|
currentDatabase = _a.sent();
|
viewsCondition = viewNames.map(function (tableName) {
|
var _a = tslib_1.__read(tableName.split("."), 2), database = _a[0], name = _a[1];
|
if (!name) {
|
name = database;
|
database = _this.driver.database || currentDatabase;
|
}
|
return "(`t`.`schema` = '" + database + "' AND `t`.`name` = '" + name + "')";
|
}).join(" OR ");
|
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` = 'VIEW' " + (viewsCondition ? "AND (" + viewsCondition + ")" : ""));
|
return [4 /*yield*/, this.query(query)];
|
case 3:
|
dbViews = _a.sent();
|
return [2 /*return*/, dbViews.map(function (dbView) {
|
var view = new View_1.View();
|
var db = dbView["schema"] === currentDatabase ? undefined : 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.
|
*/
|
MysqlQueryRunner.prototype.loadTables = function (tableNames) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var currentDatabase, tablesCondition, tablesSql, columnsSql, primaryKeySql, collationsSql, indicesCondition, indicesSql, foreignKeysCondition, foreignKeysSql, _a, dbTables, dbColumns, dbPrimaryKeys, dbCollations, dbIndices, dbForeignKeys, isMariaDb;
|
var _this = this;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0:
|
// if no tables given then no need to proceed
|
if (!tableNames || !tableNames.length)
|
return [2 /*return*/, []];
|
return [4 /*yield*/, this.getCurrentDatabase()];
|
case 1:
|
currentDatabase = _b.sent();
|
tablesCondition = tableNames.map(function (tableName) {
|
var _a = tslib_1.__read(tableName.split("."), 2), database = _a[0], name = _a[1];
|
if (!name) {
|
name = database;
|
database = _this.driver.database || currentDatabase;
|
}
|
return "(`TABLE_SCHEMA` = '" + database + "' AND `TABLE_NAME` = '" + name + "')";
|
}).join(" OR ");
|
tablesSql = "SELECT * FROM `INFORMATION_SCHEMA`.`TABLES` WHERE " + tablesCondition;
|
columnsSql = "SELECT * FROM `INFORMATION_SCHEMA`.`COLUMNS` WHERE " + tablesCondition;
|
primaryKeySql = "SELECT * FROM `INFORMATION_SCHEMA`.`KEY_COLUMN_USAGE` WHERE `CONSTRAINT_NAME` = 'PRIMARY' AND (" + tablesCondition + ")";
|
collationsSql = "SELECT `SCHEMA_NAME`, `DEFAULT_CHARACTER_SET_NAME` as `CHARSET`, `DEFAULT_COLLATION_NAME` AS `COLLATION` FROM `INFORMATION_SCHEMA`.`SCHEMATA`";
|
indicesCondition = tableNames.map(function (tableName) {
|
var _a = tslib_1.__read(tableName.split("."), 2), database = _a[0], name = _a[1];
|
if (!name) {
|
name = database;
|
database = _this.driver.database || currentDatabase;
|
}
|
return "(`s`.`TABLE_SCHEMA` = '" + database + "' AND `s`.`TABLE_NAME` = '" + name + "')";
|
}).join(" OR ");
|
indicesSql = "SELECT `s`.* FROM `INFORMATION_SCHEMA`.`STATISTICS` `s` " +
|
"LEFT JOIN `INFORMATION_SCHEMA`.`REFERENTIAL_CONSTRAINTS` `rc` ON `s`.`INDEX_NAME` = `rc`.`CONSTRAINT_NAME` " +
|
("WHERE (" + indicesCondition + ") AND `s`.`INDEX_NAME` != 'PRIMARY' AND `rc`.`CONSTRAINT_NAME` IS NULL");
|
foreignKeysCondition = tableNames.map(function (tableName) {
|
var _a = tslib_1.__read(tableName.split("."), 2), database = _a[0], name = _a[1];
|
if (!name) {
|
name = database;
|
database = _this.driver.database || currentDatabase;
|
}
|
return "(`kcu`.`TABLE_SCHEMA` = '" + database + "' AND `kcu`.`TABLE_NAME` = '" + name + "')";
|
}).join(" OR ");
|
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 `INFORMATION_SCHEMA`.`KEY_COLUMN_USAGE` `kcu` " +
|
"INNER JOIN `INFORMATION_SCHEMA`.`REFERENTIAL_CONSTRAINTS` `rc` ON `rc`.`constraint_name` = `kcu`.`constraint_name` " +
|
"WHERE " + foreignKeysCondition;
|
return [4 /*yield*/, Promise.all([
|
this.query(tablesSql),
|
this.query(columnsSql),
|
this.query(primaryKeySql),
|
this.query(collationsSql),
|
this.query(indicesSql),
|
this.query(foreignKeysSql)
|
])];
|
case 2:
|
_a = tslib_1.__read.apply(void 0, [_b.sent(), 6]), dbTables = _a[0], dbColumns = _a[1], dbPrimaryKeys = _a[2], dbCollations = _a[3], dbIndices = _a[4], dbForeignKeys = _a[5];
|
// if tables were not found in the db, no need to proceed
|
if (!dbTables.length)
|
return [2 /*return*/, []];
|
isMariaDb = this.driver.options.type === "mariadb";
|
// create tables for loaded tables
|
return [2 /*return*/, Promise.all(dbTables.map(function (dbTable) { return tslib_1.__awaiter(_this, void 0, void 0, function () {
|
var table, dbCollation, defaultCollation, defaultCharset, db, tableFullName, tableForeignKeyConstraints, tableIndexConstraints;
|
var _this = this;
|
return tslib_1.__generator(this, function (_a) {
|
table = new Table_1.Table();
|
dbCollation = dbCollations.find(function (coll) { return coll["SCHEMA_NAME"] === dbTable["TABLE_SCHEMA"]; });
|
defaultCollation = dbCollation["COLLATION"];
|
defaultCharset = dbCollation["CHARSET"];
|
db = dbTable["TABLE_SCHEMA"] === currentDatabase ? undefined : dbTable["TABLE_SCHEMA"];
|
table.name = this.driver.buildTableName(dbTable["TABLE_NAME"], undefined, db);
|
tableFullName = this.driver.buildTableName(dbTable["TABLE_NAME"], undefined, dbTable["TABLE_SCHEMA"]);
|
// create columns from the loaded columns
|
table.columns = dbColumns
|
.filter(function (dbColumn) { return _this.driver.buildTableName(dbColumn["TABLE_NAME"], undefined, dbColumn["TABLE_SCHEMA"]) === tableFullName; })
|
.map(function (dbColumn) {
|
var columnUniqueIndex = dbIndices.find(function (dbIndex) {
|
return _this.driver.buildTableName(dbIndex["TABLE_NAME"], undefined, dbIndex["TABLE_SCHEMA"]) === tableFullName
|
&& dbIndex["COLUMN_NAME"] === dbColumn["COLUMN_NAME"] && dbIndex["NON_UNIQUE"] === "0";
|
});
|
var tableMetadata = _this.connection.entityMetadatas.find(function (metadata) { return metadata.tablePath === table.name; });
|
var hasIgnoredIndex = columnUniqueIndex && tableMetadata && tableMetadata.indices
|
.some(function (index) { return index.name === columnUniqueIndex["INDEX_NAME"] && index.synchronize === false; });
|
var isConstraintComposite = columnUniqueIndex
|
? !!dbIndices.find(function (dbIndex) { return dbIndex["INDEX_NAME"] === columnUniqueIndex["INDEX_NAME"] && dbIndex["COLUMN_NAME"] !== dbColumn["COLUMN_NAME"]; })
|
: false;
|
var tableColumn = new TableColumn_1.TableColumn();
|
tableColumn.name = dbColumn["COLUMN_NAME"];
|
tableColumn.type = dbColumn["DATA_TYPE"].toLowerCase();
|
if (_this.driver.withWidthColumnTypes.indexOf(tableColumn.type) !== -1) {
|
var width = dbColumn["COLUMN_TYPE"].substring(dbColumn["COLUMN_TYPE"].indexOf("(") + 1, dbColumn["COLUMN_TYPE"].indexOf(")"));
|
tableColumn.width = width && !_this.isDefaultColumnWidth(table, tableColumn, parseInt(width)) ? parseInt(width) : undefined;
|
}
|
if (dbColumn["COLUMN_DEFAULT"] === null
|
|| dbColumn["COLUMN_DEFAULT"] === undefined
|
|| (isMariaDb && dbColumn["COLUMN_DEFAULT"] === "NULL")) {
|
tableColumn.default = undefined;
|
}
|
else {
|
tableColumn.default = dbColumn["COLUMN_DEFAULT"] === "CURRENT_TIMESTAMP" ? dbColumn["COLUMN_DEFAULT"] : "'" + dbColumn["COLUMN_DEFAULT"] + "'";
|
}
|
if (dbColumn["EXTRA"].indexOf("on update") !== -1) {
|
tableColumn.onUpdate = dbColumn["EXTRA"].substring(dbColumn["EXTRA"].indexOf("on update") + 10);
|
}
|
if (dbColumn["GENERATION_EXPRESSION"]) {
|
tableColumn.asExpression = dbColumn["GENERATION_EXPRESSION"];
|
tableColumn.generatedType = dbColumn["EXTRA"].indexOf("VIRTUAL") !== -1 ? "VIRTUAL" : "STORED";
|
}
|
tableColumn.isUnique = !!columnUniqueIndex && !hasIgnoredIndex && !isConstraintComposite;
|
tableColumn.isNullable = dbColumn["IS_NULLABLE"] === "YES";
|
tableColumn.isPrimary = dbPrimaryKeys.some(function (dbPrimaryKey) {
|
return _this.driver.buildTableName(dbPrimaryKey["TABLE_NAME"], undefined, dbPrimaryKey["TABLE_SCHEMA"]) === tableFullName && dbPrimaryKey["COLUMN_NAME"] === tableColumn.name;
|
});
|
tableColumn.zerofill = dbColumn["COLUMN_TYPE"].indexOf("zerofill") !== -1;
|
tableColumn.unsigned = tableColumn.zerofill ? true : dbColumn["COLUMN_TYPE"].indexOf("unsigned") !== -1;
|
tableColumn.isGenerated = dbColumn["EXTRA"].indexOf("auto_increment") !== -1;
|
if (tableColumn.isGenerated)
|
tableColumn.generationStrategy = "increment";
|
tableColumn.comment = 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"]) {
|
var length = dbColumn["CHARACTER_MAXIMUM_LENGTH"].toString();
|
tableColumn.length = !_this.isDefaultColumnLength(table, tableColumn, length) ? length : "";
|
}
|
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") {
|
var colType = dbColumn["COLUMN_TYPE"];
|
var items = colType.substring(colType.indexOf("(") + 1, colType.indexOf(")")).split(",");
|
tableColumn.enum = items.map(function (item) {
|
return item.substring(1, item.length - 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;
|
});
|
tableForeignKeyConstraints = OrmUtils_1.OrmUtils.uniq(dbForeignKeys.filter(function (dbForeignKey) {
|
return _this.driver.buildTableName(dbForeignKey["TABLE_NAME"], undefined, dbForeignKey["TABLE_SCHEMA"]) === tableFullName;
|
}), function (dbForeignKey) { return dbForeignKey["CONSTRAINT_NAME"]; });
|
table.foreignKeys = tableForeignKeyConstraints.map(function (dbForeignKey) {
|
var foreignKeys = dbForeignKeys.filter(function (dbFk) { return 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.
|
var database = dbForeignKey["REFERENCED_TABLE_SCHEMA"] === currentDatabase ? undefined : dbForeignKey["REFERENCED_TABLE_SCHEMA"];
|
var referencedTableName = _this.driver.buildTableName(dbForeignKey["REFERENCED_TABLE_NAME"], undefined, database);
|
return new TableForeignKey_1.TableForeignKey({
|
name: dbForeignKey["CONSTRAINT_NAME"],
|
columnNames: foreignKeys.map(function (dbFk) { return dbFk["COLUMN_NAME"]; }),
|
referencedTableName: referencedTableName,
|
referencedColumnNames: foreignKeys.map(function (dbFk) { return dbFk["REFERENCED_COLUMN_NAME"]; }),
|
onDelete: dbForeignKey["ON_DELETE"],
|
onUpdate: dbForeignKey["ON_UPDATE"]
|
});
|
});
|
tableIndexConstraints = OrmUtils_1.OrmUtils.uniq(dbIndices.filter(function (dbIndex) {
|
return _this.driver.buildTableName(dbIndex["TABLE_NAME"], undefined, dbIndex["TABLE_SCHEMA"]) === tableFullName;
|
}), function (dbIndex) { return dbIndex["INDEX_NAME"]; });
|
table.indices = tableIndexConstraints.map(function (constraint) {
|
var indices = dbIndices.filter(function (index) {
|
return index["TABLE_SCHEMA"] === constraint["TABLE_SCHEMA"]
|
&& index["TABLE_NAME"] === constraint["TABLE_NAME"]
|
&& index["INDEX_NAME"] === constraint["INDEX_NAME"];
|
});
|
return new TableIndex_1.TableIndex({
|
table: table,
|
name: constraint["INDEX_NAME"],
|
columnNames: indices.map(function (i) { return i["COLUMN_NAME"]; }),
|
isUnique: constraint["NON_UNIQUE"] === "0",
|
isSpatial: constraint["INDEX_TYPE"] === "SPATIAL",
|
isFulltext: constraint["INDEX_TYPE"] === "FULLTEXT"
|
});
|
});
|
return [2 /*return*/, table];
|
});
|
}); }))];
|
}
|
});
|
});
|
};
|
/**
|
* Builds create table sql
|
*/
|
MysqlQueryRunner.prototype.createTableSql = function (table, createForeignKeys) {
|
var _this = this;
|
var columnDefinitions = table.columns.map(function (column) { return _this.buildCreateColumnSql(column, true); }).join(", ");
|
var 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(function (column) { return column.isUnique; })
|
.forEach(function (column) {
|
var isUniqueIndexExist = table.indices.some(function (index) {
|
return index.columnNames.length === 1 && !!index.isUnique && index.columnNames.indexOf(column.name) !== -1;
|
});
|
var isUniqueConstraintExist = table.uniques.some(function (unique) {
|
return unique.columnNames.length === 1 && unique.columnNames.indexOf(column.name) !== -1;
|
});
|
if (!isUniqueIndexExist && !isUniqueConstraintExist)
|
table.indices.push(new TableIndex_1.TableIndex({
|
name: _this.connection.namingStrategy.uniqueConstraintName(table.name, [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(function (unique) {
|
var uniqueExist = table.indices.some(function (index) { return 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) {
|
var indicesSql = table.indices.map(function (index) {
|
var columnNames = index.columnNames.map(function (columnName) { return "`" + columnName + "`"; }).join(", ");
|
if (!index.name)
|
index.name = _this.connection.namingStrategy.indexName(table.name, index.columnNames, index.where);
|
var indexType = "";
|
if (index.isUnique)
|
indexType += "UNIQUE ";
|
if (index.isSpatial)
|
indexType += "SPATIAL ";
|
if (index.isFulltext)
|
indexType += "FULLTEXT ";
|
return indexType + "INDEX `" + index.name + "` (" + columnNames + ")";
|
}).join(", ");
|
sql += ", " + indicesSql;
|
}
|
if (table.foreignKeys.length > 0 && createForeignKeys) {
|
var foreignKeysSql = table.foreignKeys.map(function (fk) {
|
var columnNames = fk.columnNames.map(function (columnName) { return "`" + columnName + "`"; }).join(", ");
|
if (!fk.name)
|
fk.name = _this.connection.namingStrategy.foreignKeyName(table.name, fk.columnNames);
|
var referencedColumnNames = fk.referencedColumnNames.map(function (columnName) { return "`" + columnName + "`"; }).join(", ");
|
var constraint = "CONSTRAINT `" + fk.name + "` FOREIGN KEY (" + columnNames + ") REFERENCES " + _this.escapePath(fk.referencedTableName) + " (" + 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) {
|
var columnNames = table.primaryColumns.map(function (column) { return "`" + column.name + "`"; }).join(", ");
|
sql += ", PRIMARY KEY (" + columnNames + ")";
|
}
|
sql += ") ENGINE=" + (table.engine || "InnoDB");
|
return new Query_1.Query(sql);
|
};
|
/**
|
* Builds drop table sql
|
*/
|
MysqlQueryRunner.prototype.dropTableSql = function (tableOrName) {
|
return new Query_1.Query("DROP TABLE " + this.escapePath(tableOrName));
|
};
|
MysqlQueryRunner.prototype.createViewSql = function (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.connection).getQuery());
|
}
|
};
|
MysqlQueryRunner.prototype.insertViewDefinitionSql = function (view) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var currentDatabase, expression, _a, query, parameters;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0: return [4 /*yield*/, this.getCurrentDatabase()];
|
case 1:
|
currentDatabase = _b.sent();
|
expression = typeof view.expression === "string" ? view.expression.trim() : view.expression(this.connection).getQuery();
|
_a = tslib_1.__read(this.connection.createQueryBuilder()
|
.insert()
|
.into(this.getTypeormMetadataTableName())
|
.values({ type: "VIEW", schema: currentDatabase, name: view.name, value: expression })
|
.getQueryAndParameters(), 2), query = _a[0], parameters = _a[1];
|
return [2 /*return*/, new Query_1.Query(query, parameters)];
|
}
|
});
|
});
|
};
|
/**
|
* Builds drop view sql.
|
*/
|
MysqlQueryRunner.prototype.dropViewSql = function (viewOrPath) {
|
return new Query_1.Query("DROP VIEW " + this.escapePath(viewOrPath));
|
};
|
/**
|
* Builds remove view sql.
|
*/
|
MysqlQueryRunner.prototype.deleteViewDefinitionSql = function (viewOrPath) {
|
return tslib_1.__awaiter(this, void 0, void 0, function () {
|
var currentDatabase, viewName, qb, _a, query, parameters;
|
return tslib_1.__generator(this, function (_b) {
|
switch (_b.label) {
|
case 0: return [4 /*yield*/, this.getCurrentDatabase()];
|
case 1:
|
currentDatabase = _b.sent();
|
viewName = viewOrPath instanceof View_1.View ? viewOrPath.name : viewOrPath;
|
qb = this.connection.createQueryBuilder();
|
_a = tslib_1.__read(qb.delete()
|
.from(this.getTypeormMetadataTableName())
|
.where(qb.escape("type") + " = 'VIEW'")
|
.andWhere(qb.escape("schema") + " = :schema", { schema: currentDatabase })
|
.andWhere(qb.escape("name") + " = :name", { name: viewName })
|
.getQueryAndParameters(), 2), query = _a[0], parameters = _a[1];
|
return [2 /*return*/, new Query_1.Query(query, parameters)];
|
}
|
});
|
});
|
};
|
/**
|
* Builds create index sql.
|
*/
|
MysqlQueryRunner.prototype.createIndexSql = function (table, index) {
|
var columns = index.columnNames.map(function (columnName) { return "`" + columnName + "`"; }).join(", ");
|
var indexType = "";
|
if (index.isUnique)
|
indexType += "UNIQUE ";
|
if (index.isSpatial)
|
indexType += "SPATIAL ";
|
if (index.isFulltext)
|
indexType += "FULLTEXT ";
|
return new Query_1.Query("CREATE " + indexType + "INDEX `" + index.name + "` ON " + this.escapePath(table) + " (" + columns + ")");
|
};
|
/**
|
* Builds drop index sql.
|
*/
|
MysqlQueryRunner.prototype.dropIndexSql = function (table, indexOrName) {
|
var indexName = indexOrName instanceof TableIndex_1.TableIndex ? indexOrName.name : indexOrName;
|
return new Query_1.Query("DROP INDEX `" + indexName + "` ON " + this.escapePath(table));
|
};
|
/**
|
* Builds create primary key sql.
|
*/
|
MysqlQueryRunner.prototype.createPrimaryKeySql = function (table, columnNames) {
|
var columnNamesString = columnNames.map(function (columnName) { return "`" + columnName + "`"; }).join(", ");
|
return new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " ADD PRIMARY KEY (" + columnNamesString + ")");
|
};
|
/**
|
* Builds drop primary key sql.
|
*/
|
MysqlQueryRunner.prototype.dropPrimaryKeySql = function (table) {
|
return new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " DROP PRIMARY KEY");
|
};
|
/**
|
* Builds create foreign key sql.
|
*/
|
MysqlQueryRunner.prototype.createForeignKeySql = function (table, foreignKey) {
|
var columnNames = foreignKey.columnNames.map(function (column) { return "`" + column + "`"; }).join(", ");
|
var referencedColumnNames = foreignKey.referencedColumnNames.map(function (column) { return "`" + column + "`"; }).join(",");
|
var sql = "ALTER TABLE " + this.escapePath(table) + " ADD CONSTRAINT `" + foreignKey.name + "` FOREIGN KEY (" + columnNames + ") " +
|
("REFERENCES " + this.escapePath(foreignKey.referencedTableName) + "(" + 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.
|
*/
|
MysqlQueryRunner.prototype.dropForeignKeySql = function (table, foreignKeyOrName) {
|
var foreignKeyName = foreignKeyOrName instanceof TableForeignKey_1.TableForeignKey ? foreignKeyOrName.name : foreignKeyOrName;
|
return new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " DROP FOREIGN KEY `" + foreignKeyName + "`");
|
};
|
MysqlQueryRunner.prototype.parseTableName = function (target) {
|
var tableName = target instanceof Table_1.Table ? target.name : target;
|
return {
|
database: tableName.indexOf(".") !== -1 ? tableName.split(".")[0] : this.driver.database,
|
tableName: tableName.indexOf(".") !== -1 ? tableName.split(".")[1] : tableName
|
};
|
};
|
/**
|
* Escapes given table or view path.
|
*/
|
MysqlQueryRunner.prototype.escapePath = function (target, disableEscape) {
|
var tableName = target instanceof Table_1.Table || target instanceof View_1.View ? target.name : target;
|
return tableName.split(".").map(function (i) { return disableEscape ? i : "`" + i + "`"; }).join(".");
|
};
|
/**
|
* Builds a part of query to create/change a column.
|
*/
|
MysqlQueryRunner.prototype.buildCreateColumnSql = function (column, skipPrimary, skipName) {
|
if (skipName === void 0) { skipName = false; }
|
var c = "";
|
if (skipName) {
|
c = this.connection.driver.createFullType(column);
|
}
|
else {
|
c = "`" + column.name + "` " + this.connection.driver.createFullType(column);
|
}
|
if (column.asExpression)
|
c += " AS (" + column.asExpression + ") " + (column.generatedType ? column.generatedType : "VIRTUAL");
|
// if you specify ZEROFILL for a numeric column, MySQL automatically adds the UNSIGNED attribute to that column.
|
if (column.zerofill) {
|
c += " ZEROFILL";
|
}
|
else if (column.unsigned) {
|
c += " UNSIGNED";
|
}
|
if (column.enum)
|
c += " (" + column.enum.map(function (value) { return "'" + value + "'"; }).join(", ") + ")";
|
if (column.charset)
|
c += " CHARACTER SET \"" + column.charset + "\"";
|
if (column.collation)
|
c += " COLLATE \"" + column.collation + "\"";
|
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)
|
c += " COMMENT '" + column.comment + "'";
|
if (column.default !== undefined && column.default !== null)
|
c += " DEFAULT " + column.default;
|
if (column.onUpdate)
|
c += " ON UPDATE " + column.onUpdate;
|
return c;
|
};
|
return MysqlQueryRunner;
|
}(BaseQueryRunner_1.BaseQueryRunner));
|
exports.MysqlQueryRunner = MysqlQueryRunner;
|
|
//# sourceMappingURL=MysqlQueryRunner.js.map
|
