Merge pull request #2029 from taozhi8833998/fix-type

fix: type definition
taozhi8833998 · Jul 29, 2024 · d4f8ea1 · d4f8ea1
2 parents 3f9c7b3 + 47162ff
commit d4f8ea1
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/ast/postgresql.ts b/ast/postgresql.ts
@@ -125,13 +125,15 @@ export type create_function_stmt_t = {
 
 export type create_function_stmt = AstStatement<create_function_stmt_t>;
 
+export type create_type_stmt_option = { as: 'as'; resource: string; create_definitions: expr_list | create_column_definition_list; } | ignore;
+
 export type create_type_stmt_t = {
         type: 'create',
         keyword: 'type',
         name: { schema: string; name: string },
         as?: string,
         resource?: string,
-        create_definitions?: any
+        create_definitions?: expr_list | create_column_definition_list;
       }
 
 export type create_type_stmt = AstStatement<create_type_stmt_t>;
@@ -294,6 +296,8 @@ export type column_definition_opt_list = {
         reference_definition?: reference_definition;
       };
 
+export type create_column_definition_list = create_column_definition[];
+
 
 
 export type create_column_definition = {

diff --git a/pegjs/postgresql.pegjs b/pegjs/postgresql.pegjs
@@ -734,6 +734,7 @@ create_function_stmt
 
 create_type_stmt_option
   = KW_AS __ r:(KW_ENUM / 'RANGE'i) __ LPAREN __ e:expr_list? __ RPAREN {
+    // => { as: 'as'; resource: string; create_definitions: expr_list | create_column_definition_list; }
     e.parentheses = true
     return {
       as: 'as',
@@ -742,6 +743,7 @@ create_type_stmt_option
     }
   }
   / KW_AS __ LPAREN __ e:create_column_definition_list? __ RPAREN {
+    // => ignore
     return {
       as: 'as',
       create_definitions: e,