typeorm 0.3.27-dev.fa3cd43 → 0.3.28-dev.02e7b71

package/driver/postgres/PostgresDriver.d.ts

@@ -83,8 +83,7 @@ export declare class PostgresDriver implements Driver {
     /**
      * Gets list of supported column data types by a driver.
      *
-     * @see https://www.tutorialspoint.com/postgresql/postgresql_data_types.htm
-     * @see https://www.postgresql.org/docs/9.2/static/datatype.html
+     * @see https://www.postgresql.org/docs/current/datatype.html
      */
     supportedDataTypes: ColumnType[];
     /**
@@ -147,6 +146,7 @@ export declare class PostgresDriver implements Driver {
         hasCubeColumns: boolean;
         hasGeometryColumns: boolean;
         hasLtreeColumns: boolean;
+        hasVectorColumns: boolean;
         hasExclusionConstraints: boolean;
         hasExtensions: boolean;
     }>;

package/driver/postgres/PostgresDriver.js

@@ -45,8 +45,7 @@ class PostgresDriver {
         /**
          * Gets list of supported column data types by a driver.
          *
-         * @see https://www.tutorialspoint.com/postgresql/postgresql_data_types.htm
-         * @see https://www.postgresql.org/docs/9.2/static/datatype.html
+         * @see https://www.postgresql.org/docs/current/datatype.html
          */
         this.supportedDataTypes = [
             "int",
@@ -105,6 +104,7 @@ class PostgresDriver {
             "xml",
             "json",
             "jsonb",
+            "jsonpath",
             "int4range",
             "int8range",
             "numrange",
@@ -121,6 +121,8 @@ class PostgresDriver {
             "geography",
             "cube",
             "ltree",
+            "vector",
+            "halfvec",
         ];
         /**
          * Returns type of upsert supported by driver if any
@@ -141,6 +143,8 @@ class PostgresDriver {
             "bit",
             "varbit",
             "bit varying",
+            "vector",
+            "halfvec",
         ];
         /**
          * Gets list of column data types that support precision by a driver.
@@ -288,7 +292,7 @@ class PostgresDriver {
     }
     async enableExtensions(extensionsMetadata, connection) {
         const { logger } = this.connection;
-        const { hasUuidColumns, hasCitextColumns, hasHstoreColumns, hasCubeColumns, hasGeometryColumns, hasLtreeColumns, hasExclusionConstraints, } = extensionsMetadata;
+        const { hasUuidColumns, hasCitextColumns, hasHstoreColumns, hasCubeColumns, hasGeometryColumns, hasLtreeColumns, hasVectorColumns, hasExclusionConstraints, } = extensionsMetadata;
         if (hasUuidColumns)
             try {
                 await this.executeQuery(connection, `CREATE EXTENSION IF NOT EXISTS "${this.options.uuidExtension || "uuid-ossp"}"`);
@@ -331,6 +335,13 @@ class PostgresDriver {
             catch (_) {
                 logger.log("warn", "At least one of the entities has a ltree column, but the 'ltree' extension cannot be installed automatically. Please install it manually using superuser rights");
             }
+        if (hasVectorColumns)
+            try {
+                await this.executeQuery(connection, `CREATE EXTENSION IF NOT EXISTS "vector"`);
+            }
+            catch (_) {
+                logger.log("warn", "At least one of the entities has a vector column, but the 'vector' extension (pgvector) cannot be installed automatically. Please install it manually using superuser rights");
+            }
         if (hasExclusionConstraints)
             try {
                 // The btree_gist extension provides operator support in PostgreSQL exclusion constraints
@@ -361,6 +372,9 @@ class PostgresDriver {
             return (metadata.columns.filter((column) => column.type === "ltree")
                 .length > 0);
         });
+        const hasVectorColumns = this.connection.entityMetadatas.some((metadata) => {
+            return metadata.columns.some((column) => column.type === "vector" || column.type === "halfvec");
+        });
         const hasExclusionConstraints = this.connection.entityMetadatas.some((metadata) => {
             return metadata.exclusions.length > 0;
         });
@@ -371,6 +385,7 @@ class PostgresDriver {
             hasCubeColumns,
             hasGeometryColumns,
             hasLtreeColumns,
+            hasVectorColumns,
             hasExclusionConstraints,
             hasExtensions: hasUuidColumns ||
                 hasCitextColumns ||
@@ -378,6 +393,7 @@ class PostgresDriver {
                 hasGeometryColumns ||
                 hasCubeColumns ||
                 hasLtreeColumns ||
+                hasVectorColumns ||
                 hasExclusionConstraints,
         };
     }
@@ -432,6 +448,15 @@ class PostgresDriver {
         else if (["json", "jsonb", ...this.spatialTypes].indexOf(columnMetadata.type) >= 0) {
             return JSON.stringify(value);
         }
+        else if (columnMetadata.type === "vector" ||
+            columnMetadata.type === "halfvec") {
+            if (Array.isArray(value)) {
+                return `[${value.join(",")}]`;
+            }
+            else {
+                return value;
+            }
+        }
         else if (columnMetadata.type === "hstore") {
             if (typeof value === "string") {
                 return value;
@@ -505,6 +530,16 @@ class PostgresDriver {
         else if (columnMetadata.type === "time") {
             value = DateUtils_1.DateUtils.mixedTimeToString(value);
         }
+        else if (columnMetadata.type === "vector" ||
+            columnMetadata.type === "halfvec") {
+            if (typeof value === "string" &&
+                value.startsWith("[") &&
+                value.endsWith("]")) {
+                if (value === "[]")
+                    return [];
+                return value.slice(1, -1).split(",").map(Number);
+            }
+        }
         else if (columnMetadata.type === "hstore") {
             if (columnMetadata.hstoreType === "object") {
                 const unescapeString = (str) => str.replace(/\\./g, (m) => m[1]);
@@ -860,6 +895,10 @@ class PostgresDriver {
                 type = column.type;
             }
         }
+        else if (column.type === "vector" || column.type === "halfvec") {
+            type =
+                column.type + (column.length ? "(" + column.length + ")" : "");
+        }
         if (column.isArray)
             type += " array";
         return type;