@lakutata/orm 3.0.0-beta.6 → 3.1.0

package/dist/vendor/typeorm/driver/cockroachdb/CockroachDriver.js

@@ -0,0 +1,1006 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CockroachDriver = void 0;
+const error_1 = require("../../error");
+const ConnectionIsNotSetError_1 = require("../../error/ConnectionIsNotSetError");
+const DriverPackageNotInstalledError_1 = require("../../error/DriverPackageNotInstalledError");
+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 ObjectUtils_1 = require("../../util/ObjectUtils");
+const OrmUtils_1 = require("../../util/OrmUtils");
+const DriverUtils_1 = require("../DriverUtils");
+const CockroachQueryRunner_1 = require("./CockroachQueryRunner");
+/**
+ * Organizes communication with Cockroach DBMS.
+ */
+class CockroachDriver {
+    // -------------------------------------------------------------------------
+    // Static Properties
+    // -------------------------------------------------------------------------
+    /**
+     * Transaction isolation levels supported by this driver.
+     *
+     * @see https://www.cockroachlabs.com/docs/stable/transactions#isolation-levels
+     */
+    static { this.supportedIsolationLevels = [
+        "READ UNCOMMITTED",
+        "READ COMMITTED",
+        "REPEATABLE READ",
+        "SERIALIZABLE",
+    ]; }
+    /**
+     * 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 = CockroachDriver.supportedIsolationLevels;
+        /**
+         * Pool for slave databases.
+         * Used in replication.
+         */
+        this.slaves = [];
+        /**
+         * We store all created query runners because we need to release them.
+         */
+        this.connectedQueryRunners = [];
+        /**
+         * 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.cockroachlabs.com/docs/stable/data-types.html
+         */
+        this.supportedDataTypes = [
+            "array",
+            "bool",
+            "boolean",
+            "bytes",
+            "bytea",
+            "blob",
+            "date",
+            "enum",
+            "geometry",
+            "geography",
+            "numeric",
+            "decimal",
+            "dec",
+            "float",
+            "float4",
+            "float8",
+            "double precision",
+            "real",
+            "inet",
+            "int",
+            "int4",
+            "integer",
+            "int2",
+            "int8",
+            "int64",
+            "smallint",
+            "bigint",
+            "interval",
+            "string",
+            "character varying",
+            "character",
+            "char",
+            "char varying",
+            "varchar",
+            "text",
+            "time",
+            "time without time zone",
+            "timestamp",
+            "timestamptz",
+            "timestamp without time zone",
+            "timestamp with time zone",
+            "json",
+            "jsonb",
+            "uuid",
+        ];
+        /**
+         * Returns type of upsert supported by driver if any
+         */
+        this.supportedUpsertTypes = [
+            "on-conflict-do-update",
+            "primary-key",
+        ];
+        /**
+         * Gets list of spatial column data types.
+         */
+        this.spatialTypes = ["geometry", "geography"];
+        /**
+         * Gets list of column data types that support length by a driver.
+         */
+        this.withLengthColumnTypes = [
+            "character varying",
+            "char varying",
+            "varchar",
+            "character",
+            "char",
+            "string",
+        ];
+        /**
+         * Gets list of column data types that support precision by a driver.
+         */
+        this.withPrecisionColumnTypes = ["numeric", "decimal", "dec"];
+        /**
+         * Gets list of column data types that support scale by a driver.
+         */
+        this.withScaleColumnTypes = ["numeric", "decimal", "dec"];
+        /**
+         * Orm has special columns and we need to know what database column types should be for those types.
+         * Column types are driver dependent.
+         */
+        this.mappedDataTypes = {
+            createDate: "timestamptz",
+            createDateDefault: "now()",
+            updateDate: "timestamptz",
+            updateDateDefault: "now()",
+            deleteDate: "timestamptz",
+            deleteDateNullable: true,
+            version: Number,
+            treeLevel: Number,
+            migrationId: Number,
+            migrationName: "varchar",
+            migrationTimestamp: "int8",
+            cacheId: Number,
+            cacheIdentifier: "varchar",
+            cacheTime: "int8",
+            cacheDuration: Number,
+            cacheQuery: "string",
+            cacheResult: "string",
+            metadataType: "varchar",
+            metadataDatabase: "varchar",
+            metadataSchema: "varchar",
+            metadataTable: "varchar",
+            metadataName: "varchar",
+            metadataValue: "string",
+        };
+        /**
+         * The prefix used for the parameters
+         */
+        this.parametersPrefix = "$";
+        /**
+         * 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 = {
+            char: { length: 1 },
+        };
+        this.cteCapabilities = {
+            enabled: true,
+            writable: true,
+            materializedHint: true,
+            requiresRecursiveHint: true,
+        };
+        this.dataSource = dataSource;
+        this.options = dataSource.options;
+        this.isReplicated = this.options.replication ? true : false;
+        // load postgres package
+        this.loadDependencies();
+        this.database = DriverUtils_1.DriverUtils.buildDriverOptions(this.options.replication
+            ? this.options.replication.master
+            : this.options).database;
+        this.schema = DriverUtils_1.DriverUtils.buildDriverOptions(this.options).schema;
+        // ObjectUtils.assign(this.options, DriverUtils.buildDriverOptions(connection.options)); // todo: do it better way
+        // validate options to make sure everything is set
+        // todo: revisit validation with replication in mind
+        // if (!this.options.host)
+        //     throw new DriverOptionNotSetError("host");
+        // if (!this.options.username)
+        //     throw new DriverOptionNotSetError("username");
+        // if (!this.options.database)
+        //     throw new DriverOptionNotSetError("database");
+    }
+    // -------------------------------------------------------------------------
+    // Public Implemented Methods
+    // -------------------------------------------------------------------------
+    /**
+     * Performs connection to the database.
+     * Based on pooling options, it can either create connection immediately,
+     * either create a pool and create connection when needed.
+     */
+    async connect() {
+        if (this.options.replication) {
+            this.slaves = await Promise.all(this.options.replication.slaves.map((slave) => {
+                return this.createPool(this.options, slave);
+            }));
+            this.master = await this.createPool(this.options, this.options.replication.master);
+        }
+        else {
+            this.master = await this.createPool(this.options, this.options);
+        }
+        if (!this.database || !this.searchSchema) {
+            const queryRunner = this.createQueryRunner("master");
+            this.database ??= await queryRunner.getCurrentDatabase();
+            this.searchSchema ??= await queryRunner.getCurrentSchema();
+            await queryRunner.release();
+        }
+        this.schema ??= this.searchSchema;
+    }
+    /**
+     * Makes any action after connection (e.g. create extensions in Postgres driver).
+     */
+    async afterConnect() {
+        // enable time travel queries
+        if (this.options.timeTravelQueries) {
+            await this.dataSource.query(`SET default_transaction_use_follower_reads = 'on';`);
+        }
+        // enable experimental alter column type support (we need it to alter enum types)
+        await this.dataSource.query("SET enable_experimental_alter_column_type_general = true");
+        return Promise.resolve();
+    }
+    /**
+     * Closes connection with database.
+     */
+    async disconnect() {
+        if (!this.master) {
+            throw new ConnectionIsNotSetError_1.ConnectionIsNotSetError("cockroachdb");
+        }
+        await this.closePool(this.master);
+        await Promise.all(this.slaves.map((slave) => this.closePool(slave)));
+        this.master = undefined;
+        this.slaves = [];
+    }
+    /**
+     * 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 CockroachQueryRunner_1.CockroachQueryRunner(this, mode);
+    }
+    /**
+     * 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 (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 === "datetime" ||
+            columnMetadata.type === Date ||
+            columnMetadata.type === "timestamp" ||
+            columnMetadata.type === "timestamptz" ||
+            columnMetadata.type === "timestamp with time zone" ||
+            columnMetadata.type === "timestamp without time zone") {
+            return DateUtils_1.DateUtils.mixedDateToDate(value);
+        }
+        else if (["json", "jsonb", ...this.spatialTypes].indexOf(columnMetadata.type) >= 0) {
+            return JSON.stringify(value);
+        }
+        else if (columnMetadata.type === "simple-array") {
+            return DateUtils_1.DateUtils.simpleArrayToString(value);
+        }
+        else if (columnMetadata.type === "simple-json") {
+            return DateUtils_1.DateUtils.simpleJsonToString(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;
+        // unique_rowid() generates bigint value and should not be converted to number
+        if (([Number, "int4", "smallint", "int2"].some((v) => v === columnMetadata.type) &&
+            !columnMetadata.isArray) ||
+            columnMetadata.generationStrategy === "increment") {
+            value = parseInt(value);
+        }
+        else if (columnMetadata.type === Boolean) {
+            value = value ? true : false;
+        }
+        else if (columnMetadata.type === "datetime" ||
+            columnMetadata.type === Date ||
+            columnMetadata.type === "timestamp" ||
+            columnMetadata.type === "timestamptz" ||
+            columnMetadata.type === "timestamp with time zone" ||
+            columnMetadata.type === "timestamp without time zone") {
+            value = DateUtils_1.DateUtils.normalizeHydratedDate(value);
+        }
+        else if (columnMetadata.type === "date") {
+            value = DateUtils_1.DateUtils.mixedDateToDateString(value, {
+                utc: columnMetadata.utc,
+            });
+        }
+        else if (columnMetadata.type === "time") {
+            value = DateUtils_1.DateUtils.mixedTimeToString(value);
+        }
+        else if (columnMetadata.type === "simple-array") {
+            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") {
+            if (columnMetadata.isArray) {
+                if (value === "{}")
+                    return [];
+                if (Array.isArray(value))
+                    return value;
+                // manually convert enum array to array of values (pg does not support, see https://github.com/brianc/node-pg-types/issues/56)
+                value = value
+                    .slice(1, -1)
+                    .split(",")
+                    .map((val) => {
+                    // replace double quotes from the beginning and from the end
+                    if (val.startsWith(`"`) && val.endsWith(`"`))
+                        val = val.slice(1, -1);
+                    // replace escaped backslash and double quotes
+                    return val.replaceAll(/\\(\\|")/g, "$1");
+                });
+                // convert to number if that exists in possible enum options
+                value = value.map((val) => {
+                    return !isNaN(+val) &&
+                        columnMetadata.enum.indexOf(parseInt(val)) >= 0
+                        ? parseInt(val)
+                        : val;
+                });
+            }
+            else {
+                // convert to number if that exists in possible enum options
+                value =
+                    !isNaN(+value) &&
+                        columnMetadata.enum.indexOf(parseInt(value)) >= 0
+                        ? parseInt(value)
+                        : value;
+            }
+        }
+        if (columnMetadata.transformer)
+            value = ApplyValueTransformers_1.ApplyValueTransformers.transformFrom(columnMetadata.transformer, value);
+        return value;
+    }
+    /**
+     * 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];
+        const parameterIndexMap = new Map();
+        sql = sql.replaceAll(/:(\.\.\.)?([A-Za-z0-9_.]+)/g, (full, isArray, key) => {
+            if (!parameters.hasOwnProperty(key)) {
+                return full;
+            }
+            if (parameterIndexMap.has(key)) {
+                return this.parametersPrefix + parameterIndexMap.get(key);
+            }
+            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);
+            parameterIndexMap.set(key, escapedParameters.length);
+            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 schema name and table name.
+     * E.g. myDB.mySchema.myTable
+     *
+     * @param tableName
+     * @param schema
+     */
+    buildTableName(tableName, schema) {
+        const tablePath = [tableName];
+        if (schema) {
+            tablePath.unshift(schema);
+        }
+        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 = this.schema;
+        if (InstanceChecker_1.InstanceChecker.isTable(target) || InstanceChecker_1.InstanceChecker.isView(target)) {
+            // name is sometimes a path
+            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)) {
+            // referencedTableName is sometimes a path
+            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: driverDatabase,
+            schema: (parts.length > 1 ? parts[0] : undefined) ?? driverSchema,
+            tableName: parts.length > 1 ? parts[1] : parts[0],
+        };
+    }
+    /**
+     * Creates a database type from a given column metadata.
+     *
+     * @param column
+     * @param column.type
+     * @param column.length
+     * @param column.precision
+     * @param column.scale
+     * @param column.isArray
+     * @param column.isGenerated
+     * @param column.generationStrategy
+     */
+    normalizeType(column) {
+        if (column.type === Number ||
+            column.type === "integer" ||
+            column.type === "int" ||
+            column.type === "bigint" ||
+            column.type === "int64") {
+            return "int8";
+        }
+        else if (column.type === String ||
+            column.type === "character varying" ||
+            column.type === "char varying") {
+            return "varchar";
+        }
+        else if (column.type === Date ||
+            column.type === "timestamp without time zone") {
+            return "timestamp";
+        }
+        else if (column.type === "timestamp with time zone") {
+            return "timestamptz";
+        }
+        else if (column.type === "time without time zone") {
+            return "time";
+        }
+        else if (column.type === Boolean || column.type === "boolean") {
+            return "bool";
+        }
+        else if (column.type === "simple-array" ||
+            column.type === "simple-json" ||
+            column.type === "text") {
+            return "string";
+        }
+        else if (column.type === "bytea" || column.type === "blob") {
+            return "bytes";
+        }
+        else if (column.type === "smallint") {
+            return "int2";
+        }
+        else if (column.type === "numeric" || column.type === "dec") {
+            return "decimal";
+        }
+        else if (column.type === "double precision" ||
+            column.type === "float") {
+            return "float8";
+        }
+        else if (column.type === "real") {
+            return "float4";
+        }
+        else if (column.type === "character") {
+            return "char";
+        }
+        else if (column.type === "simple-enum") {
+            return "enum";
+        }
+        else if (column.type === "json") {
+            return "jsonb";
+        }
+        else {
+            return column.type || "";
+        }
+    }
+    /**
+     * Normalizes "default" value of the column.
+     *
+     * @param columnMetadata
+     */
+    normalizeDefault(columnMetadata) {
+        const defaultValue = columnMetadata.default;
+        if (defaultValue === undefined || defaultValue === null) {
+            return undefined;
+        }
+        if (typeof defaultValue === "function") {
+            const value = defaultValue();
+            if (value.toUpperCase() === "CURRENT_TIMESTAMP") {
+                return "current_timestamp()";
+            }
+            else if (value.toUpperCase() === "CURRENT_DATE") {
+                return "current_date()";
+            }
+            return value;
+        }
+        if ((columnMetadata.type === "enum" ||
+            columnMetadata.type === "simple-enum") &&
+            defaultValue !== undefined) {
+            if (columnMetadata.isArray) {
+                const enumName = this.buildEnumName(columnMetadata);
+                let arrayValue = defaultValue;
+                if (typeof defaultValue === "string") {
+                    if (defaultValue === "{}")
+                        return `ARRAY[]::${enumName}[]`;
+                    arrayValue = defaultValue
+                        .replace("{", "")
+                        .replace("}", "")
+                        .split(",");
+                }
+                if (Array.isArray(arrayValue)) {
+                    const expr = `ARRAY[${arrayValue
+                        .map((it) => `'${it}'`)
+                        .join(",")}]`;
+                    return `${expr}::${enumName}[]`;
+                }
+            }
+            else {
+                return `'${defaultValue}'`;
+            }
+        }
+        else if (typeof defaultValue === "number") {
+            return `(${defaultValue})`;
+        }
+        if (typeof defaultValue === "boolean") {
+            return defaultValue ? "true" : "false";
+        }
+        if (typeof defaultValue === "string") {
+            const arrayCast = columnMetadata.isArray
+                ? `::${columnMetadata.type}[]`
+                : "";
+            return `'${defaultValue}'${arrayCast}`;
+        }
+        if (ObjectUtils_1.ObjectUtils.isObject(defaultValue) && defaultValue !== null) {
+            return `'${JSON.stringify(defaultValue).replaceAll("'", "''")}'`;
+        }
+        return `${defaultValue}`;
+    }
+    /**
+     * Compares "default" value of the column.
+     *
+     * @param columnMetadata
+     * @param tableColumn
+     */
+    defaultEqual(columnMetadata, tableColumn) {
+        // defaults are equal if both are undefined or null
+        if ((columnMetadata.default === null ||
+            columnMetadata.default === undefined) &&
+            (tableColumn.default === null || tableColumn.default === undefined))
+            return true;
+        if (["json", "jsonb"].includes(columnMetadata.type) &&
+            !["function", "undefined"].includes(typeof columnMetadata.default)) {
+            return this.compareJsonDefaults(columnMetadata, tableColumn);
+        }
+        const columnDefault = this.lowerDefaultValueIfNecessary(this.normalizeDefault(columnMetadata));
+        return columnDefault === tableColumn.default;
+    }
+    /**
+     * Compares json/jsonb default values of the column.
+     *
+     * @param columnMetadata
+     * @param tableColumn
+     */
+    compareJsonDefaults(columnMetadata, tableColumn) {
+        let jsonString = tableColumn.default;
+        if (typeof jsonString === "string") {
+            jsonString = jsonString.trim();
+            if (jsonString.startsWith("e'") && // CockroachDB might escape JSON/JSONB default values with e'...'
+                jsonString.endsWith("'")) {
+                jsonString = jsonString
+                    .slice(2, -1)
+                    .replaceAll(String.raw `\'`, "'");
+            }
+            else if (jsonString.startsWith("'") && jsonString.endsWith("'")) {
+                jsonString = jsonString.slice(1, -1).replaceAll("''", "'");
+            }
+        }
+        if (typeof jsonString === "string") {
+            try {
+                const tableColumnDefaultParsed = JSON.parse(jsonString);
+                return OrmUtils_1.OrmUtils.deepCompare(columnMetadata.default, tableColumnDefaultParsed);
+            }
+            catch (err) {
+                if (!(err instanceof SyntaxError)) {
+                    throw new error_1.TypeORMError(`Failed to compare default values of ${columnMetadata.propertyName} column`);
+                }
+            }
+        }
+        else {
+            return OrmUtils_1.OrmUtils.deepCompare(columnMetadata.default, jsonString);
+        }
+        return false;
+    }
+    /**
+     * Normalizes "isUnique" value of the column.
+     *
+     * @param column
+     */
+    normalizeIsUnique(column) {
+        return column.entityMetadata.uniques.some((uq) => uq.columns.length === 1 && uq.columns[0] === column);
+    }
+    /**
+     * Returns default column lengths, which is required on column creation.
+     *
+     * @param column
+     */
+    getColumnLength(column) {
+        return column.length ? column.length.toString() : "";
+    }
+    /**
+     * Creates column type definition including length, precision and scale
+     *
+     * @param column
+     */
+    createFullType(column) {
+        let type = column.type;
+        if (column.length) {
+            type += "(" + column.length + ")";
+        }
+        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 + ")";
+        }
+        else if (this.spatialTypes.indexOf(column.type) >= 0) {
+            if (column.spatialFeatureType != null && column.srid != null) {
+                type = `${column.type}(${column.spatialFeatureType},${column.srid})`;
+            }
+            else if (column.spatialFeatureType != null) {
+                type = `${column.type}(${column.spatialFeatureType})`;
+            }
+            else {
+                type = column.type;
+            }
+        }
+        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.
+     */
+    async obtainMasterConnection() {
+        if (!this.master) {
+            throw new error_1.TypeORMError("Driver not Connected");
+        }
+        return new Promise((ok, fail) => {
+            this.master.connect((err, connection, release) => {
+                if (err) {
+                    fail(err);
+                }
+                else {
+                    ok([connection, release]);
+                }
+            });
+        });
+    }
+    /**
+     * 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.
+     */
+    async obtainSlaveConnection() {
+        if (!this.slaves.length)
+            return this.obtainMasterConnection();
+        const random = Math.floor(Math.random() * this.slaves.length);
+        return new Promise((ok, fail) => {
+            this.slaves[random].connect((err, connection, release) => {
+                if (err) {
+                    fail(err);
+                }
+                else {
+                    ok([connection, release]);
+                }
+            });
+        });
+    }
+    /**
+     * Creates generated map of values generated or returned by database after INSERT query.
+     *
+     * todo: slow. optimize Object.keys(), OrmUtils.mergeDeep and column.createValueMap parts
+     *
+     * @param metadata
+     * @param insertResult
+     */
+    createGeneratedMap(metadata, insertResult) {
+        if (!insertResult)
+            return undefined;
+        return Object.keys(insertResult).reduce((map, key) => {
+            const column = metadata.findColumnWithDatabaseName(key);
+            if (column) {
+                OrmUtils_1.OrmUtils.mergeDeep(map, column.createValueMap(this.prepareHydratedValue(insertResult[key], column)));
+            }
+            return map;
+        }, {});
+    }
+    /**
+     * 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
+            return (tableColumn.name !== columnMetadata.databaseName ||
+                tableColumn.type !== this.normalizeType(columnMetadata) ||
+                tableColumn.length !== columnMetadata.length ||
+                tableColumn.isArray !== columnMetadata.isArray ||
+                tableColumn.precision !== columnMetadata.precision ||
+                (columnMetadata.scale !== undefined &&
+                    tableColumn.scale !== columnMetadata.scale) ||
+                tableColumn.comment !==
+                    this.escapeComment(columnMetadata.comment) ||
+                (!tableColumn.isGenerated &&
+                    !this.defaultEqual(columnMetadata, tableColumn)) || // we included check for generated here, because generated columns already can have default values
+                tableColumn.isPrimary !== columnMetadata.isPrimary ||
+                tableColumn.isNullable !== columnMetadata.isNullable ||
+                tableColumn.isUnique !==
+                    this.normalizeIsUnique(columnMetadata) ||
+                tableColumn.enumName !== columnMetadata.enumName ||
+                !!(tableColumn.enum &&
+                    columnMetadata.enum &&
+                    !OrmUtils_1.OrmUtils.isArraysEqual(tableColumn.enum, columnMetadata.enum.map((val) => val + ""))) || // enums in postgres are always strings
+                tableColumn.isGenerated !== columnMetadata.isGenerated ||
+                tableColumn.generatedType !== columnMetadata.generatedType ||
+                (tableColumn.asExpression ?? "").trim() !==
+                    (columnMetadata.asExpression ?? "").trim() ||
+                // Spatial columns declared without a feature type or SRID are
+                // created as plain geometry, which is reported back as
+                // "Geometry" with SRID 0, so compare against those defaults.
+                (tableColumn.spatialFeatureType ?? "geometry").toLowerCase() !==
+                    (columnMetadata.spatialFeatureType ?? "geometry").toLowerCase() ||
+                (tableColumn.srid ?? 0) !== (columnMetadata.srid ?? 0));
+        });
+    }
+    lowerDefaultValueIfNecessary(value) {
+        if (!value) {
+            return value;
+        }
+        return value
+            .split(`'`)
+            .map((v, i) => {
+            return i % 2 === 1 ? v : v.toLowerCase();
+        })
+            .join(`'`);
+    }
+    /**
+     * Returns true if driver supports RETURNING / OUTPUT statement.
+     *
+     * @param _returningType
+     */
+    isReturningSqlSupported(_returningType) {
+        return true;
+    }
+    /**
+     * Returns true if driver supports uuid values generation on its own.
+     */
+    isUUIDGenerationSupported() {
+        return true;
+    }
+    /**
+     * Returns true if driver supports fulltext indices.
+     */
+    isFullTextColumnTypeSupported() {
+        return false;
+    }
+    /**
+     * Creates an escaped parameter.
+     *
+     * @param parameterName
+     * @param index
+     */
+    createParameter(parameterName, index) {
+        return this.parametersPrefix + (index + 1);
+    }
+    // -------------------------------------------------------------------------
+    // Public Methods
+    // -------------------------------------------------------------------------
+    /**
+     * Loads postgres query stream package.
+     */
+    loadStreamDependency() {
+        try {
+            return PlatformTools_1.PlatformTools.load("pg-query-stream");
+        }
+        catch {
+            // todo: better error for browser env
+            throw new error_1.TypeORMError(`To use streams you should install pg-query-stream package. Please run "npm i pg-query-stream".`);
+        }
+    }
+    // -------------------------------------------------------------------------
+    // Protected Methods
+    // -------------------------------------------------------------------------
+    /**
+     * If driver dependency is not given explicitly, then try to load it via "require".
+     */
+    loadDependencies() {
+        try {
+            const postgres = this.options.driver ?? PlatformTools_1.PlatformTools.load("pg");
+            this.postgres = postgres;
+            try {
+                const pgNative = this.options.nativeDriver ?? PlatformTools_1.PlatformTools.load("pg-native");
+                if (pgNative && this.postgres.native)
+                    this.postgres = this.postgres.native;
+            }
+            catch (e) { }
+        }
+        catch (e) {
+            // todo: better error for browser env
+            throw new DriverPackageNotInstalledError_1.DriverPackageNotInstalledError("Postgres", "pg");
+        }
+    }
+    /**
+     * Creates a new connection pool for a given database credentials.
+     *
+     * @param options
+     * @param credentials
+     */
+    async createPool(options, credentials) {
+        credentials = Object.assign({}, credentials, DriverUtils_1.DriverUtils.buildDriverOptions(credentials)); // todo: do it better way
+        // build connection options for the driver
+        const connectionOptions = Object.assign({}, {
+            host: credentials.host,
+            user: credentials.username,
+            password: credentials.password,
+            database: credentials.database,
+            port: credentials.port,
+            ssl: credentials.ssl,
+            application_name: options.applicationName,
+            max: options.poolSize,
+        }, options.extra ?? {});
+        // create a connection pool
+        const pool = new this.postgres.Pool(connectionOptions);
+        const { logger } = this.dataSource;
+        const poolErrorHandler = options.poolErrorHandler ??
+            ((error) => logger.log("warn", `Postgres pool raised an error. ${error}`));
+        /*
+          Attaching an error handler to pool errors is essential, as, otherwise, errors raised will go unhandled and
+          cause the hosting app to crash.
+         */
+        pool.on("error", poolErrorHandler);
+        return new Promise((ok, fail) => {
+            pool.connect((err, connection, release) => {
+                if (err)
+                    return fail(err);
+                release();
+                ok(pool);
+            });
+        });
+    }
+    /**
+     * Closes connection pool.
+     *
+     * @param pool
+     */
+    async closePool(pool) {
+        while (this.connectedQueryRunners.length) {
+            await this.connectedQueryRunners[0].release();
+        }
+        return new Promise((ok, fail) => {
+            pool.end((err) => (err ? fail(err) : ok()));
+        });
+    }
+    /**
+     * Escapes a given comment.
+     *
+     * @param comment
+     */
+    escapeComment(comment) {
+        if (!comment)
+            return comment;
+        comment = comment.replaceAll("'", "''").replaceAll("\u0000", ""); // Null bytes aren't allowed in comments
+        return comment;
+    }
+    /**
+     * Builds ENUM type name from given table and column.
+     *
+     * @param column
+     */
+    buildEnumName(column) {
+        const { schema, tableName } = this.parseTableName(column.entityMetadata);
+        let enumName = column.enumName ??
+            `${tableName}_${column.databaseName.toLowerCase()}_enum`;
+        if (schema)
+            enumName = `${schema}.${enumName}`;
+        return enumName
+            .split(".")
+            .map((i) => {
+            return `"${i}"`;
+        })
+            .join(".");
+    }
+}
+exports.CockroachDriver = CockroachDriver;
+//# sourceMappingURL=CockroachDriver.js.map