Merge pull request #2050 from taozhi8833998/fix-quoted-data-type-pg

fix: quoted column data type in pg
taozhi8833998 · Aug 3, 2024 · 650501f · 650501f
2 parents 94393f6 + 99cd4ae
commit 650501f
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/pegjs/postgresql.pegjs b/pegjs/postgresql.pegjs
@@ -1315,7 +1315,7 @@ create_column_definition_list
 
 create_column_definition
   = c:column_ref __
-    d:data_type __
+    d:(data_type / double_quoted_ident) __
     cdo:column_definition_opt_list? {
       /*
       => {
@@ -1335,6 +1335,7 @@ create_column_definition
       }
       */
       columnList.add(`create::${c.table}::${c.column.expr.value}`)
+      if (d.type === 'double_quote_string') d = { dataType: `"${d.value}"` }
       return {
         column: c,
         definition: d,

diff --git a/test/postgres.spec.js b/test/postgres.spec.js
@@ -1595,6 +1595,13 @@ describe('Postgres', () => {
         `CREATE TABLE "public"."tnotok" ("id" SERIAL CONSTRAINT users_PK PRIMARY KEY, description TEXT NOT NULL DEFAULT ''::TEXT)`,
       ]
     },
+    {
+      title: 'create table with quoted data type',
+      sql: [
+        'CREATE TABLE "User" ("gender" "Gender" NOT NULL);',
+        'CREATE TABLE "User" ("gender" "Gender" NOT NULL)',
+      ]
+    },
   ]
   function neatlyNestTestedSQL(sqlList){
     sqlList.forEach(sqlInfo => {