UNPKG

typeorm

Version:

Data-Mapper ORM for TypeScript and ES2023+. Supports MySQL/MariaDB, PostgreSQL, MS SQL Server, Oracle, SAP HANA, SQLite, MongoDB databases.

880 lines • 32.9 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.AuroraMysqlDriver = void 0; const error_1 = require("../../error"); const PlatformTools_1 = require("../../platform/PlatformTools"); const RdbmsSchemaBuilder_1 = require("../../schema-builder/RdbmsSchemaBuilder"); const ApplyValueTransformers_1 = require("../../util/ApplyValueTransformers"); const DateUtils_1 = require("../../util/DateUtils"); const InstanceChecker_1 = require("../../util/InstanceChecker"); const OrmUtils_1 = require("../../util/OrmUtils"); const DriverUtils_1 = require("../DriverUtils"); const AuroraMysqlQueryRunner_1 = require("./AuroraMysqlQueryRunner"); /** * Organizes communication with MySQL DBMS. */ class AuroraMysqlDriver { // ------------------------------------------------------------------------- // Static Properties // ------------------------------------------------------------------------- /** * Aurora MySQL cannot honor per-transaction isolation levels over the RDS * Data API: BeginTransaction accepts no isolation parameter, the API is * stateless so a preceding `SET TRANSACTION` has no guaranteed connection * affinity to the subsequent BeginTransaction, multi-statement SQL is * rejected, and MySQL disallows `SET TRANSACTION` inside an active * transaction (error 1568). See the transactions documentation for links. */ static { this.supportedIsolationLevels = []; } /** * DataSource used by the driver. * * @deprecated since 1.0.0. Use {@link dataSource} instance instead. */ get connection() { return this.dataSource; } // ------------------------------------------------------------------------- // Constructor // ------------------------------------------------------------------------- constructor(dataSource) { /** * Isolation levels supported by this driver. */ this.supportedIsolationLevels = AuroraMysqlDriver.supportedIsolationLevels; /** * Indicates if replication is enabled. */ this.isReplicated = false; /** * Indicates if tree tables are supported by this driver. */ this.treeSupport = true; /** * Represent transaction support by this driver */ this.transactionSupport = "nested"; /** * Gets list of supported column data types by a driver. * * @see https://www.tutorialspoint.com/mysql/mysql-data-types.htm * @see https://dev.mysql.com/doc/refman/8.0/en/data-types.html */ this.supportedDataTypes = [ // numeric types "bit", "int", "integer", // synonym for int "tinyint", "smallint", "mediumint", "bigint", "float", "double", "double precision", // synonym for double "real", // synonym for double "decimal", "dec", // synonym for decimal "numeric", // synonym for decimal "fixed", // synonym for decimal "bool", // synonym for tinyint "boolean", // synonym for tinyint // date and time types "date", "datetime", "timestamp", "time", "year", // string types "char", "nchar", // synonym for national char "national char", "varchar", "nvarchar", // synonym for national varchar "national varchar", "blob", "text", "tinyblob", "tinytext", "mediumblob", "mediumtext", "longblob", "longtext", "enum", "set", "binary", "varbinary", // json data type "json", // spatial data types "geometry", "point", "linestring", "polygon", "multipoint", "multilinestring", "multipolygon", "geometrycollection", ]; /** * Returns type of upsert supported by driver if any */ this.supportedUpsertTypes = ["on-duplicate-key-update"]; /** * Gets list of spatial column data types. */ this.spatialTypes = [ "geometry", "point", "linestring", "polygon", "multipoint", "multilinestring", "multipolygon", "geometrycollection", ]; /** * Gets list of column data types that support length by a driver. */ this.withLengthColumnTypes = [ "char", "varchar", "nvarchar", "binary", "varbinary", ]; /** * Gets list of column data types that support precision by a driver. */ this.withPrecisionColumnTypes = [ "decimal", "dec", "numeric", "fixed", "float", "double", "double precision", "real", "time", "datetime", "timestamp", ]; /** * Gets list of column data types that supports scale by a driver. */ this.withScaleColumnTypes = [ "decimal", "dec", "numeric", "fixed", "float", "double", "double precision", "real", ]; /** * ORM has special columns and we need to know what database column types should be for those columns. * Column types are driver dependant. */ this.mappedDataTypes = { createDate: "datetime", createDatePrecision: 6, createDateDefault: "CURRENT_TIMESTAMP(6)", updateDate: "datetime", updateDatePrecision: 6, updateDateDefault: "CURRENT_TIMESTAMP(6)", deleteDate: "datetime", deleteDatePrecision: 6, deleteDateNullable: true, version: "int", treeLevel: "int", migrationId: "int", migrationName: "varchar", migrationTimestamp: "bigint", cacheId: "int", cacheIdentifier: "varchar", cacheTime: "bigint", cacheDuration: "int", cacheQuery: "text", cacheResult: "text", metadataType: "varchar", metadataDatabase: "varchar", metadataSchema: "varchar", metadataTable: "varchar", metadataName: "varchar", metadataValue: "text", }; /** * Default values of length, precision and scale depends on column data type. * Used in the cases when length/precision/scale is not specified by user. */ this.dataTypeDefaults = { varchar: { length: 255 }, nvarchar: { length: 255 }, "national varchar": { length: 255 }, char: { length: 1 }, binary: { length: 1 }, varbinary: { length: 255 }, decimal: { precision: 10, scale: 0 }, dec: { precision: 10, scale: 0 }, numeric: { precision: 10, scale: 0 }, fixed: { precision: 10, scale: 0 }, float: { precision: 12 }, double: { precision: 22 }, time: { precision: 0 }, datetime: { precision: 0 }, timestamp: { precision: 0 }, }; /** * Max length allowed by MySQL for aliases. * * @see https://dev.mysql.com/doc/refman/5.5/en/identifiers.html */ this.maxAliasLength = 63; this.cteCapabilities = { enabled: false, }; this.dataSource = dataSource; this.options = dataSource.options; // load mysql package this.loadDependencies(); this.client = new this.DataApiDriver(this.options.region, this.options.secretArn, this.options.resourceArn, this.options.database, (query, parameters) => this.dataSource.logger.logQuery(query, parameters), this.options.serviceConfigOptions, this.options.formatOptions); this.database = DriverUtils_1.DriverUtils.buildDriverOptions(this.options).database; // validate options to make sure everything is set // todo: revisit validation with replication in mind // if (!(this.options.host || (this.options.extra && this.options.extra.socketPath)) && !this.options.socketPath) // throw new DriverOptionNotSetError("socketPath and host"); // if (!this.options.username) // throw new DriverOptionNotSetError("username"); // if (!this.options.database) // throw new DriverOptionNotSetError("database"); // todo: check what is going on when connection is setup without database and how to connect to a database then? // todo: provide options to auto-create a database if it does not exist yet } // ------------------------------------------------------------------------- // Public Methods // ------------------------------------------------------------------------- /** * Performs connection to the database. */ async connect() { if (!this.database) { const queryRunner = this.createQueryRunner("master"); this.database = await queryRunner.getCurrentDatabase(); await queryRunner.release(); } } /** * Makes any action after connection (e.g. create extensions in Postgres driver). */ afterConnect() { return Promise.resolve(); } /** * Closes connection with the database. */ async disconnect() { } /** * Creates a schema builder used to build and sync a schema. */ createSchemaBuilder() { return new RdbmsSchemaBuilder_1.RdbmsSchemaBuilder(this.dataSource); } /** * Creates a query runner used to execute database queries. * * @param mode */ createQueryRunner(mode) { return new AuroraMysqlQueryRunner_1.AuroraMysqlQueryRunner(this, new this.DataApiDriver(this.options.region, this.options.secretArn, this.options.resourceArn, this.options.database, (query, parameters) => this.dataSource.logger.logQuery(query, parameters), this.options.serviceConfigOptions, this.options.formatOptions)); } /** * Replaces parameters in the given sql with special escaping character * and an array of parameter names to be passed to a query. * * @param sql * @param parameters */ escapeQueryWithParameters(sql, parameters) { const escapedParameters = []; if (!parameters || !Object.keys(parameters).length) return [sql, escapedParameters]; sql = sql.replaceAll(/:(\.\.\.)?([A-Za-z0-9_.]+)/g, (full, isArray, key) => { if (!parameters.hasOwnProperty(key)) { return full; } const value = parameters[key]; if (isArray) { return value .map((v) => { escapedParameters.push(v); return this.createParameter(key, escapedParameters.length - 1); }) .join(", "); } if (typeof value === "function") { return value(); } escapedParameters.push(value); return this.createParameter(key, escapedParameters.length - 1); }); // todo: make replace only in value statements, otherwise problems return [sql, escapedParameters]; } /** * Escapes a column name. * * @param columnName */ escape(columnName) { return "`" + columnName.replaceAll("`", "``") + "`"; } /** * Build full table name with database name, schema name and table name. * E.g. myDB.mySchema.myTable * * @param tableName * @param schema * @param database */ buildTableName(tableName, schema, database) { const tablePath = [tableName]; if (database) { tablePath.unshift(database); } return tablePath.join("."); } /** * Parse a target table name or other types and return a normalized table definition. * * @param target */ parseTableName(target) { const driverDatabase = this.database; const driverSchema = undefined; if (InstanceChecker_1.InstanceChecker.isTable(target) || InstanceChecker_1.InstanceChecker.isView(target)) { const parsed = this.parseTableName(target.name); return { database: target.database ?? parsed.database ?? driverDatabase, schema: target.schema ?? parsed.schema ?? driverSchema, tableName: parsed.tableName, }; } if (InstanceChecker_1.InstanceChecker.isTableForeignKey(target)) { const parsed = this.parseTableName(target.referencedTableName); return { database: target.referencedDatabase ?? parsed.database ?? driverDatabase, schema: target.referencedSchema ?? parsed.schema ?? driverSchema, tableName: parsed.tableName, }; } if (InstanceChecker_1.InstanceChecker.isEntityMetadata(target)) { // EntityMetadata tableName is never a path return { database: target.database ?? driverDatabase, schema: target.schema ?? driverSchema, tableName: target.tableName, }; } const parts = target.split("."); return { database: (parts.length > 1 ? parts[0] : undefined) ?? driverDatabase, schema: driverSchema, tableName: parts.length > 1 ? parts[1] : parts[0], }; } /** * Prepares given value to a value to be persisted, based on its column type and metadata. * * @param value * @param columnMetadata */ preparePersistentValue(value, columnMetadata) { if (columnMetadata.transformer) value = ApplyValueTransformers_1.ApplyValueTransformers.transformTo(columnMetadata.transformer, value); if (this.options.formatOptions?.castParameters !== false) { return this.client.preparePersistentValue(value, columnMetadata); } if (value === null || value === undefined) return value; if (columnMetadata.type === Boolean) { return value === true ? 1 : 0; } else if (columnMetadata.type === "date") { return DateUtils_1.DateUtils.mixedDateToDateString(value, { utc: columnMetadata.utc, }); } else if (columnMetadata.type === "time") { return DateUtils_1.DateUtils.mixedDateToTimeString(value); } else if (columnMetadata.type === "json") { return JSON.stringify(value); } else if (columnMetadata.type === "timestamp" || columnMetadata.type === "datetime" || columnMetadata.type === Date) { return DateUtils_1.DateUtils.mixedDateToDate(value); } else if (columnMetadata.type === "simple-array" || columnMetadata.type === "set") { return DateUtils_1.DateUtils.simpleArrayToString(value); } else if (columnMetadata.type === "simple-json") { return DateUtils_1.DateUtils.simpleJsonToString(value); } else if (columnMetadata.type === "enum" || columnMetadata.type === "simple-enum") { return "" + value; } return value; } /** * Prepares given value to a value to be persisted, based on its column type or metadata. * * @param value * @param columnMetadata */ prepareHydratedValue(value, columnMetadata) { if (value === null || value === undefined) return columnMetadata.transformer ? ApplyValueTransformers_1.ApplyValueTransformers.transformFrom(columnMetadata.transformer, value) : value; if (this.options.formatOptions?.castParameters !== false) { return this.client.prepareHydratedValue(value, columnMetadata); } if (columnMetadata.type === Boolean || columnMetadata.type === "bool" || columnMetadata.type === "boolean") { value = value ? true : false; } else if (columnMetadata.type === "datetime" || columnMetadata.type === Date) { value = DateUtils_1.DateUtils.normalizeHydratedDate(value); } else if (columnMetadata.type === "date") { value = DateUtils_1.DateUtils.mixedDateToDateString(value, { utc: columnMetadata.utc, }); } else if (columnMetadata.type === "json") { value = typeof value === "string" ? JSON.parse(value) : value; } else if (columnMetadata.type === "time") { value = DateUtils_1.DateUtils.mixedTimeToString(value); } else if (columnMetadata.type === "simple-array" || columnMetadata.type === "set") { value = DateUtils_1.DateUtils.stringToSimpleArray(value); } else if (columnMetadata.type === "simple-json") { value = DateUtils_1.DateUtils.stringToSimpleJson(value); } else if ((columnMetadata.type === "enum" || columnMetadata.type === "simple-enum") && columnMetadata.enum && !isNaN(value) && columnMetadata.enum.indexOf(parseInt(value)) >= 0) { // convert to number if that exists in possible enum options value = parseInt(value); } else if (columnMetadata.type === Number) { // convert to number if number value = !isNaN(+value) ? parseInt(value) : value; } if (columnMetadata.transformer) value = ApplyValueTransformers_1.ApplyValueTransformers.transformFrom(columnMetadata.transformer, value); return value; } /** * Creates a database type from a given column metadata. * * @param column * @param column.type * @param column.length * @param column.precision * @param column.scale */ normalizeType(column) { if (column.type === Number || column.type === "integer") { return "int"; } else if (column.type === String) { return "varchar"; } else if (column.type === Date) { return "datetime"; } else if ( // (column.type as any) === Uint8Array || // (typeof Buffer !== "undefined" && (column.type as any) === Buffer) typeof column.type === "function" && column.type.prototype instanceof Uint8Array) { return "blob"; } else if (column.type === Boolean) { return "tinyint"; } else if (column.type === "uuid") { return "varchar"; } else if (column.type === "simple-array" || column.type === "simple-json") { return "text"; } else if (column.type === "simple-enum") { return "enum"; } else if (column.type === "double precision" || column.type === "real") { return "double"; } else if (column.type === "dec" || column.type === "numeric" || column.type === "fixed") { return "decimal"; } else if (column.type === "bool" || column.type === "boolean") { return "tinyint"; } else if (column.type === "nvarchar" || column.type === "national varchar") { return "varchar"; } else if (column.type === "nchar" || column.type === "national char") { return "char"; } else { return column.type || ""; } } /** * Normalizes "default" value of the column. * * @param columnMetadata */ normalizeDefault(columnMetadata) { const defaultValue = columnMetadata.default; if (defaultValue === null) { return undefined; } if ((columnMetadata.type === "enum" || columnMetadata.type === "simple-enum") && defaultValue !== undefined) { return `'${defaultValue}'`; } if (columnMetadata.type === "set" && defaultValue !== undefined) { return `'${DateUtils_1.DateUtils.simpleArrayToString(defaultValue)}'`; } if (typeof defaultValue === "number") { return `${defaultValue}`; } if (typeof defaultValue === "boolean") { return defaultValue ? "1" : "0"; } if (typeof defaultValue === "function") { return defaultValue(); } if (typeof defaultValue === "string") { return `'${defaultValue}'`; } if (defaultValue === undefined) { return undefined; } return `${defaultValue}`; } /** * Normalizes "isUnique" value of the column. * * @param column */ normalizeIsUnique(column) { return column.entityMetadata.indices.some((idx) => idx.isUnique && idx.columns.length === 1 && idx.columns[0] === column); } /** * Returns default column lengths, which is required on column creation. * * @param column */ getColumnLength(column) { if (column.length) return column.length.toString(); /** * fix https://github.com/typeorm/typeorm/issues/1139 */ if (column.generationStrategy === "uuid") return "36"; switch (column.type) { case String: case "varchar": case "nvarchar": case "national varchar": return "255"; case "varbinary": return "255"; default: return ""; } } /** * Creates column type definition including length, precision and scale * * @param column */ createFullType(column) { let type = column.type; // used 'getColumnLength()' method, because MySQL requires column length for `varchar`, `nvarchar` and `varbinary` data types if (this.getColumnLength(column)) { type += `(${this.getColumnLength(column)})`; } else if (column.precision !== null && column.precision !== undefined && column.scale !== null && column.scale !== undefined) { type += `(${column.precision},${column.scale})`; } else if (column.precision !== null && column.precision !== undefined) { type += `(${column.precision})`; } if (column.isArray) type += " array"; return type; } /** * Obtains a new database connection to a master server. * Used for replication. * If replication is not setup then returns default connection's database connection. */ obtainMasterConnection() { return new Promise((ok, fail) => { if (this.poolCluster) { this.poolCluster.getConnection("MASTER", (err, dbConnection) => { if (err) { fail(err); } else { ok(this.prepareDbConnection(dbConnection)); } }); } else if (this.pool) { this.pool.getConnection((err, dbConnection) => { if (err) { fail(err); } else { ok(this.prepareDbConnection(dbConnection)); } }); } else { fail(new error_1.TypeORMError(`Connection is not established with mysql database`)); } }); } /** * Obtains a new database connection to a slave server. * Used for replication. * If replication is not setup then returns master (default) connection's database connection. */ obtainSlaveConnection() { if (!this.poolCluster) return this.obtainMasterConnection(); return new Promise((ok, fail) => { this.poolCluster.getConnection("SLAVE*", (err, dbConnection) => { if (err) { fail(err); } else { ok(this.prepareDbConnection(dbConnection)); } }); }); } /** * Creates generated map of values generated or returned by database after INSERT query. * * @param metadata * @param insertResult * @param entityIndex */ createGeneratedMap(metadata, insertResult, entityIndex) { const generatedMap = metadata.generatedColumns.reduce((map, generatedColumn) => { let value; if (generatedColumn.generationStrategy === "increment" && insertResult.insertId) { // NOTE: When multiple rows is inserted by a single INSERT statement, // `insertId` is the value generated for the first inserted row only. value = insertResult.insertId + entityIndex; // } else if (generatedColumn.generationStrategy === "uuid") { // console.log("getting db value:", generatedColumn.databaseName); // value = generatedColumn.getEntityValue(uuidMap); } return OrmUtils_1.OrmUtils.mergeDeep(map, generatedColumn.createValueMap(value)); }, {}); return Object.keys(generatedMap).length > 0 ? generatedMap : undefined; } /** * Differentiate columns of this table and columns from the given column metadatas columns * and returns only changed. * * @param tableColumns * @param columnMetadatas */ findChangedColumns(tableColumns, columnMetadatas) { return columnMetadatas.filter((columnMetadata) => { const tableColumn = tableColumns.find((c) => c.name === columnMetadata.databaseName); if (!tableColumn) return false; // we don't need new columns, we only need exist and changed let columnMetadataLength = columnMetadata.length; if (!columnMetadataLength && columnMetadata.generationStrategy === "uuid") { // fixing #3374 columnMetadataLength = this.getColumnLength(columnMetadata); } return (tableColumn.name !== columnMetadata.databaseName || tableColumn.type !== this.normalizeType(columnMetadata) || tableColumn.length !== columnMetadataLength || tableColumn.precision !== columnMetadata.precision || tableColumn.scale !== columnMetadata.scale || tableColumn.unsigned !== columnMetadata.unsigned || tableColumn.asExpression !== columnMetadata.asExpression || tableColumn.generatedType !== columnMetadata.generatedType || tableColumn.comment !== this.escapeComment(columnMetadata.comment) || !this.compareDefaultValues(this.normalizeDefault(columnMetadata), tableColumn.default) || !!(tableColumn.enum && columnMetadata.enum && !OrmUtils_1.OrmUtils.isArraysEqual(tableColumn.enum, columnMetadata.enum.map((val) => val + ""))) || tableColumn.onUpdate !== columnMetadata.onUpdate || tableColumn.isPrimary !== columnMetadata.isPrimary || tableColumn.isNullable !== columnMetadata.isNullable || tableColumn.isUnique !== this.normalizeIsUnique(columnMetadata) || (columnMetadata.generationStrategy !== "uuid" && tableColumn.isGenerated !== columnMetadata.isGenerated)); }); } /** * Returns true if driver supports RETURNING / OUTPUT statement. */ isReturningSqlSupported() { return false; } /** * Returns true if driver supports uuid values generation on its own. */ isUUIDGenerationSupported() { return false; } /** * Returns true if driver supports fulltext indices. */ isFullTextColumnTypeSupported() { return true; } /** * Creates an escaped parameter. * * @param parameterName * @param index */ createParameter(parameterName, index) { return "?"; } // ------------------------------------------------------------------------- // Protected Methods // ------------------------------------------------------------------------- /** * Loads all driver dependencies. */ loadDependencies() { const DataApiDriver = this.options.driver ?? PlatformTools_1.PlatformTools.load("typeorm-aurora-data-api-driver"); this.DataApiDriver = DataApiDriver; // Driver uses rollup for publishing, which has issues when using typeorm in combination with webpack // See https://github.com/webpack/webpack/issues/4742#issuecomment-295556787 this.DataApiDriver = this.DataApiDriver.default ?? this.DataApiDriver; } /** * Creates a new connection pool for a given database credentials. * * @param options * @param credentials */ createConnectionOptions(options, credentials) { credentials = Object.assign({}, credentials, DriverUtils_1.DriverUtils.buildDriverOptions(credentials)); // todo: do it better way // build connection options for the driver return Object.assign({}, { resourceArn: options.resourceArn, secretArn: options.secretArn, database: options.database, region: options.region, type: options.type, }, { host: credentials.host, user: credentials.username, password: credentials.password, database: credentials.database, port: credentials.port, ssl: options.ssl, }, options.extra ?? {}); } /** * Creates a new connection pool for a given database credentials. * * @param connectionOptions */ async createPool(connectionOptions) { return {}; } /** * Attaches all required base handlers to a database connection, such as the unhandled error handler. * * @param connection */ prepareDbConnection(connection) { const { logger } = this.dataSource; /** * Attaching an error handler to connection errors is essential, as, otherwise, errors raised will go unhandled and * cause the hosting app to crash. */ if (connection.listeners("error").length === 0) { connection.on("error", (error) => logger.log("warn", `MySQL connection raised an error. ${error}`)); } return connection; } /** * Checks if "DEFAULT" values in the column metadata and in the database are equal. * * @param columnMetadataValue * @param databaseValue */ compareDefaultValues(columnMetadataValue, databaseValue) { if (typeof columnMetadataValue === "string" && typeof databaseValue === "string") { // we need to cut out "'" because in mysql we can understand returned value is a string or a function // as result compare cannot understand if default is really changed or not columnMetadataValue = columnMetadataValue.replaceAll(/^'+|'+$/g, ""); databaseValue = databaseValue.replaceAll(/^'+|'+$/g, ""); } return columnMetadataValue === databaseValue; } /** * Escapes a given comment. * * @param comment */ escapeComment(comment) { if (!comment) return comment; comment = comment.replaceAll("\u0000", ""); // Null bytes aren't allowed in comments return comment; } } exports.AuroraMysqlDriver = AuroraMysqlDriver; //# sourceMappingURL=AuroraMysqlDriver.js.map