typeorm
Version:
Data-Mapper ORM for TypeScript, ES7, ES6, ES5. Supports MySQL, PostgreSQL, MariaDB, SQLite, MS SQL Server, Oracle, MongoDB databases.
917 lines • 193 kB
JavaScript
"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.PostgresQueryRunner = void 0;
var tslib_1 = require("tslib");
var QueryFailedError_1 = require("../../error/QueryFailedError");
var QueryRunnerAlreadyReleasedError_1 = require("../../error/QueryRunnerAlreadyReleasedError");
var TransactionAlreadyStartedError_1 = require("../../error/TransactionAlreadyStartedError");
var TransactionNotStartedError_1 = require("../../error/TransactionNotStartedError");
var BaseQueryRunner_1 = require("../../query-runner/BaseQueryRunner");
var Table_1 = require("../../schema-builder/table/Table");
var TableCheck_1 = require("../../schema-builder/table/TableCheck");
var TableColumn_1 = require("../../schema-builder/table/TableColumn");
var TableExclusion_1 = require("../../schema-builder/table/TableExclusion");
var TableForeignKey_1 = require("../../schema-builder/table/TableForeignKey");
var TableIndex_1 = require("../../schema-builder/table/TableIndex");
var TableUnique_1 = require("../../schema-builder/table/TableUnique");
var View_1 = require("../../schema-builder/view/View");
var Broadcaster_1 = require("../../subscriber/Broadcaster");
var OrmUtils_1 = require("../../util/OrmUtils");
var Query_1 = require("../Query");
var BroadcasterResult_1 = require("../../subscriber/BroadcasterResult");
/**
* Runs queries on a single postgres database connection.
*/
var PostgresQueryRunner = /** @class */ (function (_super) {
tslib_1.__extends(PostgresQueryRunner, _super);
// -------------------------------------------------------------------------
// Constructor
// -------------------------------------------------------------------------
function PostgresQueryRunner(driver, mode) {
var _this = _super.call(this) || this;
_this.driver = driver;
_this.connection = driver.connection;
_this.mode = mode;
_this.broadcaster = new Broadcaster_1.Broadcaster(_this);
return _this;
}
// -------------------------------------------------------------------------
// Public Methods
// -------------------------------------------------------------------------
/**
* Creates/uses database connection from the connection pool to perform further operations.
* Returns obtained database connection.
*/
PostgresQueryRunner.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 (_a) {
var _b = tslib_1.__read(_a, 2), connection = _b[0], release = _b[1];
_this.driver.connectedQueryRunners.push(_this);
_this.databaseConnection = connection;
var onErrorCallback = function () { return _this.release(); };
_this.releaseCallback = function () {
_this.databaseConnection.removeListener("error", onErrorCallback);
release();
};
_this.databaseConnection.on("error", onErrorCallback);
return _this.databaseConnection;
});
}
else { // master
this.databaseConnectionPromise = this.driver.obtainMasterConnection().then(function (_a) {
var _b = tslib_1.__read(_a, 2), connection = _b[0], release = _b[1];
_this.driver.connectedQueryRunners.push(_this);
_this.databaseConnection = connection;
var onErrorCallback = function () { return _this.release(); };
_this.releaseCallback = function () {
_this.databaseConnection.removeListener("error", onErrorCallback);
release();
};
_this.databaseConnection.on("error", onErrorCallback);
return _this.databaseConnection;
});
}
return this.databaseConnectionPromise;
};
/**
* Releases used database connection.
* You cannot use query runner methods once its released.
*/
PostgresQueryRunner.prototype.release = function () {
if (this.isReleased) {
return Promise.resolve();
}
this.isReleased = true;
if (this.releaseCallback)
this.releaseCallback();
var index = this.driver.connectedQueryRunners.indexOf(this);
if (index !== -1)
this.driver.connectedQueryRunners.splice(index, 1);
return Promise.resolve();
};
/**
* Starts transaction.
*/
PostgresQueryRunner.prototype.startTransaction = function (isolationLevel) {
return tslib_1.__awaiter(this, void 0, void 0, function () {
var beforeBroadcastResult, afterBroadcastResult;
return tslib_1.__generator(this, function (_a) {
switch (_a.label) {
case 0:
if (this.isTransactionActive)
throw new TransactionAlreadyStartedError_1.TransactionAlreadyStartedError();
beforeBroadcastResult = new BroadcasterResult_1.BroadcasterResult();
this.broadcaster.broadcastBeforeTransactionStartEvent(beforeBroadcastResult);
if (!(beforeBroadcastResult.promises.length > 0)) return [3 /*break*/, 2];
return [4 /*yield*/, Promise.all(beforeBroadcastResult.promises)];
case 1:
_a.sent();
_a.label = 2;
case 2:
this.isTransactionActive = true;
return [4 /*yield*/, this.query("START TRANSACTION")];
case 3:
_a.sent();
if (!isolationLevel) return [3 /*break*/, 5];
return [4 /*yield*/, this.query("SET TRANSACTION ISOLATION LEVEL " + isolationLevel)];
case 4:
_a.sent();
_a.label = 5;
case 5:
afterBroadcastResult = new BroadcasterResult_1.BroadcasterResult();
this.broadcaster.broadcastAfterTransactionStartEvent(afterBroadcastResult);
if (!(afterBroadcastResult.promises.length > 0)) return [3 /*break*/, 7];
return [4 /*yield*/, Promise.all(afterBroadcastResult.promises)];
case 6:
_a.sent();
_a.label = 7;
case 7: return [2 /*return*/];
}
});
});
};
/**
* Commits transaction.
* Error will be thrown if transaction was not started.
*/
PostgresQueryRunner.prototype.commitTransaction = function () {
return tslib_1.__awaiter(this, void 0, void 0, function () {
var beforeBroadcastResult, afterBroadcastResult;
return tslib_1.__generator(this, function (_a) {
switch (_a.label) {
case 0:
if (!this.isTransactionActive)
throw new TransactionNotStartedError_1.TransactionNotStartedError();
beforeBroadcastResult = new BroadcasterResult_1.BroadcasterResult();
this.broadcaster.broadcastBeforeTransactionCommitEvent(beforeBroadcastResult);
if (!(beforeBroadcastResult.promises.length > 0)) return [3 /*break*/, 2];
return [4 /*yield*/, Promise.all(beforeBroadcastResult.promises)];
case 1:
_a.sent();
_a.label = 2;
case 2: return [4 /*yield*/, this.query("COMMIT")];
case 3:
_a.sent();
this.isTransactionActive = false;
afterBroadcastResult = new BroadcasterResult_1.BroadcasterResult();
this.broadcaster.broadcastAfterTransactionCommitEvent(afterBroadcastResult);
if (!(afterBroadcastResult.promises.length > 0)) return [3 /*break*/, 5];
return [4 /*yield*/, Promise.all(afterBroadcastResult.promises)];
case 4:
_a.sent();
_a.label = 5;
case 5: return [2 /*return*/];
}
});
});
};
/**
* Rollbacks transaction.
* Error will be thrown if transaction was not started.
*/
PostgresQueryRunner.prototype.rollbackTransaction = function () {
return tslib_1.__awaiter(this, void 0, void 0, function () {
var beforeBroadcastResult, afterBroadcastResult;
return tslib_1.__generator(this, function (_a) {
switch (_a.label) {
case 0:
if (!this.isTransactionActive)
throw new TransactionNotStartedError_1.TransactionNotStartedError();
beforeBroadcastResult = new BroadcasterResult_1.BroadcasterResult();
this.broadcaster.broadcastBeforeTransactionRollbackEvent(beforeBroadcastResult);
if (!(beforeBroadcastResult.promises.length > 0)) return [3 /*break*/, 2];
return [4 /*yield*/, Promise.all(beforeBroadcastResult.promises)];
case 1:
_a.sent();
_a.label = 2;
case 2: return [4 /*yield*/, this.query("ROLLBACK")];
case 3:
_a.sent();
this.isTransactionActive = false;
afterBroadcastResult = new BroadcasterResult_1.BroadcasterResult();
this.broadcaster.broadcastAfterTransactionRollbackEvent(afterBroadcastResult);
if (!(afterBroadcastResult.promises.length > 0)) return [3 /*break*/, 5];
return [4 /*yield*/, Promise.all(afterBroadcastResult.promises)];
case 4:
_a.sent();
_a.label = 5;
case 5: return [2 /*return*/];
}
});
});
};
/**
* Executes a given SQL query.
*/
PostgresQueryRunner.prototype.query = function (query, parameters) {
return tslib_1.__awaiter(this, void 0, void 0, function () {
var databaseConnection, queryStartTime, result, maxQueryExecutionTime, queryEndTime, queryExecutionTime, err_1;
return tslib_1.__generator(this, function (_a) {
switch (_a.label) {
case 0:
if (this.isReleased)
throw new QueryRunnerAlreadyReleasedError_1.QueryRunnerAlreadyReleasedError();
return [4 /*yield*/, this.connect()];
case 1:
databaseConnection = _a.sent();
this.driver.connection.logger.logQuery(query, parameters, this);
_a.label = 2;
case 2:
_a.trys.push([2, 4, , 5]);
queryStartTime = +new Date();
return [4 /*yield*/, databaseConnection.query(query, parameters)];
case 3:
result = _a.sent();
maxQueryExecutionTime = this.driver.connection.options.maxQueryExecutionTime;
queryEndTime = +new Date();
queryExecutionTime = queryEndTime - queryStartTime;
if (maxQueryExecutionTime && queryExecutionTime > maxQueryExecutionTime)
this.driver.connection.logger.logQuerySlow(queryExecutionTime, query, parameters, this);
switch (result.command) {
case "DELETE":
case "UPDATE":
// for UPDATE and DELETE query additionally return number of affected rows
return [2 /*return*/, [result.rows, result.rowCount]];
break;
default:
return [2 /*return*/, result.rows];
}
return [3 /*break*/, 5];
case 4:
err_1 = _a.sent();
this.driver.connection.logger.logQueryError(err_1, query, parameters, this);
throw new QueryFailedError_1.QueryFailedError(query, parameters, err_1);
case 5: return [2 /*return*/];
}
});
});
};
/**
* Returns raw data stream.
*/
PostgresQueryRunner.prototype.stream = function (query, parameters, onEnd, onError) {
var _this = this;
var QueryStream = this.driver.loadStreamDependency();
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(new QueryStream(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.
*/
PostgresQueryRunner.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.
*/
PostgresQueryRunner.prototype.getSchemas = function (database) {
return tslib_1.__awaiter(this, void 0, void 0, function () {
return tslib_1.__generator(this, function (_a) {
return [2 /*return*/, Promise.resolve([])];
});
});
};
/**
* Checks if database with the given name exist.
*/
PostgresQueryRunner.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 pg_database WHERE datname='" + database + "';")];
case 1:
result = _a.sent();
return [2 /*return*/, result.length ? true : false];
}
});
});
};
/**
* Loads currently using database
*/
PostgresQueryRunner.prototype.getCurrentDatabase = function () {
return tslib_1.__awaiter(this, void 0, void 0, function () {
var query;
return tslib_1.__generator(this, function (_a) {
switch (_a.label) {
case 0: return [4 /*yield*/, this.query("SELECT * FROM current_database()")];
case 1:
query = _a.sent();
return [2 /*return*/, query[0]["current_database"]];
}
});
});
};
/**
* Checks if schema with the given name exist.
*/
PostgresQueryRunner.prototype.hasSchema = function (schema) {
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\" = '" + schema + "'")];
case 1:
result = _a.sent();
return [2 /*return*/, result.length ? true : false];
}
});
});
};
/**
* Loads currently using database schema
*/
PostgresQueryRunner.prototype.getCurrentSchema = function () {
return tslib_1.__awaiter(this, void 0, void 0, function () {
var query;
return tslib_1.__generator(this, function (_a) {
switch (_a.label) {
case 0: return [4 /*yield*/, this.query("SELECT * FROM current_schema()")];
case 1:
query = _a.sent();
return [2 /*return*/, query[0]["current_schema"]];
}
});
});
};
/**
* Checks if table with the given name exist in the database.
*/
PostgresQueryRunner.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\".\"tables\" WHERE \"table_schema\" = " + parsedTableName.schema + " 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.
*/
PostgresQueryRunner.prototype.hasColumn = function (tableOrName, columnName) {
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.schema + " 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.
* Note: Postgres does not support database creation inside a transaction block.
*/
PostgresQueryRunner.prototype.createDatabase = function (database, ifNotExist) {
return tslib_1.__awaiter(this, void 0, void 0, function () {
var databaseAlreadyExists, up, down;
return tslib_1.__generator(this, function (_a) {
switch (_a.label) {
case 0:
if (!ifNotExist) return [3 /*break*/, 2];
return [4 /*yield*/, this.hasDatabase(database)];
case 1:
databaseAlreadyExists = _a.sent();
if (databaseAlreadyExists)
return [2 /*return*/, Promise.resolve()];
_a.label = 2;
case 2:
up = "CREATE DATABASE \"" + database + "\"";
down = "DROP DATABASE \"" + database + "\"";
return [4 /*yield*/, this.executeQueries(new Query_1.Query(up), new Query_1.Query(down))];
case 3:
_a.sent();
return [2 /*return*/];
}
});
});
};
/**
* Drops database.
* Note: Postgres does not support database dropping inside a transaction block.
*/
PostgresQueryRunner.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.
*/
PostgresQueryRunner.prototype.createSchema = function (schema, 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 SCHEMA IF NOT EXISTS \"" + schema + "\"" : "CREATE SCHEMA \"" + schema + "\"";
down = "DROP SCHEMA \"" + schema + "\" CASCADE";
return [4 /*yield*/, this.executeQueries(new Query_1.Query(up), new Query_1.Query(down))];
case 1:
_a.sent();
return [2 /*return*/];
}
});
});
};
/**
* Drops table schema.
*/
PostgresQueryRunner.prototype.dropSchema = function (schemaPath, ifExist, isCascade) {
return tslib_1.__awaiter(this, void 0, void 0, function () {
var schema, up, down;
return tslib_1.__generator(this, function (_a) {
switch (_a.label) {
case 0:
schema = schemaPath.indexOf(".") === -1 ? schemaPath : schemaPath.split(".")[0];
up = ifExist ? "DROP SCHEMA IF EXISTS \"" + schema + "\" " + (isCascade ? "CASCADE" : "") : "DROP SCHEMA \"" + schema + "\" " + (isCascade ? "CASCADE" : "");
down = "CREATE SCHEMA \"" + schema + "\"";
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.
*/
PostgresQueryRunner.prototype.createTable = function (table, ifNotExist, createForeignKeys, createIndices) {
if (ifNotExist === void 0) { ifNotExist = false; }
if (createForeignKeys === void 0) { createForeignKeys = true; }
if (createIndices === void 0) { createIndices = true; }
return tslib_1.__awaiter(this, void 0, void 0, function () {
var isTableExist, upQueries, downQueries, enumColumns, createdEnumTypes, enumColumns_1, enumColumns_1_1, column, hasEnum, enumName, e_1_1;
var e_1, _a;
var _this = this;
return tslib_1.__generator(this, function (_b) {
switch (_b.label) {
case 0:
if (!ifNotExist) return [3 /*break*/, 2];
return [4 /*yield*/, this.hasTable(table)];
case 1:
isTableExist = _b.sent();
if (isTableExist)
return [2 /*return*/, Promise.resolve()];
_b.label = 2;
case 2:
upQueries = [];
downQueries = [];
enumColumns = table.columns.filter(function (column) { return column.type === "enum" || column.type === "simple-enum"; });
createdEnumTypes = [];
_b.label = 3;
case 3:
_b.trys.push([3, 8, 9, 10]);
enumColumns_1 = tslib_1.__values(enumColumns), enumColumns_1_1 = enumColumns_1.next();
_b.label = 4;
case 4:
if (!!enumColumns_1_1.done) return [3 /*break*/, 7];
column = enumColumns_1_1.value;
return [4 /*yield*/, this.hasEnumType(table, column)];
case 5:
hasEnum = _b.sent();
enumName = this.buildEnumName(table, column);
// if enum with the same "enumName" is defined more then once, me must prevent double creation
if (!hasEnum && createdEnumTypes.indexOf(enumName) === -1) {
createdEnumTypes.push(enumName);
upQueries.push(this.createEnumTypeSql(table, column, enumName));
downQueries.push(this.dropEnumTypeSql(table, column, enumName));
}
_b.label = 6;
case 6:
enumColumns_1_1 = enumColumns_1.next();
return [3 /*break*/, 4];
case 7: return [3 /*break*/, 10];
case 8:
e_1_1 = _b.sent();
e_1 = { error: e_1_1 };
return [3 /*break*/, 10];
case 9:
try {
if (enumColumns_1_1 && !enumColumns_1_1.done && (_a = enumColumns_1.return)) _a.call(enumColumns_1);
}
finally { if (e_1) throw e_1.error; }
return [7 /*endfinally*/];
case 10:
upQueries.push(this.createTableSql(table, createForeignKeys));
downQueries.push(this.dropTableSql(table));
// 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)); });
if (createIndices) {
table.indices.forEach(function (index) {
// 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);
upQueries.push(_this.createIndexSql(table, index));
downQueries.push(_this.dropIndexSql(table, index));
});
}
return [4 /*yield*/, this.executeQueries(upQueries, downQueries)];
case 11:
_b.sent();
return [2 /*return*/];
}
});
});
};
/**
* Drops the table.
*/
PostgresQueryRunner.prototype.dropTable = function (target, ifExist, dropForeignKeys, dropIndices) {
if (dropForeignKeys === void 0) { dropForeignKeys = true; }
if (dropIndices === void 0) { dropIndices = 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 (dropIndices) {
table.indices.forEach(function (index) {
upQueries.push(_this.dropIndexSql(table, index));
downQueries.push(_this.createIndexSql(table, index));
});
}
if (dropForeignKeys)
table.foreignKeys.forEach(function (foreignKey) { return upQueries.push(_this.dropForeignKeySql(table, foreignKey)); });
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.
*/
PostgresQueryRunner.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.
*/
PostgresQueryRunner.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 the given table.
*/
PostgresQueryRunner.prototype.renameTable = function (oldTableOrName, newTableName) {
return tslib_1.__awaiter(this, void 0, void 0, function () {
var upQueries, downQueries, oldTable, _a, newTable, oldTableName, schemaName, columnNames, oldPkName, newPkName, enumColumns, enumColumns_2, enumColumns_2_1, column, oldEnumType, e_2_1;
var e_2, _b;
var _this = this;
return tslib_1.__generator(this, function (_c) {
switch (_c.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 = _c.sent();
_c.label = 3;
case 3:
oldTable = _a;
newTable = oldTable.clone();
oldTableName = oldTable.name.indexOf(".") === -1 ? oldTable.name : oldTable.name.split(".")[1];
schemaName = oldTable.name.indexOf(".") === -1 ? undefined : oldTable.name.split(".")[0];
newTable.name = schemaName ? schemaName + "." + newTableName : newTableName;
upQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(oldTable) + " RENAME TO \"" + newTableName + "\""));
downQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(newTable) + " RENAME TO \"" + oldTableName + "\""));
// rename column primary key constraint
if (newTable.primaryColumns.length > 0) {
columnNames = newTable.primaryColumns.map(function (column) { return column.name; });
oldPkName = this.connection.namingStrategy.primaryKeyName(oldTable, columnNames);
newPkName = this.connection.namingStrategy.primaryKeyName(newTable, columnNames);
upQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(newTable) + " RENAME CONSTRAINT \"" + oldPkName + "\" TO \"" + newPkName + "\""));
downQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(newTable) + " RENAME CONSTRAINT \"" + newPkName + "\" TO \"" + oldPkName + "\""));
}
// rename sequences
newTable.columns.map(function (col) {
if (col.isGenerated && col.generationStrategy === "increment") {
var seqName = _this.buildSequenceName(oldTable, col.name, undefined, true, true);
var newSeqName = _this.buildSequenceName(newTable, col.name, undefined, true, true);
var up = schemaName ? "ALTER SEQUENCE \"" + schemaName + "\".\"" + seqName + "\" RENAME TO \"" + newSeqName + "\"" : "ALTER SEQUENCE \"" + seqName + "\" RENAME TO \"" + newSeqName + "\"";
var down = schemaName ? "ALTER SEQUENCE \"" + schemaName + "\".\"" + newSeqName + "\" RENAME TO \"" + seqName + "\"" : "ALTER SEQUENCE \"" + newSeqName + "\" RENAME TO \"" + seqName + "\"";
upQueries.push(new Query_1.Query(up));
downQueries.push(new Query_1.Query(down));
}
});
// rename unique constraints
newTable.uniques.forEach(function (unique) {
// build new constraint name
var newUniqueName = _this.connection.namingStrategy.uniqueConstraintName(newTable, unique.columnNames);
// build queries
upQueries.push(new Query_1.Query("ALTER TABLE " + _this.escapePath(newTable) + " RENAME CONSTRAINT \"" + unique.name + "\" TO \"" + newUniqueName + "\""));
downQueries.push(new Query_1.Query("ALTER TABLE " + _this.escapePath(newTable) + " RENAME CONSTRAINT \"" + newUniqueName + "\" TO \"" + unique.name + "\""));
// replace constraint name
unique.name = newUniqueName;
});
// rename index constraints
newTable.indices.forEach(function (index) {
// build new constraint name
var schema = _this.extractSchema(newTable);
var newIndexName = _this.connection.namingStrategy.indexName(newTable, index.columnNames, index.where);
// build queries
var up = schema ? "ALTER INDEX \"" + schema + "\".\"" + index.name + "\" RENAME TO \"" + newIndexName + "\"" : "ALTER INDEX \"" + index.name + "\" RENAME TO \"" + newIndexName + "\"";
var down = schema ? "ALTER INDEX \"" + schema + "\".\"" + newIndexName + "\" RENAME TO \"" + index.name + "\"" : "ALTER INDEX \"" + newIndexName + "\" RENAME TO \"" + index.name + "\"";
upQueries.push(new Query_1.Query(up));
downQueries.push(new Query_1.Query(down));
// replace constraint name
index.name = newIndexName;
});
// rename foreign key constraints
newTable.foreignKeys.forEach(function (foreignKey) {
// build new constraint name
var newForeignKeyName = _this.connection.namingStrategy.foreignKeyName(newTable, foreignKey.columnNames, foreignKey.referencedTableName, foreignKey.referencedColumnNames);
// build queries
upQueries.push(new Query_1.Query("ALTER TABLE " + _this.escapePath(newTable) + " RENAME CONSTRAINT \"" + foreignKey.name + "\" TO \"" + newForeignKeyName + "\""));
downQueries.push(new Query_1.Query("ALTER TABLE " + _this.escapePath(newTable) + " RENAME CONSTRAINT \"" + newForeignKeyName + "\" TO \"" + foreignKey.name + "\""));
// replace constraint name
foreignKey.name = newForeignKeyName;
});
enumColumns = newTable.columns.filter(function (column) { return column.type === "enum" || column.type === "simple-enum"; });
_c.label = 4;
case 4:
_c.trys.push([4, 9, 10, 11]);
enumColumns_2 = tslib_1.__values(enumColumns), enumColumns_2_1 = enumColumns_2.next();
_c.label = 5;
case 5:
if (!!enumColumns_2_1.done) return [3 /*break*/, 8];
column = enumColumns_2_1.value;
// skip renaming for user-defined enum name
if (column.enumName)
return [3 /*break*/, 7];
return [4 /*yield*/, this.getUserDefinedTypeName(oldTable, column)];
case 6:
oldEnumType = _c.sent();
upQueries.push(new Query_1.Query("ALTER TYPE \"" + oldEnumType.schema + "\".\"" + oldEnumType.name + "\" RENAME TO " + this.buildEnumName(newTable, column, false)));
downQueries.push(new Query_1.Query("ALTER TYPE " + this.buildEnumName(newTable, column) + " RENAME TO \"" + oldEnumType.name + "\""));
_c.label = 7;
case 7:
enumColumns_2_1 = enumColumns_2.next();
return [3 /*break*/, 5];
case 8: return [3 /*break*/, 11];
case 9:
e_2_1 = _c.sent();
e_2 = { error: e_2_1 };
return [3 /*break*/, 11];
case 10:
try {
if (enumColumns_2_1 && !enumColumns_2_1.done && (_b = enumColumns_2.return)) _b.call(enumColumns_2);
}
finally { if (e_2) throw e_2.error; }
return [7 /*endfinally*/];
case 11: return [4 /*yield*/, this.executeQueries(upQueries, downQueries)];
case 12:
_c.sent();
return [2 /*return*/];
}
});
});
};
/**
* Creates a new column from the column in the table.
*/
PostgresQueryRunner.prototype.addColumn = function (tableOrName, column) {
return tslib_1.__awaiter(this, void 0, void 0, function () {
var table, _a, clonedTable, upQueries, downQueries, hasEnum, primaryColumns, pkName_1, columnNames_1, pkName, columnNames, columnIndex, uniqueConstraint;
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 = [];
if (!(column.type === "enum" || column.type === "simple-enum")) return [3 /*break*/, 5];
return [4 /*yield*/, this.hasEnumType(table, column)];
case 4:
hasEnum = _b.sent();
if (!hasEnum) {
upQueries.push(this.createEnumTypeSql(table, column));
downQueries.push(this.dropEnumTypeSql(table, column));
}
_b.label = 5;
case 5:
upQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " ADD " + this.buildCreateColumnSql(table, column)));
downQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " DROP COLUMN \"" + column.name + "\""));
// create or update primary key constraint
if (column.isPrimary) {
primaryColumns = clonedTable.primaryColumns;
// if table already have primary key, me must drop it and recreate again
if (primaryColumns.length > 0) {
pkName_1 = this.connection.namingStrategy.primaryKeyName(clonedTable.name, primaryColumns.map(function (column) { return column.name; }));
columnNames_1 = primaryColumns.map(function (column) { return "\"" + column.name + "\""; }).join(", ");
upQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " DROP CONSTRAINT \"" + pkName_1 + "\""));
downQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " ADD CONSTRAINT \"" + pkName_1 + "\" PRIMARY KEY (" + columnNames_1 + ")"));
}
primaryColumns.push(column);
pkName = this.connection.namingStrategy.primaryKeyName(clonedTable.name, primaryColumns.map(function (column) { return column.name; }));
columnNames = primaryColumns.map(function (column) { return "\"" + column.name + "\""; }).join(", ");
upQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " ADD CONSTRAINT \"" + pkName + "\" PRIMARY KEY (" + columnNames + ")"));
downQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " DROP CONSTRAINT \"" + pkName + "\""));
}
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));
}
// create unique constraint
if (column.isUnique) {
uniqueConstraint = new TableUnique_1.TableUnique({
name: this.connection.namingStrategy.uniqueConstraintName(table.name, [column.name]),
columnNames: [column.name]
});
clonedTable.uniques.push(uniqueConstraint);
upQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " ADD CONSTRAINT \"" + uniqueConstraint.name + "\" UNIQUE (\"" + column.name + "\")"));
downQueries.push(new Query_1.Query("ALTER TABLE " + this.escapePath(table) + " DROP CONSTRAINT \"" + uniqueConstraint.name + "\""));
}
// create column's comment
if (column.comment) {
upQueries.push(new Query_1.Query("COMMENT ON COLUMN " + this.escapePath(table) + ".\"" + column.name + "\" IS " + this.escapeComment(column.comment)));
downQueries.push(new Query_1.Query("COMMENT ON COLUMN " + this.escapePath(table) + ".\"" + column.name + "\" IS " + this.escapeComment(column.comment)));
}
return [4 /*yield*/, this.executeQueries(upQueries, downQueries)];
case 6:
_b.sent();
clonedTable.addColumn(column);
this.replaceCachedTable(table, clonedTable);
return [2 /*return*/];
}
});
});
};
/**
* Creates a new columns from the column in the table.
*/
PostgresQueryRunner.prototype.addColumns = function (tableOrName, columns) {
return tslib_1.__awaiter(this, void 0, void 0, function () {
var columns_1, columns_1_1, column, e_3_1;
var e_3, _a;
return tslib_1.__generator(this, function (_b) {
switch (_b.label) {
case 0:
_b.trys.push([0, 5, 6, 7]);
columns_1 = tslib_1.__values(columns), columns_1_1 = columns_1.next();
_b.label = 1;
case 1:
if (!!columns_1_1.done) return [3 /*break*/, 4];
column = columns_1_1.value;
return [4 /*yield*/, this.addColumn(tableOrName, column)];
case 2:
_b.sent();
_b.label = 3;
case 3:
columns_1_1 = columns_1.next();
return [3 /*break*/, 1];
case 4: return [3 /*break*/, 7];
case 5:
e_3_1 = _b.sent();
e_3 = { error: e_3_1 };
return [3 /*break*/, 7];
case 6:
try {
if (columns_1_1 && !columns_1_1.done && (_a =