Merge pull request #2528 from taozhi8833998/feat-generated-by-default-pg

feat: support generate by default in pg

Author: taozhi8833998

pegjs/postgresql.pegjs

@@ -1295,6 +1295,13 @@ column_definition_opt
     // => { comment: keyword_comment; }
     return { comment: co }
   }
+  / gbd:('GENERATED'i)? __ 'BY'i __ 'DEFAULT'i __ 'AS'i __ 'IDENTITY'i {
+    // => { generated_by_default: { type: 'origin', value: string } }
+    const sql = []
+    if (gbd) sql.push('generated')
+    sql.push('by', 'default', 'as', 'identity')
+    return { generated_by_default: { type: 'origin', value: sql.join(' ').toLowerCase('') } }
+  }
   / ca:collate_expr {
     // => { collate: collate_expr; }
     return { collate: ca }

src/column.js

@@ -111,6 +111,7 @@ function columnOption(definition) {
     primary_key: primaryKey,
     column_format: columnFormat,
     reference_definition: referenceDefinition,
+    generated_by_default: generateByDefault,
   } = definition
   const nullSQL = [toUpper(nullable && nullable.action), toUpper(nullable && nullable.value)].filter(hasVal).join(' ')
   if (!generated) columnOpt.push(nullSQL)
@@ -123,7 +124,7 @@ function columnOption(definition) {
   columnOpt.push(constraintDefinitionToSQL(check))
   columnOpt.push(generatedExpressionToSQL(generated))
   if (generated) columnOpt.push(nullSQL)
-  columnOpt.push(autoIncrementToSQL(autoIncrement), toUpper(primaryKey), toUpper(uniqueKey), commentToSQL(comment))
+  columnOpt.push(autoIncrementToSQL(autoIncrement), toUpper(primaryKey), toUpper(uniqueKey), literalToSQL(generateByDefault), commentToSQL(comment))
   columnOpt.push(...commonTypeValue(characterSet))
   if (database.toLowerCase() !== 'sqlite') columnOpt.push(exprToSQL(collate))
   columnOpt.push(...commonTypeValue(columnFormat))

test/postgres.spec.js

@@ -1913,6 +1913,15 @@ describe('Postgres', () => {
         `SELECT "p".description AS "description", "p".invoice_date AS "invoice_date" FROM "anonymized_schema"."payments" AS "p" INNER JOIN "anonymized_schema"."entities" AS "e" ON "p".entity_id = "e".entity_id LEFT JOIN "anonymized_schema"."entities" AS "pe" ON "p".parent_entity_id = "pe".entity_id LEFT JOIN "anonymized_schema"."entities" AS "ppe" ON "pe".parent_ids[array_upper("pe".parent_ids, 1)] = "ppe".entity_id WHERE "e".entity_type IN ('A', 'B') AND "p".invoice_date >= '2025-08-01' AND "p".invoice_date < '2025-09-01' AND CAST("p".amount AS NUMERIC) > 0`
       ]
     },
+    {
+      title: 'generate by default',
+      sql: [
+        `CREATE TABLE "t_logs" (
+          "log_id" INTEGER NOT NULL UNIQUE GENERATED BY DEFAULT AS IDENTITY
+        );`,
+        'CREATE TABLE "t_logs" ("log_id" INTEGER NOT NULL UNIQUE GENERATED BY DEFAULT AS IDENTITY)'
+      ]
+    },
   ]
   function neatlyNestTestedSQL(sqlList){
     sqlList.forEach(sqlInfo => {