Merge pull request #6714 from nocodb/nc-fix/2561-pg

feat: support for pg identity column

packages/nocodb/src/db/sql-client/lib/pg/PgClient.ts

@@ -781,6 +781,18 @@ class PGClient extends KnexClient {
     const result = new Result();
     log.api(`${_func}:args:`, args);
     try {
+      const versionQuery = await this.sqlClient.raw('SELECT version()');
+
+      // Example output of `SELECT version()`
+      // PostgreSQL 14.4 (Debian 14.4-1.pgdg110+1) on x86_64-pc-linux-gnu, compiled by gcc (Debian 10.2.1-6) 10.2.1 20210110, 64-bit
+      const majorVersion = versionQuery.rows[0]?.version
+        ?.split(' ')?.[1]
+        ?.split('.')?.[0];
+
+      // PostgreSQL 10 and above supports identity columns
+      const identitySelector =
+        +majorVersion >= 10 ? 'c.is_identity as ii,' : '';
+
       args.databaseName = this.connectionConfig.connection.database;
       const response = await this.sqlClient.raw(
         `select
@@ -800,6 +812,7 @@ class PGClient extends KnexClient {
                     -- c.collation_name as clnn,
                     pk.ordinal_position as pk_ordinal_position, pk.constraint_name as pk_constraint_name,
                     c.udt_name,
+                    ${identitySelector}
 
        (SELECT count(*)
             FROM INFORMATION_SCHEMA.TABLE_CONSTRAINTS tc1
@@ -893,6 +906,13 @@ class PGClient extends KnexClient {
           column.dtxp = response.rows[i].enum_values;
         }
 
+        // handle identity column
+        if (+majorVersion >= 10) {
+          if (response.rows[i].ii === 'YES') {
+            column.ai = true;
+          }
+        }
+
         columns.push(column);
       }