fix: custom type support schema prefix in pg

Author: taozhi8833998

pegjs/postgresql.pegjs

@@ -6272,7 +6272,11 @@ network_address_type
   = t:(KW_INET / KW_CIDR / KW_MACADDR8 / KW_MACADDR) {/* =>  data_type */  return { dataType: t }}
 
 custom_types
-  = name:ident_name &{ return customTypes.has(name) } {
+  = schema:ident_name __ DOT __ name:ident_without_kw &{ return customTypes.has(`${schema}.${name}`) }  {
+      // => data_type
+      return { schema: schema, dataType: name }
+  }
+  / name:ident_name &{ return customTypes.has(name) } {
       // => data_type
       return { dataType: name }
   }

src/column.js

@@ -65,9 +65,9 @@ function columnRefToSQL(expr) {
 
 function columnDataType(definition) {
   if (!definition) return
-  const { dataType, length, suffix, scale, expr } = definition
+  const { schema, dataType, length, suffix, scale, expr } = definition
   const parentheses = length != null && true || false
-  let result = dataTypeToSQL({ dataType, length, suffix, scale, parentheses })
+  let result = dataTypeToSQL({ schema, dataType, length, suffix, scale, parentheses })
   if (expr) result += exprToSQL(expr)
   if (definition.array) {
     const arrayExpr = arrayDimensionToSymbol(definition)

src/util.js

@@ -292,12 +292,13 @@ function onPartitionsToSQL(expr) {
 }
 
 function dataTypeToSQL(expr) {
-  const { dataType, length, parentheses, scale, suffix } = expr
+  const { schema, dataType, length, parentheses, scale, suffix } = expr
   let str = ''
   if (length != null) str = scale ? `${length}, ${scale}` : length
   if (parentheses) str = `(${str})`
   if (suffix && suffix.length) str += ` ${suffix.join(' ')}`
-  return `${dataType}${str}`
+  const prefix = schema ? `${schema}.` : ''
+  return `${prefix}${dataType}${str}`
 }
 
 function arrayStructTypeToSQL(expr) {

test/postgres.spec.js

@@ -2322,6 +2322,16 @@ describe('Postgres', () => {
           'CREATE TABLE "t_pgsql_generate_test" (id_col SERIAL NOT NULL PRIMARY KEY, small_int_col SMALLINT NOT NULL, small_serial_col SMALLSERIAL NOT NULL, integer_type_col INTEGER NOT NULL, big_int_col BIGINT NOT NULL)'
         ]
       },
+      {
+        title: 'custom data type for create table',
+        sql: [
+          `create type public.Gender AS ENUM ('MALE', 'FEMALE');
+            create table users (
+              gender public."Gender"
+            );`,
+          `CREATE TYPE "public"."Gender" AS ENUM ('MALE', 'FEMALE') ; CREATE TABLE "users" (gender public.Gender)`
+        ]
+      },
     ]
     neatlyNestTestedSQL(SQL_LIST)
   })