typeorm
Version:
Data-Mapper ORM for TypeScript, ES7, ES6, ES5. Supports MySQL, PostgreSQL, MariaDB, SQLite, MS SQL Server, Oracle, MongoDB databases.
484 lines (482 loc) • 24.6 kB
JavaScript
var __extends = (this && this.__extends) || (function () {
var extendStatics = Object.setPrototypeOf ||
({ __proto__: [] } instanceof Array && function (d, b) { d.__proto__ = b; }) ||
function (d, b) { for (var p in b) if (b.hasOwnProperty(p)) d[p] = b[p]; };
return function (d, b) {
extendStatics(d, b);
function __() { this.constructor = d; }
d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __());
};
})();
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : new P(function (resolve) { resolve(result.value); }).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __generator = (this && this.__generator) || function (thisArg, body) {
var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g;
return g = { next: verb(0), "throw": verb(1), "return": verb(2) }, typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
function verb(n) { return function (v) { return step([n, v]); }; }
function step(op) {
if (f) throw new TypeError("Generator is already executing.");
while (_) try {
if (f = 1, y && (t = y[op[0] & 2 ? "return" : op[0] ? "throw" : "next"]) && !(t = t.call(y, op[1])).done) return t;
if (y = 0, t) op = [0, t.value];
switch (op[0]) {
case 0: case 1: t = op; break;
case 4: _.label++; return { value: op[1], done: false };
case 5: _.label++; y = op[1]; op = [0]; continue;
case 7: op = _.ops.pop(); _.trys.pop(); continue;
default:
if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
if (t[2]) _.ops.pop();
_.trys.pop(); continue;
}
op = body.call(thisArg, _);
} catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
}
};
import { QueryBuilder } from "./QueryBuilder";
import { SqlServerDriver } from "../driver/sqlserver/SqlServerDriver";
import { PostgresDriver } from "../driver/postgres/PostgresDriver";
import { EntityMetadata } from "../metadata/EntityMetadata";
import { UpdateResult } from "./result/UpdateResult";
import { ReturningStatementNotSupportedError } from "../error/ReturningStatementNotSupportedError";
import { ReturningResultsEntityUpdator } from "./ReturningResultsEntityUpdator";
import { SqljsDriver } from "../driver/sqljs/SqljsDriver";
import { MysqlDriver } from "../driver/mysql/MysqlDriver";
import { BroadcasterResult } from "../subscriber/BroadcasterResult";
import { AbstractSqliteDriver } from "../driver/sqlite-abstract/AbstractSqliteDriver";
import { LimitOnUpdateNotSupportedError } from "../error/LimitOnUpdateNotSupportedError";
import { OracleDriver } from "../driver/oracle/OracleDriver";
/**
* Allows to build complex sql queries in a fashion way and execute those queries.
*/
var UpdateQueryBuilder = /** @class */ (function (_super) {
__extends(UpdateQueryBuilder, _super);
// -------------------------------------------------------------------------
// Constructor
// -------------------------------------------------------------------------
function UpdateQueryBuilder(connectionOrQueryBuilder, queryRunner) {
var _this = _super.call(this, connectionOrQueryBuilder, queryRunner) || this;
_this.expressionMap.aliasNamePrefixingEnabled = false;
return _this;
}
// -------------------------------------------------------------------------
// Public Implemented Methods
// -------------------------------------------------------------------------
/**
* Gets generated sql query without parameters being replaced.
*/
UpdateQueryBuilder.prototype.getQuery = function () {
var sql = this.createUpdateExpression();
sql += this.createOrderByExpression();
sql += this.createLimitExpression();
return sql.trim();
};
/**
* Executes sql generated by query builder and returns raw database results.
*/
UpdateQueryBuilder.prototype.execute = function () {
return __awaiter(this, void 0, void 0, function () {
var queryRunner, transactionStartedByUs, broadcastResult, returningResultsEntityUpdator, _a, sql, parameters, updateResult, _b, broadcastResult, error_1, rollbackError_1;
return __generator(this, function (_c) {
switch (_c.label) {
case 0:
queryRunner = this.obtainQueryRunner();
transactionStartedByUs = false;
_c.label = 1;
case 1:
_c.trys.push([1, 13, 18, 23]);
if (!(this.expressionMap.useTransaction === true && queryRunner.isTransactionActive === false)) return [3 /*break*/, 3];
return [4 /*yield*/, queryRunner.startTransaction()];
case 2:
_c.sent();
transactionStartedByUs = true;
_c.label = 3;
case 3:
if (!(this.expressionMap.callListeners === true && this.expressionMap.mainAlias.hasMetadata)) return [3 /*break*/, 5];
broadcastResult = new BroadcasterResult();
queryRunner.broadcaster.broadcastBeforeUpdateEvent(broadcastResult, this.expressionMap.mainAlias.metadata);
if (!(broadcastResult.promises.length > 0)) return [3 /*break*/, 5];
return [4 /*yield*/, Promise.all(broadcastResult.promises)];
case 4:
_c.sent();
_c.label = 5;
case 5:
returningResultsEntityUpdator = new ReturningResultsEntityUpdator(queryRunner, this.expressionMap);
if (this.expressionMap.updateEntity === true &&
this.expressionMap.mainAlias.hasMetadata &&
this.expressionMap.whereEntities.length > 0) {
this.expressionMap.extraReturningColumns = returningResultsEntityUpdator.getUpdationReturningColumns();
}
_a = this.getQueryAndParameters(), sql = _a[0], parameters = _a[1];
updateResult = new UpdateResult();
_b = updateResult;
return [4 /*yield*/, queryRunner.query(sql, parameters)];
case 6:
_b.raw = _c.sent();
if (!(this.expressionMap.updateEntity === true &&
this.expressionMap.mainAlias.hasMetadata &&
this.expressionMap.whereEntities.length > 0)) return [3 /*break*/, 8];
return [4 /*yield*/, returningResultsEntityUpdator.update(updateResult, this.expressionMap.whereEntities)];
case 7:
_c.sent();
_c.label = 8;
case 8:
if (!(this.expressionMap.callListeners === true && this.expressionMap.mainAlias.hasMetadata)) return [3 /*break*/, 10];
broadcastResult = new BroadcasterResult();
queryRunner.broadcaster.broadcastAfterUpdateEvent(broadcastResult, this.expressionMap.mainAlias.metadata);
if (!(broadcastResult.promises.length > 0)) return [3 /*break*/, 10];
return [4 /*yield*/, Promise.all(broadcastResult.promises)];
case 9:
_c.sent();
_c.label = 10;
case 10:
if (!transactionStartedByUs) return [3 /*break*/, 12];
return [4 /*yield*/, queryRunner.commitTransaction()];
case 11:
_c.sent();
_c.label = 12;
case 12: return [2 /*return*/, updateResult];
case 13:
error_1 = _c.sent();
if (!transactionStartedByUs) return [3 /*break*/, 17];
_c.label = 14;
case 14:
_c.trys.push([14, 16, , 17]);
return [4 /*yield*/, queryRunner.rollbackTransaction()];
case 15:
_c.sent();
return [3 /*break*/, 17];
case 16:
rollbackError_1 = _c.sent();
return [3 /*break*/, 17];
case 17: throw error_1;
case 18:
if (!(queryRunner !== this.queryRunner)) return [3 /*break*/, 20];
return [4 /*yield*/, queryRunner.release()];
case 19:
_c.sent();
_c.label = 20;
case 20:
if (!(this.connection.driver instanceof SqljsDriver && !queryRunner.isTransactionActive)) return [3 /*break*/, 22];
return [4 /*yield*/, this.connection.driver.autoSave()];
case 21:
_c.sent();
_c.label = 22;
case 22: return [7 /*endfinally*/];
case 23: return [2 /*return*/];
}
});
});
};
// -------------------------------------------------------------------------
// Public Methods
// -------------------------------------------------------------------------
/**
* Values needs to be updated.
*/
UpdateQueryBuilder.prototype.set = function (values) {
this.expressionMap.valuesSet = values;
return this;
};
/**
* Sets WHERE condition in the query builder.
* If you had previously WHERE expression defined,
* calling this function will override previously set WHERE conditions.
* Additionally you can add parameters used in where expression.
*/
UpdateQueryBuilder.prototype.where = function (where, parameters) {
this.expressionMap.wheres = []; // don't move this block below since computeWhereParameter can add where expressions
var condition = this.computeWhereParameter(where);
if (condition)
this.expressionMap.wheres = [{ type: "simple", condition: condition }];
if (parameters)
this.setParameters(parameters);
return this;
};
/**
* Adds new AND WHERE condition in the query builder.
* Additionally you can add parameters used in where expression.
*/
UpdateQueryBuilder.prototype.andWhere = function (where, parameters) {
this.expressionMap.wheres.push({ type: "and", condition: this.computeWhereParameter(where) });
if (parameters)
this.setParameters(parameters);
return this;
};
/**
* Adds new OR WHERE condition in the query builder.
* Additionally you can add parameters used in where expression.
*/
UpdateQueryBuilder.prototype.orWhere = function (where, parameters) {
this.expressionMap.wheres.push({ type: "or", condition: this.computeWhereParameter(where) });
if (parameters)
this.setParameters(parameters);
return this;
};
/**
* Adds new AND WHERE with conditions for the given ids.
*/
UpdateQueryBuilder.prototype.whereInIds = function (ids) {
return this.where(this.createWhereIdsExpression(ids));
};
/**
* Adds new AND WHERE with conditions for the given ids.
*/
UpdateQueryBuilder.prototype.andWhereInIds = function (ids) {
return this.andWhere(this.createWhereIdsExpression(ids));
};
/**
* Adds new OR WHERE with conditions for the given ids.
*/
UpdateQueryBuilder.prototype.orWhereInIds = function (ids) {
return this.orWhere(this.createWhereIdsExpression(ids));
};
/**
* Optional returning/output clause.
*/
UpdateQueryBuilder.prototype.output = function (output) {
return this.returning(output);
};
/**
* Optional returning/output clause.
*/
UpdateQueryBuilder.prototype.returning = function (returning) {
// not all databases support returning/output cause
if (!this.connection.driver.isReturningSqlSupported())
throw new ReturningStatementNotSupportedError();
this.expressionMap.returning = returning;
return this;
};
/**
* Sets ORDER BY condition in the query builder.
* If you had previously ORDER BY expression defined,
* calling this function will override previously set ORDER BY conditions.
*/
UpdateQueryBuilder.prototype.orderBy = function (sort, order, nulls) {
if (order === void 0) { order = "ASC"; }
if (sort) {
if (sort instanceof Object) {
this.expressionMap.orderBys = sort;
}
else {
if (nulls) {
this.expressionMap.orderBys = (_a = {}, _a[sort] = { order: order, nulls: nulls }, _a);
}
else {
this.expressionMap.orderBys = (_b = {}, _b[sort] = order, _b);
}
}
}
else {
this.expressionMap.orderBys = {};
}
return this;
var _a, _b;
};
/**
* Adds ORDER BY condition in the query builder.
*/
UpdateQueryBuilder.prototype.addOrderBy = function (sort, order, nulls) {
if (order === void 0) { order = "ASC"; }
if (nulls) {
this.expressionMap.orderBys[sort] = { order: order, nulls: nulls };
}
else {
this.expressionMap.orderBys[sort] = order;
}
return this;
};
/**
* Sets LIMIT - maximum number of rows to be selected.
*/
UpdateQueryBuilder.prototype.limit = function (limit) {
this.expressionMap.limit = limit;
return this;
};
/**
* Indicates if entity must be updated after update operation.
* This may produce extra query or use RETURNING / OUTPUT statement (depend on database).
* Enabled by default.
*/
UpdateQueryBuilder.prototype.whereEntity = function (entity) {
var _this = this;
if (!this.expressionMap.mainAlias.hasMetadata)
throw new Error(".whereEntity method can only be used on queries which update real entity table.");
this.expressionMap.wheres = [];
var entities = entity instanceof Array ? entity : [entity];
entities.forEach(function (entity) {
var entityIdMap = _this.expressionMap.mainAlias.metadata.getEntityIdMap(entity);
if (!entityIdMap)
throw new Error("Provided entity does not have ids set, cannot perform operation.");
_this.orWhereInIds(entityIdMap);
});
this.expressionMap.whereEntities = entities;
return this;
};
/**
* Indicates if entity must be updated after update operation.
* This may produce extra query or use RETURNING / OUTPUT statement (depend on database).
* Enabled by default.
*/
UpdateQueryBuilder.prototype.updateEntity = function (enabled) {
this.expressionMap.updateEntity = enabled;
return this;
};
// -------------------------------------------------------------------------
// Protected Methods
// -------------------------------------------------------------------------
/**
* Creates UPDATE express used to perform insert query.
*/
UpdateQueryBuilder.prototype.createUpdateExpression = function () {
var _this = this;
var valuesSet = this.getValueSet();
var metadata = this.expressionMap.mainAlias.hasMetadata ? this.expressionMap.mainAlias.metadata : undefined;
// prepare columns and values to be updated
var updateColumnAndValues = [];
var newParameters = {};
var parametersCount = this.connection.driver instanceof MysqlDriver ||
this.connection.driver instanceof OracleDriver ||
this.connection.driver instanceof AbstractSqliteDriver
? 0 : Object.keys(this.expressionMap.nativeParameters).length;
if (metadata) {
EntityMetadata.createPropertyPath(metadata, valuesSet).forEach(function (propertyPath) {
// todo: make this and other query builder to work with properly with tables without metadata
var columns = metadata.findColumnsWithPropertyPath(propertyPath);
columns.forEach(function (column) {
var paramName = "upd_" + column.databaseName;
//
var value = column.getEntityValue(valuesSet);
if (column.referencedColumn && value instanceof Object) {
value = column.referencedColumn.getEntityValue(value);
}
value = _this.connection.driver.preparePersistentValue(value, column);
// todo: duplication zone
if (value instanceof Function) { // support for SQL expressions in update query
updateColumnAndValues.push(_this.escape(column.databaseName) + " = " + value());
}
else {
if (_this.connection.driver instanceof SqlServerDriver) {
value = _this.connection.driver.parametrizeValue(column, value);
// } else if (value instanceof Array) {
// value = new ArrayParameter(value);
}
if (_this.connection.driver instanceof MysqlDriver ||
_this.connection.driver instanceof OracleDriver ||
_this.connection.driver instanceof AbstractSqliteDriver) {
newParameters[paramName] = value;
}
else {
_this.expressionMap.nativeParameters[paramName] = value;
}
updateColumnAndValues.push(_this.escape(column.databaseName) + " = " + _this.connection.driver.createParameter(paramName, parametersCount));
parametersCount++;
}
});
});
if (metadata.versionColumn)
updateColumnAndValues.push(this.escape(metadata.versionColumn.databaseName) + " = " + this.escape(metadata.versionColumn.databaseName) + " + 1");
if (metadata.updateDateColumn)
updateColumnAndValues.push(this.escape(metadata.updateDateColumn.databaseName) + " = CURRENT_TIMESTAMP"); // todo: fix issue with CURRENT_TIMESTAMP(6) being used, can "DEFAULT" be used?!
}
else {
Object.keys(valuesSet).map(function (key) {
var value = valuesSet[key];
// todo: duplication zone
if (value instanceof Function) { // support for SQL expressions in update query
updateColumnAndValues.push(_this.escape(key) + " = " + value());
}
else {
// we need to store array values in a special class to make sure parameter replacement will work correctly
// if (value instanceof Array)
// value = new ArrayParameter(value);
if (_this.connection.driver instanceof MysqlDriver ||
_this.connection.driver instanceof OracleDriver ||
_this.connection.driver instanceof AbstractSqliteDriver) {
newParameters[key] = value;
}
else {
_this.expressionMap.nativeParameters[key] = value;
}
updateColumnAndValues.push(_this.escape(key) + " = " + _this.connection.driver.createParameter(key, parametersCount));
parametersCount++;
}
});
}
// we re-write parameters this way because we want our "UPDATE ... SET" parameters to be first in the list of "nativeParameters"
// because some drivers like mysql depend on order of parameters
if (this.connection.driver instanceof MysqlDriver ||
this.connection.driver instanceof OracleDriver ||
this.connection.driver instanceof AbstractSqliteDriver) {
this.expressionMap.nativeParameters = Object.assign(newParameters, this.expressionMap.nativeParameters);
}
// get a table name and all column database names
var whereExpression = this.createWhereExpression();
var returningExpression = this.createReturningExpression();
// generate and return sql update query
if (returningExpression && (this.connection.driver instanceof PostgresDriver || this.connection.driver instanceof OracleDriver)) {
return "UPDATE " + this.getTableName(this.getMainTableName()) + " SET " + updateColumnAndValues.join(", ") + whereExpression + " RETURNING " + returningExpression;
}
else if (returningExpression && this.connection.driver instanceof SqlServerDriver) {
return "UPDATE " + this.getTableName(this.getMainTableName()) + " SET " + updateColumnAndValues.join(", ") + " OUTPUT " + returningExpression + whereExpression;
}
else {
return "UPDATE " + this.getTableName(this.getMainTableName()) + " SET " + updateColumnAndValues.join(", ") + whereExpression; // todo: how do we replace aliases in where to nothing?
}
};
/**
* Creates "ORDER BY" part of SQL query.
*/
UpdateQueryBuilder.prototype.createOrderByExpression = function () {
var _this = this;
var orderBys = this.expressionMap.orderBys;
if (Object.keys(orderBys).length > 0)
return " ORDER BY " + Object.keys(orderBys)
.map(function (columnName) {
if (typeof orderBys[columnName] === "string") {
return _this.replacePropertyNames(columnName) + " " + orderBys[columnName];
}
else {
return _this.replacePropertyNames(columnName) + " " + orderBys[columnName].order + " " + orderBys[columnName].nulls;
}
})
.join(", ");
return "";
};
/**
* Creates "LIMIT" parts of SQL query.
*/
UpdateQueryBuilder.prototype.createLimitExpression = function () {
var limit = this.expressionMap.limit;
if (limit) {
if (this.connection.driver instanceof MysqlDriver) {
return " LIMIT " + limit;
}
else {
throw new LimitOnUpdateNotSupportedError();
}
}
return "";
};
/**
* Gets array of values need to be inserted into the target table.
*/
UpdateQueryBuilder.prototype.getValueSet = function () {
if (this.expressionMap.valuesSet instanceof Object)
return this.expressionMap.valuesSet;
throw new Error("Cannot perform update query because update values are not defined. Call \"qb.set(...)\" method to specify inserted values.");
};
return UpdateQueryBuilder;
}(QueryBuilder));
export { UpdateQueryBuilder };
//# sourceMappingURL=UpdateQueryBuilder.js.map